有 WA 有 TLE
#include<bits/stdc++.h>
using namespace std;
struct node{
int color,number;
}p[1000010];
unsigned long long sum;
int n,m;
int main(){
cin>>n>>m;
for(int i=1;i<=n;i++)
cin>>p[i].number;
for(int i=1;i<=n;i++)
cin>>p[i].color;
for(int x=1;x<=n;x++)
for(int z=x+2;z<=n;z+=2){
if(p[x].color!=p[z].color)
continue;
sum+=(x+z)*(p[x].number+p[z].number)%10007;
}
cout<<sum;
return 0;
}