题目
思路
有题目可得,num同奇偶性时,才满足题目条件。
所以先求出同奇偶性的,在分别求值就行了
注:s数组每次更新
代码
#include<cstdio>
using namespace std;
#define N 100001
#define mod 10007
int n,m,sum;
int num[N],c[N],s[N];
void cal(){
for(int len=1;len<s[0];len++)
for(int i=1;i+len<=s[0];i++){
sum+=((s[i]%mod+s[i+len]%mod)*(num[s[i]]%mod+num[s[i+len]]%mod));
sum%=mod;
}
}
int main()
{
// freopen("sum.in","r",stdin);
// freopen("sum.out","w",stdout);
scanf("%d %d",&n,&m);
if(n==m) {printf("0");return 0;}
for(int i=1;i<=n;i++)
scanf("%d",&num[i]);
for(int i=1;i<=n;i++)
scanf("%d",&c[i]);
for(int i=1;i<=n;i++){
if(c[i]){
s[0]=0;
s[++s[0]]=i;
for(int k=i+2;k<=n;k+=2)
if(c[i]==c[k]){
s[++s[0]]=k;
c[k]=0;
}
c[i]=0;
cal();
}
}
printf("%d",sum);
return 0;
}