vijosP1059 积木城堡

vijosP1059 积木城堡

链接：https://vijos.org/p/1059

【思路】

   01背包。

   刚开始想麻烦了，想的是二分答案然后01背包判断是否可行，但是首先答案不满足单调性所以不能二分(这点以后做题之前一定要想清楚)，其次如果从大到小枚举依次判定的话会TLE。

   不得不说自己真是笨。

   其实可以对每一组积木用一次01背包，用一个cnt记录满足该高度的城堡数目。然后从大到小检查如果为n输出即可。时间上是O(n^3)。

【代码】

#include<iostream>
#include<cstring>
#define FOR(a,b,c) for(int a=(b);a<(c);a++)
using namespace std;

const int maxn = 100+10;

int h[maxn][maxn],sum[maxn];
bool d[maxn*maxn];
int cnt[maxn*maxn];
int n;

void dp(int tot,int* A) {
    int len=A[0];
    d[0]=true;
    FOR(i,1,len)
       for(int j=tot;j>=A[i];j--)
       {
          d[j] = d[j] || d[j-A[i]];
       }
    FOR(i,0,tot+1) cnt[i] += d[i];
}

int main() {
    ios::sync_with_stdio(false);
    cin>>n;
    int x,R=0;
    FOR(i,0,n) {
        h[i][0]=1;
        while(cin>>x && x!=-1) {
            h[i][h[i][0]++]=x;
            sum[i] += x;
        }
        R=max(R,sum[i]);
    }

    FOR(j,0,n) 
    {
       memset(d,0,sizeof(d));
       dp(sum[j],h[j]);
    }
    for(int i=R;i>=0;i--)
       if(cnt[i]>=n) {
              cout<<i<<"
";
              break;
       }
    return 0;
}