牛逼的思路(90分求助QAQ)
查看原帖
牛逼的思路(90分求助QAQ)
965646
孟令然2013楼主2025/6/14 21:24
#include<bits/stdc++.h>
using namespace std;
long long n,m,num[100005],c[100005],ans=0;
vector<int> q[100005];
int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	cin>>n>>m;
	for(int i=1;i<=n;i++)
	{
		cin>>num[i];
	}
	for(int i=1;i<=n;i++)
	{
		cin>>c[i];
		q[c[i]].push_back(i);
	}
	for(int i=1;i<=m;i++)
	{
		for(int j=0;j<q[i].size();j++)
		{
			for(int k=1;k<q[i].size()-j;k++)
			{
				if((q[i][j+k]-q[i][j]+1)%2==1)
				{
					ans=(ans+(q[i][j]+q[i][j+k])*(num[q[i][j]]+num[q[i][j+k]])%10007)%10007;
				}
			}
		}
	}
	cout<<ans<<"\n";
	return 0;
}

用vector存相同颜色,剩下只需判断x和z就可以了 (能改不超时必关

2025/6/14 21:24
加载中...