#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就可以了
(能改不超时必关)