分页工具显示当前页，前几页，后几页

小灰灰 2022-03-25 15:04 302阅读 0赞

有次面试，说让写一段代码，输入当前页码，显示当前页码前几页，后几页。当时时间紧张，逻辑是写出来了，可是边界有问题，现在闲来无事，就写了一个。下面的代码显示的是当前页码的前两页，当前页，后两页。

package test;
    
    import java.util.Scanner;
    
    public class TestPageSize {
    
    	/**
    	 * 根据当前页码，显示页码工具栏，显示当前页码的前两页、当前页和后两页
    	 * @param curPage   当前页码
    	 * @param totalPage  总条数
    	 * @return
    	 */
    	public String getPageSize(int curPage,int totalPage){
    		
    		String pages = "";
    		if(totalPage>5){//总条数大于5
    			pages = "1 2 3 4 5";
    			if(1==curPage||2==curPage||3==curPage){//当前页码1，2，3都直接显示1-5
    				return pages;
    			}else{//当前页码>=4
    				if(curPage>(totalPage-2)){//当 当前页码是最后两页，不再计算，就显示 总条数-4 - 总条数
    					pages = (totalPage-4)+" "+(totalPage-3)+" "+(totalPage-2)+" "+(totalPage-1)+" "+totalPage;
    				}else{//当  4=<当前页码<总条数-2  计算显示 当前页码的前两页、当前页和后两页
    					pages = (curPage-2)+" "+(curPage-1)+" "+curPage+" "+(curPage+1)+" "+(curPage+2);
    				}
    				return pages;
    			}		
    		}else {//总条数小于5，直接显示1-总条数
    			for(int i=1;i<totalPage+1;i++){
    				pages += i+" ";
    			}
    			return pages;
    		}
    	}
    	
    	
    	public static void main(String[] args) {
    		System.out.println("请输入总页码：");
    		Scanner in = new Scanner(System.in);
    		String total = in.nextLine();
    		while(true){
    			System.out.println("请输入当前页码：");
    			String curPage = in.nextLine();
    			TestPageSize testO = new TestPageSize();
    			String pages = testO.getPageSize(Integer.parseInt(curPage),Integer.parseInt(total));
    			System.out.println(pages);
    		}
    	}
    }