【NOIP2015普及组】求和

题目

思路

有题目可得，num同奇偶性时，才满足题目条件。
所以先求出同奇偶性的，在分别求值就行了
注：s数组每次更新

代码

#include
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;
}