显示代码纯文本
#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;
}