[동적 기획] Vijos P1071 새해 에피소드 카드

1434 단어
상태가 바뀔 때 바뀐 경로를 기록해야 하는데 주석에 표시된 한 줄의 코드가 매우 중요하다.
제목 링크: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;
}

좋은 웹페이지 즐겨찾기