Codeforces Round247 C k-tree DP

题目链接点这儿。

本来打算昨晚参赛的,但是突然想起No Game No Life 0点更新第七话,然后我就去守更新去了www白真的太萌了www

然后C就早上起来做了一下,还是挺水的。

给你重量为1~k的物品(个数无限),放入一个容量为n的背包,保证至少有一件重量>=d的物品(而且物品存放是有序的)。

所以dp水一下就过去了,dp[0][i]表示容量为i的背包都由重量<d的物品存放的方法数,dp[0][i]则是必有重量>=d的方法数。

然后详细的转移过程就看代码里就好了,挺好想的。

逛cf的时候,发现有人用容斥做,dp[0][i]含义不变,dp[1][i]表示随意放物品填充满背包的方法数。这时答案就是dp[1][n] - dp[0][n]。复杂度差不多,都是o(kn),官方题解说可以过程中更新两个前缀和数组来达到o(n),这里明显是不用了。

代码见下(1e9明明是不会爆int的。。。但是我这个确有影响了。。。估计是加法的过程中爆掉了。。。)

#include <bits/stdc++.h>


#define up(i, lower, upper) for(int i = lower; i < upper; i++)
#define down(i, lower, upper) for(int i = upper-1; i >= lower; i--)


using namespace std;
typedef pair<int, int> pii;
typedef pair<double, double> pdd;
typedef vector<int> vi;
typedef vector<pii> vpii;
typedef __int64 ll;
typedef unsigned __int64 ull;


const double pi = acos(-1);
const double eps = 1.0e-9;


template<class T>


inline bool read(T &n){
    T x = 0, tmp = 1; char c = getchar();
    while((c < '0' || c > '9') && c != '-' && c != EOF) c = getchar();
    if(c == EOF) return false;
    if(c == '-') c = getchar(), tmp = -1;
    while(c >= '0' && c <= '9') x *= 10, x += (c - '0'),c = getchar();
    n = x*tmp;
    return true;
}


template <class T>
inline void write(T n) {
    if(n < 0) {
        putchar('-');
        n = -n;
    }
    int len = 0,data[20];
    while(n) {
        data[len++] = n%10;
        n /= 10;
    }
    if(!len) data[len++] = 0;
    while(len--) putchar(data[len]+48);
}

//---------------------------------------------------------
const int mod = 1e9+7;
int main() {
    ll dp[2][102] = { 0 };
    int n, k, d;
    read(n), read(k), read(d);
    if(n < d) { puts("0"); exit(0); }
    dp[0][0] = 1;
    up(i, 1, n+1) {
        up(j, 1, (i<k?i:k)+1) {
            if(j < d) {
                dp[0][i] = (dp[0][i] + dp[0][i-j])%mod;
                dp[1][i] = (dp[1][i] + dp[1][i-j])%mod;
            }
            else dp[1][i] = (dp[1][i] + dp[0][i-j] +dp[1][i-j])%mod;
        }
    }
    write(dp[1][n]);
}


你可能感兴趣的:(算法,dp,ACM,codeforces)