poj1742coins【单调队列优化多重背包】楼教主八题

早上好不容易想明白,就是不知道自己哪里错了==

我的错的

/***************
poj1742
2016.2.24
不知道哪里错了啊啊啊啊啊

***************/
#include <iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int n,m,c[102],a[102],q[102];
bool dp[100006];
int main()
{
   // freopen("cin.txt","r",stdin);
    while(~scanf("%d%d",&n,&m))
    {
        if(n==0&&m==0) break;
        for(int i=1;i<=n;i++) scanf("%d",&a[i]);
        for(int i=1;i<=n;i++) scanf("%d",&c[i]);
        int   sum=0;
        memset(dp,0,sizeof(dp));
        dp[0]=1;
        for(int i=1;i<=n;i++)
        {
            if(c[i]==1)
                for(int v=m;v>=a[i];v--)
                    if(dp[v-a[i]]) dp[v]=1;
            else if(a[i]*c[i]>=m)
                for(int v=a[i];v<=m;v++)
                    if(dp[v-a[i]]) dp[v]=1;
            else
            {
                for(int d=0;d<a[i];d++)
                {
                    int st=0,ed=-1;
                    sum=0;
                    for(int v=d;v<=m;v+=a[i])
                    {
                        if(ed-st==c[i]) sum-=q[st++];
                        q[++ed]=dp[v];
                        sum+=dp[v];
                        if(!dp[v]&&sum) dp[v]=1;
                    }
                }
            }
        }
        sum=0;
        for(int i=1;i<=m;i++) if(dp[i]) sum++;
        printf("%d\n",sum);
    }
    return 0;
}

ac的

#include <iostream>
#include<cstring>
using namespace std;
 
const int MAXN = 150;
int w[MAXN], c[MAXN];
int n, m;
bool dp[100000+10], queue[100000+10];
 
int solve_dp() {
    memset(dp, 0, sizeof(dp));
    memset(queue, 0, sizeof(queue));
    dp[0] = true;
 
    for(int i = 0; i < n; i ++) {
        if(c[i] == 1) {     // 仅允许一个包, 变成01背包问题
            for(int v = m; v >= w[i]; v--) {
                if(!dp[v] && dp[v-w[i]])
                    dp[v] = 1;
            }
        }else if(c[i]*w[i] >= m) {   // 完全背包问题, 即 w[i]*c[i] < m, 放入件数的限制是 c[i]
            for(int v = w[i]; v <= m; v++) {
                if(!dp[v] && dp[v-w[i]])
                    dp[v] = 1;
            }
        }else{  // 严格的分组背包问题
            for(int d = 0; d < w[i]; d++) {      // 对于所有余数 d [0, w[i])
                // 窗口大小为 c[i]
                int sum = 0, st = 0, ed = -1;   //st,ed 单调队列的开始和结尾, sum 队列中是否有一个 true
                for(int v = d; v <= m; v+= w[i]) {   // 完全背包 model, 但步长是 w[i]
                    if(ed - st == c[i]) {   // 窗口大小为0, 移除队首元素, 队首后移一位
                        sum -= queue[st++];
                    }
                    queue[++ed] = dp[v];
                    sum += dp[v];
                    if(!dp[v] && sum)
                        dp[v] = 1;
                }
            }
        }
    }
    int res = 0;
    for(int i = 1; i <= m; i ++)
        res += dp[i];
    return res;
 
}
int main() {
   // freopen("E:\\Copy\\ACM\\测试用例\\in.txt", "r", stdin);
    while(cin >> n >> m && n != 0) {
        for(int i = 0; i < n; i ++) {
            scanf("%d", &w[i]);
        }
        for(int i = 0; i < n; i ++) {
            scanf("%d", &c[i]);
        }
        // main function
        cout << solve_dp() << endl;
    }
    return 0;
}


你可能感兴趣的:(dp,poj,单调队列)