40分球球助,大佬们看看能优化吗
查看原帖
40分球球助,大佬们看看能优化吗
677150
ThEskY_B1ackeR楼主2022/10/5 12:38
#include<bits/stdc++.h>
using namespace std;

const int MAXN = 100005;

long long sum=0;

struct st{
	
	int number,color;	

}a[MAXN];

int n,m;

int main(){
	
	//freopen("sum.in","r",stdin);
	//freopen("sum.out","w",stdout);
	
	cin>>n>>m;
	
	if(n-m<1){
		
		cout<<0;
		
	//	fclose(stdin);
	//	fclose(stdout);	
		return 0;
	}
	
	for(int i=1;i<=n;i++) cin>>a[i].number;
	
	for(int i=1;i<=n;i++) cin>>a[i].color; 
	
	for(int i=1;i<=n;i++) for(int j=1;i+j*2<=n;j++){
		
		if(a[i].color==a[i+j+j].color){
		
			sum+=(i+i+j+j)%10007*((a[i].number+a[i+j+j].number)%10007);
			
			sum%=10007;
				
		}
		
	}
	
	cout<<sum;
	
//	fclose(stdin);
//	fclose(stdout);
	return 0;
}
2022/10/5 12:38
加载中...