蓝桥杯第7届省赛java b组 K倍区间

标题:k倍区间
给定一个长度为N的数列,A1,A2…AN,如果其中一段连续的子序列Ai, Ai+1,…Aj(i<=j)之和是K的倍数，我们就称这个区间[i,i]是K倍区间.
你能求出数列中总共有多少个K倍区间吗?

输入

第一行包含两个整数N和K.(1 <=N,K <= 100000)以下N行每行包含一个整数Ai.(1 <=Ai<= 100000)

输出

输出一个整数，代表K倍区间的数目．

例如，输入:5 2
1
2
3
4
5
程序应该输出:6
资源约定:
峰值内存消耗（含虚拟机)<256M
CPU消耗<2000ms

这道题有时间限制，所以其实还是考察我们对算法的优化

解法1：

暴力解法：
生成两个数组a[],s[]，一个用于存放数列，一个用于存放前n项和
这样通过遍历s[i]-s[j]就可以获得所有的子项的和，再把子项和对k求余，就可以获得结果

public class test {
    static int n,k;
    static int[] a = new int[n+1];
    static long[] s = new long[n+1];

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        n = sc.nextInt();
        k = sc.nextInt();
        s[0] = 0;
        for (int i = 1 ;i<=n;i++){
            a[i]=sc.nextInt();
            s[i]=s[i-1]+a[i];
        }
        
        long ans=0;
        for (int i = 1; i <=n ; i++) {
            for (int j = i; j<=n  ; j++) {
                if ((s[j]-s[i-1])%k ==0)
                    ans++;
            }
        }
        System.out.println(ans);
    }
}

解法2：

s[i]表示这个数列的前i项和，对每个s[i]对k取余， 对余数相同的结果进行分组，余数相同的结果分到一组里，每两个余数相同的结果进行做差(s[i]-s[j])，的到的结果对k取余都为0

我们可以用Hashmap来存放这个结果：

注意：
Hashmap中存放的键值对的类型必须是对象！！！！
如果取余后没有x这个余数，那么Hashmap中key=x的value=null，需要手动对value赋值0，否则会出现没有指向的错误！！！！！

public class test {
    static int n,k;
    static int[] a ;//= new int[n+1];
    static int[] s ;//= new int[n+1];
    static Map<Integer,Long> cnt = new HashMap<>();

    public static void main(String[] args){
        Scanner sc = new Scanner(System.in);
        n = sc.nextInt();
        k = sc.nextInt();
        a=new int[n+1];
        s=new int[n+1];
        s[0] = 0;
        cnt.put(0,1l);
        for (int i = 1 ;i<=n;i++){
            a[i]=sc.nextInt();
            s[i]=(s[i-1]+a[i])%k;
            if(cnt.get(s[i])==null){
                cnt.put(s[i],1l);
            }
            else {
                cnt.put(s[i],cnt.get(s[i])+1);
            }
        }
        
        long ans=0;
        for (int i=0;i<k;i++){
            Long cntI=cnt.get(i);
            if (cntI==null)
                cntI=0l;
            ans+=cntI*(cntI-1)/2;
        }
        System.out.println(ans);
    }
}