标题: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);
}
}