Ant Counting（多重背包计数 + 优化）

Ant Counting

题目

t种蚂蚁，总共n只，从中挑出i只，求从中挑出l只到r只的所有挑法。

思路

显然是dp ， 定义dp为从前i只挑出j只的方案集合，记录它的数量count。
状态转移是从 dp[i-1][j-k] ==> dp[i][j]

也即 ： dp[i][j] = SUM(dp[i-1][j-k]) (0<=k<=min(该种蚂蚁最大个数，j)

---

这是一个O（nmm）的算法（虽然这题数据挺水，也可以过）

对于10^5数据量 ，需要优化到O(nm) 。

回到dp的状态转移方程

在末尾加上一个 dp[i-1][j-1-a[i] 加上一个dp[i-1][j-1-a[i]] 可以得到在j>a[i]时的优化后的dp方程

• dp[i][j] = dp[i][j-1] + dp[i-1][j] - dp[i-1][j-1-a[i]

用相同的技巧，同样可以得到j<=a[i]的情况。

• dp[i][j] = dp[i]][j-1] + dp[i-1][j]

---

现在不用对k的枚举，成功得到O(nm)的算法。

对于空间优化，使用滚动数组进行优化即可,当然也可以优化为一维，但没有太大必要。

##AC代码

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <string>
#include <cstring>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <fstream>
#define inf 0x3f3f3f3f
#define ll long long
#define ull unsigned long long
#define endl '\n'
//#define int long long
using namespace std;
typedef pair<int, int> PII;
const int N = 10 + 1e5, mod = 1e6;
// ll dp[1010][N]; // dp[i][j] = sum dp[i-1][k]
int dp[2][N];
int a[N];
void solve()
{
    int m, n, l, r;
    cin >> m >> n >> l >> r;
    for (int i = 1; i <= n; i++)
    {
        int t;
        cin >> t;
        a[t]++;
    }
    dp[0][0] = dp[1][0] = 1;// 初始化

    int pre = 1, now = 0;
    for (int i = 1; swap(now, pre), i <= m; i++)
        for (int j = 1; j <= r; j++)
            if (j - 1 >= a[i]) // j>a[i]
                dp[now][j] = (dp[now][j - 1] + dp[pre][j] - dp[pre][j - a[i] - 1] + mod) % mod;
            else // j <= a[i]
                dp[now][j] = dp[now][j - 1] + dp[pre][j], dp[now][j] %= mod;
            /* 朴素做法
            for(int k=0;k<=min(a[i],j);k++)
                dp[i][j] += dp[i-1][j-k],dp[i][j]%=mod; */

    ll ans = 0;
    for (int i = l; i <= r; i++)
        ans = (ans + dp[pre][i]) % mod;
    cout << ans << endl;
}
signed main()
{
    ios::sync_with_stdio();
    cin.tie();
    cout.tie();

    solve();

    return 0;
}