蓝桥杯-k倍区间-java

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
请严格按要求输出，不要画蛇添足地打印类似：“请您输入…” 的多余内容。

注意：

main函数需要返回0;
只使用ANSI C/ANSI C++ 标准;
不要调用依赖于编译环境或操作系统的特殊函数。
所有依赖的函数必须明确地在源文件中 #include 不能通过工程设置而省略常用头文件。
提交程序时，注意选择所期望的语言类型和编译器类型。

思路：

这道题如果直接暴力双重循环会直接超时，所以只能借助其特殊性来进行判断，
1、区间[1，n]直接判断dp[n]%k=0，cnt++即可；
2、区间[i，j]则需要利用（dp[i]-dp[j]）%k=0判断，
所以我们借助一个数组sum[]来储存余数相等的数有多少个，当每次循环cnt+sum[dp[i]]即可。
参考：https://blog.csdn.net/shizhuba/article/details/105301239

代码如下附上：

/*
*注释部分为另外一种输入方式，在编写程序方面会更加省时
*/
package 蓝桥杯;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Scanner;

public class k倍区间 {

	public static void main(String[] args) throws IOException {
		// TODO Auto-generated method stub
		BufferedReader bReader = new BufferedReader(new InputStreamReader(System.in));
		Scanner sc = new Scanner(System.in);
		int num [] = new int [100010];
		int dp [] = new int [100010];
		int sum [] = new int [100010];
		//String s1[] = bReader.readLine().split("\\s+");
		
		int n = sc.nextInt();
		//Integer.parseInt(s1[0]);
		int k = sc.nextInt();
		//Integer.parseInt(s1[1]);
		
		long cnt = 0;
		for (int i = 1; i <= n; i++) {
			num[i] = sc.nextInt();
			//Integer.parseInt(bReader.readLine());
			dp[i] = dp[i-1] + num[i];
			dp[i] %= k;
			if(dp[i] == 0)
				cnt ++;
			cnt += sum[dp[i]]; 
			sum[dp[i]]++;
		}
		System.out.println(cnt);
	}

}