P2671 [NOIP2015 普及组] 求和(洛谷前缀和

P2671 [NOIP2015 普及组] 求和

//(x+z)*(num[x]+num[z])=
//(x1+x2)*(y1+y2)+(x1+x3)*(y1+y3)+(x2+x3)*(y2+y3) 
//=x1*(y1*(n-2)+y1+y2+...+yn) 
//+x2*(y2*(n-2)+y1+y2+...+yn)
//+xn*(yn*(n-2)+y1+y2+...+yn)
//即求出y1到yn的前缀和然后循环计算即可 

//题目要求格子颜色相同，可以把相同颜色的格子分在一个组
//且z-y=y-x，得x+z=2y，所以x和z为同奇或同偶
//所以相同格子中又可以奇数分一组偶数分一组
//操作时只针对每个组组内操作就行 
#include <bits/stdc++.h>
using namespace std;
const int maxn=1e5+5;
//数据范围 
const int mol=1e4+7;
//模数 
int n,m,ans;
//m为颜色种数，没啥用 
struct node{
    int id,num,col;
    //id为下表，没有用
    //num为格子上的数字
    //col为格子颜色 
}p[maxn];
int geshu[maxn][2];
//相同颜色格子且下标同奇或同偶格子的个数 
int tot[maxn][2];
//相同颜色格子且下标同奇或同偶格子的数字和 
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%d",&p[i].num);
        p[i].id=i;
    }
    for(int i=1;i<=n;i++){
        scanf("%d",&p[i].col);
        geshu[p[i].col][i%2]++;
        tot[p[i].col][i%2]=(tot[p[i].col][i%2]+p[i].num)%mol;
        //时刻模 
    }
    for(int i=1;i<=n;i++){
        ans=(ans+i*((geshu[p[i].col][i%2]-2)*p[i].num%mol+tot[p[i].col][i%2])%mol)%mol;
        
    }
    printf("%d",ans);
    return 0;
}