标题: 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的数目,这就是最终结果
}
}