Vijos 1090 连续数之和（鸽巢定理+组合数）

题目链接

matrix67大神出的题，做出来好激动啊。。。和前几天做的那个用鸽巢定理差不多，只不过是加上一个组合而已。开始把50000的组合数都给预处理出来了，果断TLE了，改了改处理到n，然后0ms秒过了。。。

#include <cstdio>
#include <cstring>
using namespace std;
#define MOD 1234567
int c[500001][3],p[500001],sum[500001],o[100001];
int main()
{
    int i,j,n,m,ans;
    scanf("%d%d",&n,&m);
    for(i = 0;i <= n;i ++)
    c[i][0] = 1;
    for(i = 1;i <= n;i ++)
    {
        for(j = 1;j <= 2;j ++)
        c[i][j] = (c[i-1][j-1]+c[i-1][j])%MOD;
    }
    for(i = 1;i <= n;i ++)
    {
        scanf("%d",&p[i]);
        sum[i] = (sum[i-1] + p[i])%m;
        o[sum[i]]++;
    }
    for(i = 0,ans = 0;i <= m-1;i ++)
    {
        ans = (ans + c[o[i]][2])%MOD;
    }
    ans = (ans + o[0])%MOD;
    printf("%d\n",ans);
    return 0;
}