Untitled

public ArrayList<Integer> solve(ArrayList<Integer> A) {
        ArrayList<Integer> countlist=  new  ArrayList<Integer>();
        for (int i = 0; i<= A.size()+1; i++){
             countlist.add(0);
        }
        for(int i=1;i<=A.size()+1;i++){
                for(int j=i;j<=A.size()+1;j+=i){
                    countlist.set(j, countlist.get(j)+1);
                }
        }
        ArrayList<Integer> ans = new ArrayList();
         for(int i = 2; i <= A.size()+1; i++){
             ans.add(countlist.get(i));
        }
        return ans;
    }