TopCoder SRM 572 Div2 1000 DistinctRemainders

最新推荐文章于 2018-09-27 09:14:09 发布

ymzqwq

最新推荐文章于 2018-09-27 09:14:09 发布

阅读量266

点赞数

分类专栏： TopCoder TopCoder部分题解文章标签： TopCoder

本文链接：https://blog.csdn.net/ymzqwq/article/details/82084551

版权

TopCoder部分题解同时被 2 个专栏收录

54 篇文章 4 订阅

订阅专栏

TopCoder

53 篇文章 0 订阅

订阅专栏

题目说了 $K$ 个数模 $m$ 两两互不相同，有点类似0/1背包，每个剩余类只能选一个，可以用 $f[i][j]$ 表示 $i$ 个在 $[0,M-1]$ 的数和为 $j$ 的方案数，DP一波。
对于 $f[i][j]$ ，若 $j \equiv N (mod\ M)$ 且 $j<=N$ ，则对答案的贡献为

f [i] [j] * C ((N - j) / M + i - 1, i - 1) * i!

$f[i][j]*C((N-j)/M+i-1,i-1)*i!$

为什么呢？考虑给每个数加上 $m$ 的整数倍之后模 $m$ 不会发生变化，所以肯定还是符合模 $m$ 两两互不相同，然而和却神不知鬼不觉地增大了。所以要使总和为 $N$ 就相当于把 $(N-j)/M$ 个 $M$ 分给 $i$ 个数，套用经典的小球模型就是把 $(N-j)/M$ 个小球分给 $i$ 个盒子，可以为空，这个方案数是 $C((N-j)/M+i-1,i-1)$ ，题目里是有序的，所以还要乘个阶乘。

为方便计算，上式进一步简化就是 $f[i][j]*A((N-j)/M+i-1,i-1)*i$

代码中的变量名和题目不太一样

//tc is healthy, just do it
#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N=2501;
const int M=51;
const int p=1e9+7;
ll f[M][N];

class DistinctRemainders {
public:
    int howMany( long long N, int M );
};

void Add(ll &x,ll y){
    x+=y;
    while(x>=p) x-=p;
}

ll A(ll n,int m){
    if (n<m) return 0;
    ll ans=1;
    for(int i=1;i<=m;i++)
     ans=ans*((n-i+1)%p)%p;
    return ans;
}

int DistinctRemainders::howMany(long long s, int m) {
    f[0][0]=1;
    int sum=0;
    for(int i=0;i<m;i++){
        sum+=i;
        for(int j=i+1;j;j--)  //这里要倒着for，原理类似于0/1背包的倒着for
         for(int k=i;k<=sum;k++)
          Add(f[j][k],f[j-1][k-i]);
    }
    ll y=s%m;
    ll ans=0;
    for(int i=1;i<=m;i++)
     for(ll j=y;j<=s&&j<=sum;j+=m)
      Add(ans,f[i][j]*A((s-j)/m+i-1,i-1)%p*i%p);
    return (int)ans;
}