POJ 3046 Ant Counting

题目链接

题意

一共有T种蚂蚁编号从1 到 T，给出A个蚂蚁序列，问任取n个蚂蚁一共可以组几种集合，求S到B的种类和

AC

• dp
  状态转移方程为：
  
  K表示当前蚂蚁的数目

#include <iostream>
#include <algorithm>
#include <stdio.h>
#include <vector>
#include <map>
#include <bitset>
#include <string.h>
#include <cmath>
#include <algorithm>
#define N 100015
#define P pair<int,int>
#define ll long long
#define mk(a, b) make_pair(a, b)
#define mem(a, b) memset(a, b, sizeof(a))
using namespace std;
int dp[N];
int inf = 0x3f3f3f3f;
int mod = 1e6;
int main(){
#ifndef ONLINE_JUDGE
    freopen("in.txt", "r", stdin);
#endif
    int t, a, s, b;
    scanf("%d%d%d%d", &t, &a, &s, &b);
    // 记录不同蚂蚁的数目
    vector<int> v(t + 1);
    for (int i = 0; i < a; ++i) {
        int x;
        scanf("%d", &x);
        v[x]++;
    }
    // 记录到第i个蚂蚁，蚂蚁的总数
    vector <int> sum(t + 1);
    for (int i = 1; i <= t; ++i) {
        sum[i] = sum[i - 1] + v[i];
    }
    mem(dp, 0);
    // 将一个蚂蚁作为边界
    for (int i = 0; i <= v[1]; ++i) {
        dp[i] = 1;
    }
    // 计算剩余蚂蚁
    for (int k = 2; k <= t; ++k) {
        for (int i = b; i >= 1; --i) {
            // 当前蚂蚁数目超过总蚂蚁数目，跳过循环
            if (i > sum[k]) continue;
            ll tot = 0;
            for (int j = 0; j <= v[k]; ++j) {
                if (i - j < 0)  break;
                tot += dp[i - j];
                tot %= mod;
            }
            dp[i] = tot;
        }
    }
    ll ans = 0;
    for(int i = s; i <= b; ++i) {
        ans += dp[i];
        ans %= mod;
    }
    printf("%lld\n", ans);
    // cout <<ans <<endl;
    return 0;
}