noipnoip201503求和

Posted 2020-08-12 Kaori

3. 求和

        难度级别：B； 运行时间限制：1000ms； 运行空间限制：51200KB； 代码长度限制：2000000B

题目描述

  一条狭长的纸带被均匀划分出了n个格子，格子编号从1到n。每个格子上都染了一种颜色colori用[1，m]当中的一个整数表示），并且写了一个数字numberi。

5		5	3	2	2	2
编号	1	2	3	4	5	6

 

定义一种特殊的三元组：(x, y, z)，其中x，y，z都代表纸带上格子的编号，这里的三元组要求满足以下两个条件：

组要求满足以下两个条件：

    1.  xyz是整数,x<y<z，y-x=z-y

    2.  colorx=colorz

    满足上述条件的三元组的分数规定为(x+z)*(numberx+numberz)。整个纸带的分数规定为所有满足条件的三元组的分数的和。这个分数可能会很大，你只要输出整个纸带的分数除以10,007所得的余数即可。

 

【输入输出样例1】

 

sum.in						sum.out
6	2					82
5	5	3	2	2	2
2	2	1	1	2	1

 

 

【输入输出样例1 说明】 纸带如题目描述中的图所示。

所有满足条件的三元组为：(1,3,5),(4,5,6)。

所以纸带的分数为(1+5)* (5+2)+ (4+6) *(2+2)=42+40=82。

 

【输入输出样例2】

 

sum.in	sum.out
15 4 5 10 8 2 2 2 9 9 7 7 5 6 4 2 4 2 2 3 3 4 3 3 2 4 4 4 4 1 1 1	1388

 

【数据说明】

对于第 1 组至第 2 组数据， 1 ≤ n ≤ 100, 1 ≤ m ≤ 5；

对于第 3 组至第 4 组数据， 1 ≤ n ≤ 3000, 1 ≤ m ≤ 100；

对于第 5 组至第 6 组数据， 1 ≤ n ≤ 100000, 1 ≤ m ≤ 100000，且不存在出现次数超过 20 的颜色；

对 于 全 部 10 组 数 据 ， 1 ≤ n ≤ 100000, 1 ≤ m ≤ 100000, 1 ≤ colori ≤ m，1≤ numberi ≤100000

 

输入

第一行是用一个空格隔开的两个正整数n和m，n表纸带上格子的个数，m表纸带上颜色的种类数。

第二行有n用空格隔开的正整数，第i数字number表纸带上编号为i格子上面写的数字。

第三行有n用空格隔开的正整数，第i数字color表纸带上编号为i格子染的颜色。

 

输出

共一行，一个整数，表示所求的纸带分数除以10,007 所得的余数。

 

样例输入

6 2 5 5 3 2 2 2 2 2 1 1 2 1

样例输出

82

 

 

这就尴尬了→_→

/*
对于一个格子i，设之前出现的与之同奇偶p同颜色ci的格子数目为s[ci][p]、格子序号和为sr[ci][p]、格子num值之和为sa[ci][p]、格子序号与a
之积之和为sq[ci][p]，则如下累计ans即可:
ans+=(a[i]*sr[ci][p])%mod+(i*sa[ci][p])%mod+sq[ci][p]+((s[ci][p]-1)*i*a[i]%mod);
*/
#include<iostream>
#include<cstdio>
#define LL long long
using namespace std;
using namespace std;
const int maxn=100010;
const int mod=10007; 
LL n,m,a[maxn],c[maxn],sr[maxn][2],sa[maxn][2],s[maxn][2],sq[maxn][2],ans;
//MOD用int出过错 索性改为long long

int main(){
	int i;
    scanf("%lld%lld",&n,&m);
    for(i=1;i<=n;i++) scanf("%lld",&a[i]);
    for(i=1;i<=n;i++) scanf("%lld",&c[i]);
    for(i=1;i<=n;i++){
        int p=i&1,ci=c[i];//i为奇数p为1，i为偶数p为0 
        s[ci][p]++;
        if(s[ci][p]>1) ans+=(a[i]*sr[ci][p])%mod+(i*sa[ci][p])%mod+sq[ci][p]+((s[ci][p]-1)*i*a[i]%mod);
        ans%=mod;
        sr[ci][p]+=i;
        sa[ci][p]+=a[i];
        sq[ci][p]+=i*a[i];
    }
    ans%=mod; 
    printf("%lld\n",ans);
    return 0;
}

额，再不懂就评论吧，网上到处都是题解