洛谷P2822组合数问题（组合数学 + 前缀和）

2020.6.16
这是noip某年提高组的一道题。非常不幸，这道题只拿了90pts, t了两个点，后来加了快读和快写之后95pts，可能是因为每次询问大概是O(nm)的。所以2000就会t了。

这道题思路也非常清晰，2k的数据量数组开得下，可以用组合数递推公式来获得所有组合数模k的值。组合数递推公式如下：

组合数递推方程 ：f[i][j] = f[i - 1][j] + f[i - 1][j - 1]；

然后95pts的做法是用直接挨个查，遇到模k==0的情况对答案贡献就加一，最后输出。但这样有个问题，就是组合数是二维的，每次要跑n * min（n,m）次，很容易超时。以上是本人一开始自己的做法，然而没ac也没卵用。

然后就看到了神仙的前缀和方法，代码给出如下，大家细品一下区别，反正这个前缀和方法真的太巧妙了
95分代码：

#include <bits/stdc++.h>
using namespace std;
#define limit (2000 + 5)//防止溢出
#define INF 0x3f3f3f3f
#define inf 0x3f3f3f3f3f
#define lowbit(i) i&(-i)//一步两步
#define EPS 1e-6
#define FASTIO  ios::sync_with_stdio(false);cin.tie(0);
#define ff(a) printf("%d\n",a );
#define pi(a,b) pair<a,b>
#define rep(i, a, b) for(int i = a; i <= b ; ++i)
#define per(i, a, b) for(int i = b ; i >= a ; --i)
#define mint(a,b,c) min(min(a,b), c)
#define MOD 998244353
#define FOPEN freopen("C:\\Users\\tiany\\CLionProjects\\acm_01\\data.txt", "rt", stdin)
typedef long long ll;
typedef unsigned long long ull;
ll read(){
    ll sign = 1, x = 0;char s = getchar();
    while(s > '9' || s < '0' ){if(s == '-')sign = -1;s = getchar();}
    while(s >= '0' && s <= '9'){x = x * 10 + s - '0';s = getchar();}
    return x * sign;
}//快读
void write(ll x){
    if(x < 0) putchar('-'),x = -x;
    if(x / 10) write(x / 10);
    putchar(x % 10 + '0');
}

int n,m,k;
ll f[limit][limit];//i代表n，j代表m
//组合数递推方程 f[i][j] = f[i - 1][j] + f[i - 1][j - 1]；
void get_combinatoric(){
    rep(i ,00, 2000){
        f[i][i] = f[i][0] = 1;//不选就只有一种情况
    }//初始状态
    rep(i , 1, 2000){
        rep(j , 1, i){
            f[i][j] = (f[i - 1][j] + f[i - 1][j - 1]) % k;
        }
    }
}
int main() {
#ifdef LOCAL
    FOPEN;
#endif
    int t = read();
    k = read();
    get_combinatoric();//取组合数
    while (t--){
        n = read(), m = read();
        int ans = 0;
        rep(i ,0, n){
            rep(j , 0, min(i, m)){
                if(!f[i][j]) ++ans;
            }
        }
        write(ans),putchar('\n');
    }
    return 0;
}

AC代码：

#include <bits/stdc++.h>
using namespace std;
#define limit (2000 + 5)//防止溢出
#define INF 0x3f3f3f3f
#define inf 0x3f3f3f3f3f
#define lowbit(i) i&(-i)//一步两步
#define EPS 1e-6
#define FASTIO  ios::sync_with_stdio(false);cin.tie(0);
#define ff(a) printf("%d\n",a );
#define pi(a,b) pair<a,b>
#define rep(i, a, b) for(int i = a; i <= b ; ++i)
#define per(i, a, b) for(int i = b ; i >= a ; --i)
#define mint(a,b,c) min(min(a,b), c)
#define MOD 998244353
#define FOPEN freopen("C:\\Users\\tiany\\CLionProjects\\acm_01\\data.txt", "rt", stdin)
typedef long long ll;
typedef unsigned long long ull;
ll read(){
    ll sign = 1, x = 0;char s = getchar();
    while(s > '9' || s < '0' ){if(s == '-')sign = -1;s = getchar();}
    while(s >= '0' && s <= '9'){x = x * 10 + s - '0';s = getchar();}
    return x * sign;
}//快读
void write(ll x){
    if(x < 0) putchar('-'),x = -x;
    if(x / 10) write(x / 10);
    putchar(x % 10 + '0');
}

int n,m,k;
ll f[limit][limit];//i代表n，j代表m
ll ans[limit][limit];
//组合数递推方程 f[i][j] = f[i - 1][j] + f[i - 1][j - 1]；
void get_combinatoric(){
    rep(i ,00, 2000){
        f[i][i] = f[i][0] = 1;//不选就只有一种情况
    }//初始状态
    rep(i , 1, 2000){
        rep(j , 1, i){
            f[i][j] = (f[i - 1][j] + f[i - 1][j - 1]) % k;
            ans[i][j] = ans[i-1][j] + ans[i][j-1] - ans[i-1][j-1];
            if(!f[i][j])ans[i][j]++;
        }
        ans[i][i+1] = ans[i][i];
    }
}
int main() {
#ifdef LOCAL
    FOPEN;
#endif
    int t = read();
    k = read();
    get_combinatoric();//取组合数
    while (t--){
        n = read(), m = read();
        ff(ans[n][m > n ? n : m])
    }
    return 0;
}