0分蒟蒻求助
查看原帖
0分蒟蒻求助
1041765
AKkkk_TIX楼主2024/10/20 12:53
#include <bits/stdc++.h>
using namespace std;
int main(){
	freopen("sum.in", "r", stdin);
	freopen("sum.out", "r", stdout);
	
	int n, m;
	scanf("%d%d", &n, &m);
	
	int number[n + 1], color[n + 1];
	long long num = 0;
	for(int i = 1;i <= n;i++)
		cin >> number[i];
	for(int i = 1;i <= n;i++)
		cin >> color[i];
	
	for(int i = 1;2 * i + 1 <= n;i++){
		for(int j = 1;j <= n;j++){
			int x = j, y = j + i, z = j + 2 * i;
			if(x == n || y == n || z > n)
				break;
			
			if(color[x] == color[z])
				num += (x + z) * (number[x] + number[z])
		}
	}
	num = num % 10007;
	printf("%lld", num);
	
    return 0; 
}

测评记录

2024/10/20 12:53
加载中...