根据分类分别取数据算法

好久没看到算法了，整天就是ssh来ssh去，基本已经忘了曾经对算法，对数据结构的那种乐趣与解出一个问题的快乐。

这周有一个需求，要从一个表里selece数据，但需根据分类分别取记录，每个分类的记录需根据日期字段排序。

总共取10条记录，但有个分类可能存在成百上千条，而有的只有1-2条，而且总共多少分类也是不确定的，下面是两个同事写的两个算法，其中一个还存在问题，而有一个基本可以满足需求，先整理出测试代码如下：

 

package test.java; import java.util.ArrayList; import java.util.HashMap; import java.util.Iterator; import java.util.List; import java.util.Vector; public class randget { /** * @param args */ public static void main(String[] args) { getTop(); selectTop(); } /** * 存在一些问题 */ public static void getTop(){ int[] count = new int[]{5,2,1,1,1};//注意这里必须是已经降序排序好的 int sumtmp=10; Vector v=new Vector(); if(count.length==10){ for(int i=0;i<10;i++){ v.add(1); } }else if(count.length>0){ int j=0; //记录当前已经获取到的总数 int top=0; //每个模块中取多少个 if(count.length>5){ top=sumtmp/count.length; }else{ top=sumtmp/count.length+1; } for(int i=count.length-1;i>=0;i--){ int ccount=count[i]; if(ccount>top){ //总数大于平均top if(i==0){ //list全部遍历完毕则退出for循环 v.add(sumtmp-j);//最后一次取出sumtmp-j break; }else{ j+=top; v.add(top); } if(sumtmp-j>5){ top=(sumtmp-j)/i; }else{ if(i>0) top=(sumtmp-j)/i+1; else top=sumtmp-j; } }else{ //总数小于平均top ，直接添加 v.add(ccount); j+=ccount; } } } System.out.print("getTop:"); for(int k=v.size()-1;k>=0;k--){ int n =count.length-1; System.out.print(count[n-k]+"="+v.get(k)+" "); } System.out.println(); } /** * 这个测试通过 */ public static void selectTop(){ int nPerCat=1;//每个模块显示的条目数 int nTotalCat=nPerCat*10;//TAB WIN 要显示的条目数 //2.获取当前精华各个模块有多少条记录 List countCats = new ArrayList(); //实例初始化 countCats.add(new Object[]{1000,1}); countCats.add(new Object[]{2000,1}); countCats.add(new Object[]{3000,5}); countCats.add(new Object[]{4000,2}); countCats.add(new Object[]{5000,1}); int nTotal=countCats.size(); int currentCoutPut=0; HashMap showCats=new HashMap(); int getType=0;//0表示获取模式，先取少于等于每个模块显示的条目数， 1表示大于每个模块显示的条目数 for (int i=0;i<countCats.size();i++){ Object[] o= (Object[])countCats.get(i); int catid=Integer.parseInt(o[0].toString()); int catcount=Integer.parseInt(o[1].toString()); if ( getType==0 && catcount <= nPerCat ){ showCats.put(catid, catcount); countCats.remove(i); i--; currentCoutPut+=catcount; }else if (getType==1){ if (showCats.get(catid)!=null){ int c=(Integer)showCats.get(catid); if (c==catcount){ countCats.remove(i); i--; }else if (c+nPerCat<=catcount){ showCats.put(catid, c+nPerCat); currentCoutPut+=nPerCat; } else{ showCats.put(catid, c++); currentCoutPut++; } }else{ showCats.put(catid, nPerCat); currentCoutPut+=nPerCat; } } if (currentCoutPut==nTotalCat+1 || countCats.size()<1) break; //+1多取出一条用来显示“更多”链接 if (i+1==countCats.size() && currentCoutPut<nTotalCat+1){ getType=1; i=-1; } } System.out.print("selectTop:"); for(Iterator iter =showCats.keySet().iterator();iter.hasNext(); ){ Integer key = (Integer)iter.next(); Integer count = (Integer)showCats.get(key); System.out.print(key+"="+count+" "); } System.out.println(); } }

测试结果：

getTop:5=5 2=2 1=1 1=1 1=1
selectTop:1000=1 2000=1 3000=5 4000=2 5000=1

第一种算法查询出来的数据总和为：5+2+1+1+1=10

第二种算法查询出来的数据总和为：1+1+5+2+1=10

但是：

当getTop算法里初始值为：

int[] count = new int[]{5,4,4,3,2};  //注意这里一定是排序好的

selectTop算法里初始值为：

 List countCats = new ArrayList();  //实例初始化
     countCats.add(new Object[]{1000,5});
     countCats.add(new Object[]{2000,4});
     countCats.add(new Object[]{3000,4});
     countCats.add(new Object[]{4000,3});
     countCats.add(new Object[]{5000,2});

输出测试结果：

getTop:5=0 4=2 4=3 3=3 2=2
selectTop:1000=3 2000=2 3000=2 4000=2 5000=2

 

虽然看起来第一种算法在效率上要高于第二种算法，但是第一种算法考虑并不全面，故还需改进。