蓝桥杯真题 k倍区间 c++代码实现 给定一个长度为 N 的数列，A1, A2, ··· AN，如果其中一段连续的子序列 Ai,Ai+1,⋯A j ​ ( i \leq ji≤j ) 之

题目描述

给定一个长度为 N 的数列，A₁, A₂, ···A_N，如果其中一段连续的子序列 A_i,A_i+1, ···A_j( i≤j ) 之和是 K 的倍数，我们就称这个区间 [i, j] 是 K 倍区间。

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

输入描述

第一行包含两个整数 N 和 K( 1≤N,K≤10⁵)。

以下 N 行每行包含一个整数 A_i( 1 ≤A_i≤10⁵)

输出描述

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

输入输出样例

示例

输入
5 2 1 2 3 4 5

输出
6

运行限制

最大运行时间：2s
最大运行内存: 256M

代码

#include <iostream>
using namespace std;
int main()
{
	int x, y;
	long long num;//k倍区间个数可能很多，所以使用long long数据类型
	cin >> x >> y;
	long long arr[100010] = { 0 };//保存序列的每个数的余数
	long long ans[100010] = { 0 };//保存相同余数的数字的个数
	num = 0;
	for (int i = 1; i <= x; i++)
	{
		cin >> arr[i];
		arr[i] += arr[i - 1];
		arr[i] = arr[i] % y;
		num += ans[arr[i]];
		ans[arr[i]]++;
	}
	cout << num + ans[0];//由于num没有加上[0,x]区间上的个数，所以要加上ans[0]
	return 0;
}

思路

由于要找的序列一定是连续的，所以不妨将序列的前i项和记为数组的第i个元素，从而某区间[m,n]的区间和即为arr[n]-arr[m]，若此区间为k倍区间，则(arr[n]-arr[m])%k=0成立，分析该式子得arr[n]%k=arr[m]%k。所以只需要将序列的前i项的和对k取余的余数存起来，计算相同的余数的和有多少个即可算出k倍区间的个数。