题解 点赞

#include <bits/stdc++.h>
using namespace std;
const int maxN = 100010,moD = 10007;
long long ans=0,n,m,a[maxN],c[maxN],sr[maxN][2],sa[maxN][2],s[maxN][2],sq[maxN][2];
int main()
{
	cin>>n>>m;
	for (int i=1; i<=n; i++) cin>>a[i];
	for (int i=1; i<=n; i++) cin>>c[i];
	for (int i=1; i<=n; i++)
	{
		int p=i&1,ci=c[i];
		s[ci][p]++;
		if(s[ci][p]>1) ans+=(a[i]*sr[ci][p])%moD+(i*sa[ci][p])%moD+sq[ci][p]+((s[ci][p]-1)*i*a[i]%moD);
		ans%=moD;
		sr[ci][p]+=i;
		sa[ci][p]+=a[i];
		sq[ci][p]+=i*a[i];
	}
	ans%=moD;
	cout<<ans;
	return 0;
}