【动态规划】Vijos P1071 新年趣事之打牌

最新推荐文章于 2022-05-21 16:51:51 发布

lwfcgz

最新推荐文章于 2022-05-21 16:51:51 发布

阅读量1.4k

点赞数

分类专栏： ACM_NOI

本文链接：https://blog.csdn.net/lwfcgz/article/details/8476778

版权

ACM_NOI 专栏收录该内容

12 篇文章 0 订阅

订阅专栏

博客详细解析了如何使用动态规划方法解决Vijos平台上的P1071题目，即新年趣事中的打牌问题。在求解过程中，强调了状态转移路径记录的重要性，并指出某行代码在实现动态规划策略中的关键作用。

摘要由CSDN通过智能技术生成

需要在状态转移的时候记录转移的路径，注释中标出的一行代码非常重要。

题目链接：https://vijos.org/problems/P1071

#include<iostream>
#include<stack>
#include<string.h>
#include<algorithm>
#define SIZE 100005
using namespace std;
 
int main(){
    int totalW,sum;
    int n,wei[105];
    int dp[SIZE],path[SIZE];
     
    memset(dp,0,sizeof(dp));
    memset(path,0,sizeof(path));
     
    cin>>totalW>>n;
    sum=0; 
    for(int i=0;i<n;i++){
        cin>>wei[i];
        sum+=wei[i];
    }
    int left=sum-totalW;
     
    dp[0]=1;
    for(int i=0;i<n;i++){
        for(int j=left;j>=wei[i];j--){
            //该条语句非常关键
            if((dp[j]==0) && (dp[j-wei[i]]>0))
                path[j]=i+1;
            dp[j]+=dp[j-wei[i]];
             
        }
    }
     
    if(dp[left]==0)
        cout<<0<<endl;
    else if(dp[left]>1)
        cout<<-1<<endl;
    else if(dp[left]==1){
        stack<int> s;    
        while(s.empty()==false)
            s.pop();
        while(path[left]!=0){
            s.push(path[left]);
            left=left-wei[path[left]-1];
        }
        while(s.size()>1){
            int tmp=s.top();
            cout<<tmp<<" ";
            s.pop();
        }
        cout<<s.top();
        cout<<endl;
    }
    //system("pause");
    return 0;
}