记录编号 335557 评测结果 AAAAAAAAAA
题目名称 [NOIP 2015PJ]求和 最终得分 100
用户昵称 GravatarMagic_Sheep 是否通过 通过
代码语言 C++ 运行时间 2.568 s
提交时间 2016-11-02 15:05:13 内存使用 3.40 MiB
显示代码纯文本
#include<cstdio>
#include<iostream>
#include<algorithm> 
using namespace std;
struct node
{
	long long id,col,val;
};
node t[101010];
int n,m,be,end;
long long color[101010];
long long ans;
bool cmp(const node &a,const node &b)
{
	if(a.col==b.col) return a.id<b.id;
	return a.col<b.col;
}
int main()
{
	freopen("2015sum.in","r",stdin);
	freopen("2015sum.out","w",stdout);
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		scanf("%lld",&t[i].val);
		t[i].id=i;
	}
	for(int i=1;i<=n;i++) scanf("%lld",&t[i].col),color[t[i].col]++;
	sort(t+1,t+1+n,cmp);
	for(int i=1;i<=m;i++)
	{
		be++;end=be+color[i]-1;
		for(int j=be;j<=end;j++)
		{
			for(int l=j+1;l<=end;l++)
			{
				if((t[j].id-t[l].id)&1) continue;
				ans+=(t[j].id+t[l].id)*(t[j].val+t[l].val);
				ans%=10007;
			}
		}
		be=end;
	}
	printf("%lld",ans);
	return 0;
}