题目描述
一条狭长的纸带被均匀划分出了n个格子,格子编号从1到n。每个格子上都染了一种颜色color_i用[1,m]当中的一个整数表示),并且写了一个数字number_i。
定义一种特殊的三元组:(x,y,z),其中x,y,z都代表纸带上格子的编号,这里的三元
组要求满足以下两个条件:
1.xyz是整数,x<y<z,y-x=z-y
2.colorx=colorz
满足上述条件的三元组的分数规定为(x+z)*(number_x+number_z。整个纸带的分数
规定为所有满足条件的三元组的分数的和。这个分数可能会很大,你只要输出整个纸带的分数除以10,007所得的余数即可。
-------------------------------------------------------------------------------------------------------------------
需要一点数学推导,对于一种颜色,就是sigma奇*sigma num奇+sigma偶*sigma num偶+奇*num*(cnt奇-2)+偶*num*(cnt偶-2)
预处理sum和cnt数组,然后扫一遍统计答案行了
WARN:一定多%%%%%%%%%%%%
#include#include using namespace std;const int N=1e5+5,MOD=10007;inline int read(){ char c=getchar(); int x=0,f=1; while(c<'0'||c>'9'){ if(c=='-')f=-1;c=getchar();} while(c>='0'&&c<='9'){x=x*10+c-'0';c=getchar();} return x*f;}int n,m,ans=0;int col[N],num[N],sum[2][N],cnt[2][N];int main(int argc, const char * argv[]) { n=read();m=read(); for(int i=1;i<=n;i++) num[i]=read()%MOD; for(int i=1;i<=n;i++){ col[i]=read(); sum[i%2][col[i]]=(sum[i%2][col[i]]+num[i])%MOD; cnt[i%2][col[i]]++; } for(int i=1;i<=n;i++){// ans=(ans+(i%MOD*sum[i%2][col[i]])%MOD+(i%MOD*num[i]*(cnt[i%2][col[i]]-2)+MOD)%MOD)%MOD; ans=(ans+i%10007*((sum[i%2][col[i]]+(cnt[i%2][col[i]]-2)%10007*num[i]+10007)%10007))%10007; } cout<