标题: k倍区间

给定一个长度为N的数列,A1, A2, ... AN,如果其中一段连续的子序列Ai, Ai+1, ... Aj(i <= j)之和是K的倍数,我们就称这个区间[i, j]是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


请严格按要求输出,不要画蛇添足地打印类似:“请您输入...” 的多余内容。

所有代码放在同一个源文件中,调试通过后,拷贝提交该源码。
不要使用package语句。不要使用jdk1.7及以上版本的特性。
主类的名字必须是:Main,否则按无效代码处理。

题意:不说了~~

题解:设区间[l,r],用前缀和表示区间的和为,sum[r]-sum[l-1],则:(sum[r]-sum[l-1])%k=0,就是sum[r]%k=sum[l-1]%k,这样就转换成求前缀和模k相等的数有几个然后利用排列组合的代码写法求和,最后不要忘记加上,sum[i]%k=0的数目,为什么呢?

看样例:1 2 3 4 5

sum[1]%k=1 前面没有出现 1   总和+0;

sum[2]%k=1 前面出现一次 1   总和+1;

sum[3]%k=0 前面没有出现 0   总和+0;

sum[4]%k=0 前面出现一次 0   总和+1;

sum[5]%k=1 前面出现两次 1   总和+2;    

这是排列组合的写法,结果为4,但是答案为6,因为还要加上sum[i]%k=0的数目 即:4+2=6,因为区间[4,4] ,是用公式sum[4]%k=sum[4-1]%k 求出来的,但是为0的情况很特殊,这里是:[1,3],[1,4] ,即前缀和自己为0也是可以的,相当于sum[3]%k=sum[1-1]%k。

公式:最终结果=排列组合结果+前缀和模k等于0的数目

上代码:

import java.util.*;
public class Main {
    static Scanner cin = new Scanner(System.in);
    static int [] a = new int [100002];
    static int [] dp = new int [100002];
    public static void main(String[] args){
    	int n,k;
    	n=cin.nextInt();
    	k=cin.nextInt();
    	long ans=0;//我觉得要用long,结果int也能过,不过无所谓了,为了保险还是用long吧
    	for (int i = 1; i <= n;i++) {
    		int tmp=cin.nextInt();
    		dp[i]=(tmp+dp[i-1])%k;//前缀和模k,在计算前缀和的过程中模k是可以的
    		ans+=a[dp[i]];//排列组合的写法
    		a[dp[i]]++;//排列组合的写法
            //排列组合写法,用来求和,这就是它的用处
    	}
    	System.out.println(ans+a[0]);//排列组合结果加上前缀和本身模k等于0的数目,这就是最终结果
    }
}