NOIP2015求和

[NOIP2015 普及组] 求和

题目背景

NOIP2015 普及组 T3

题目描述

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

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

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

2. $colorx=colorz$

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

输入格式

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

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

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

输出格式

一个整数，表示所求的纸带分数除以$10007$所得的余数。

样例 #1

样例输入 #1

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

样例输出 #1

82

样例 #2

样例输入 #2

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

样例输出 #2

1388

提示

【输入输出样例 1 说明】

纸带如题目描述中的图所示。

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

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

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

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

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

对 于 全 部 $10$ 组 数 据 ， $1\le n\le 100000,1\le m\le 100000,1\le colo{r}_i\le m,1\le numbe{r}_i\le 100000$

分析：

这哪是前缀和呀这题，这明明是数学，纯纯的数学公式推导SOS
根据这两个条件：

1. $xyz$是整数,$x<y<z,y-x=z-y$
2. $colorx=colorz$
   可以很清楚的看出来实际上$y$在这里根本没起作用，只要$x+z=2y,colorx=colorz$就可以了，也就是只要$x,z$的奇偶相同，颜色相同就OK。 所以，很自然的就想到可以把奇偶和颜色分开，分别计算。
   在此基础上再做优化就是数学推导，三元组规定为$(x+z)\times (numbe{r}_x+numbe{r}_z)$，把式子展开来看就是:
   $x\times numbe{r}_x+x\times numbe{r}_z+z\times numbe{r}_x+z\times numbe{r}_z$
   在对颜色和奇偶进行分类后，对我们的每一个集合内的数来分析就有：
   $(i_1+i_2)\times (n_{i_1}+n_{i_2})+(i_1+i_3)\times (n_{i_1}+n_{i_3})+......+(i_1+i_k)\times (n_{i_1}+n_{i_k})$+$(i_2+i_3)\times (n_{i_2}+n_{i_3})+(i_2+i_4)\times (n_{i_2}+n_{i_4})+......+(i_2+i_k)\times (n_{i_2}+n_{i_k})$+…+$(i_k+i_{k-1})\times (n_{i_k}+n_{i_{k-1}})$
   然后，把每一个都乘开来看就是这样的：
   $[(i_1\times n_{i_1}+i_1\times n_{i_2}+i_2\times n_{i_1}+i_2\times n_{i_2})+(i_1\times n_{i_1}+i_1\times n_{i_3}+i_3\times n_{i_1}+i_3\times n_{i_3})+......+(i_1\times n_{i_1}+i_1\times n_{i_k}+i_k\times n_{i_1}+i_k\times n_{i_k})]$+$[(i_2\times n_{i_2}+i_2\times n_{i_3}+i_3\times n_{i_2}+i_3\times n_{i_3})+(i_2\times n_{i_2}+i_2\times n_{i_4}+i_4\times n_{i_2}+i_4\times n_{i_4})+......+(i_2\times n_{i_2}+i_2\times n_{i_k}+i_k\times n_{i_2}+i_k\times n_{i_k})]$+…+$[(i_k\times n_{i_k}+i_k\times n_{i_{k-1}}+i_{k-1}\times n_{i_k}+i_{k-1}\times n_{i_{k-1}})]$
   然后先只看第一个中括号里面的数，就会发现它是：
   $（k-1）\times (i_1\times n_{i_1})+i_1\times {\sum }_{t=2}^k{n}_{i_t}+$一些其他的内容，动动你聪明的脑瓜和灵巧的小手自己展吧
   然后这个$(k-1）\times (i_1\times n_{i_1})+i_1\times {\sum }_{t=2}^k{n}_{i_t}$就可以进一步的写成$（k-2）\times (i_1\times n_{i_1})+i_1\times {\sum }_{t=1}^k{n}_{i_t}$这里我们终于看到了前缀和，感天动地。把后面的内容展开的话就会发现他们最后可以归成一个形式类似的式子也就是
   $$（k-2）\times (i_r\times n_{i_r})+i_r\times \sum _{t=1}^k{n}_{i_t}$$
   所以我们就可以先处理出前缀和和相同颜色，相同奇偶的数的个数，这样只要O(n)的时间复杂度就可以处理出来所有的结果了。
   代码：

#include <iostream>
using namespace std;
const int N = 100010;
const int mod = 10007;
int n,m;
int idx[N],color[N];
int cnt[N][2];
int sum[N][2];
int main()
{
    scanf("%d%d",&n,&m);
    for(int i = 1;i <= n;i++)
    {
        scanf("%d",&idx[i]);
        idx[i] = idx[i]%mod;
    }
    for(int i = 1;i <= n;i++)
    {
        int c;
        scanf("%d",&c);//输入对应的颜色
        color[i] = c;
        cnt[c][i%2]++;//该颜色对应的奇数或偶数个数加一
        sum[c][i%2] = (sum[c][i%2]+idx[i])%mod;//求前缀和
    }
    long long ans = 0;
    for(int i = 1;i <= n;i++)
    {
        int c = color[i];
        ans += i*((sum[c][i%2]+((cnt[c][i%2]-2)*idx[i]%mod+mod))%mod)%mod;//注意取模
        ans %= mod;
    }
    printf("%lld",ans);
    return 0;
}