P2671 [NOIP2015 普及组] 求和

Ivan（互关）

于 2024-10-07 19:48:55 发布

阅读量42

点赞数 2

文章标签：算法 c++ 图论

本文链接：https://blog.csdn.net/2301_80484635/article/details/142745202

版权

#include <bits/stdc++.h>
using namespace std;
#define int long long //要开long long
const int N=1e5+5;
struct node {
	int num,col,idx;//idx:原来的下标
}t[N];
bool cmp(node a,node b){
	return a.col<b.col;//相同的颜色放到一起方便处理
}
int x[N],y[N],ans;
int solve(int l,int r){//[l,r]是相同的颜色
	int n=0;
	int sum=0,ans=0;
	for(int i=l;i<=r;i++) if(t[i].idx&1)x[++n]=t[i].num,y[n]=t[i].idx;//如果下标是奇数那么把他存到x,y里当成一组
	for(int i=1;i<=n;i++) sum+=y[i];//预处理sum
	for(int i=1;i<=n;i++) ans=(ans+x[i]*(sum+y[i]*(n-2)))%10007;//这里就要取模小心到时候答案不对
	n=0,sum=0;
	for(int i=l;i<=r;i++) if((t[i].idx&1)==0)x[++n]=t[i].num,y[n]=t[i].idx;//如果下标是偶数那么把他存到x,y里当成一组
	for(int i=1;i<=n;i++) sum+=y[i];
	for(int i=1;i<=n;i++) ans=(ans+x[i]*(sum+y[i]*(n-2)))%10007;
	return ans;
}
signed main()
{
	int n,m;
	cin>>n>>m;
	for(int i=1;i<=n;i++) cin>>t[i].num,t[i].idx=i;
	for(int i=1;i<=n;i++) cin>>t[i].col;
	sort(t+1,t+n+1,cmp);
	int las=1;
	for(int i=2;i<=n+1;i++)
		if(t[i].col!=t[i-1].col){
			ans=(ans+solve(las,i-1))%10007;
			las=i;
		}
	cout<<ans%10007<<endl;
	return 0;
}