只有1例报错,求解答
查看原帖
只有1例报错,求解答
1261529
jiank_yanshisan楼主2024/3/4 16:29
#include <iostream>
#include <map>
using namespace std;

const int N = 1e5+5;
int A[N];
map<long long, int> cnt;

int main(){
	int n, K;
	int num= 0;
	long long sum = 0;
	cin >> n >> K;
	for(int i=1; i<=n; i++){
		cin >> A[i];
		sum += A[i];
		cnt[sum%K]++; 
	} 
	for(int i=0; i<=K; i++)
		num += (cnt[i]*(cnt[i]-1))/2;
	num += cnt[0];
	cout << num <<"\n";
	return 0;
}
//超时 ->前缀和 
//有一例错 
2024/3/4 16:29
加载中...