【前缀和】K倍区间

【前缀和】 k 倍区间

题目描述

给定一个长度为 $N$ 的数列，$A_1,A_2,\cdots A_N$，如果其中一段连续的子序列 $A_i,A_{i+1},\cdots A_j(i\le j)$ 之和是 $K$ 的倍数，我们就称这个区间 $[i,j]$ 是 $K$ 倍区间。

你能求出数列中总共有多少个 $K$ 倍区间吗？

输入格式

第一行包含两个整数 $N$ 和 $K$$(1\le N,K\le 10^5)$。

以下 $N$ 行每行包含一个整数 $A_i$$(1\le A_i\le 10^5)$。

输出格式

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

样例 #1

样例输入 #1

5 2
1  
2  
3  
4  
5

样例输出 #1

6

提示

时限 2 秒, 256M。蓝桥杯 2017 年第八届

题解

import java.util.Scanner;

public class Main {
	public static void main(String[] args) {
		Scanner in = new Scanner(System.in);
		int n = in.nextInt();
		int k = in.nextInt();
		int f[] = new int[k];
		f[0] = 1;
		
		long sum = 0, count = 0;
		for (int i = 0; i < n; i++) {
			sum += in.nextLong();
			count += f[(int) (sum % k)];
			f[(int) (sum % k)]++;
		}
		
		System.out.println(count);
		in.close();
	}
}