码迷,mamicode.com
首页 > 其他好文 > 详细

URAL 1244 Gentlemen

时间:2018-01-29 18:19:27      阅读:121      评论:0      收藏:0      [点我收藏+]

标签:style   tor   while   tar   gpo   ack   names   push   its   

URAL 1244

思路:dp,有点类似背包,不过不需要求最大价值,只要求方案数就可以了。

状态:dp[i]表示和为i的方案数

初始状态:dp[0]=1

状态转移:dp[i]=∑dp[i-a[k]] (1<=k<=n)

用一个pre[]数组来记录路径

代码:

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pb push_back
#define mem(a,b) memset(a,b,sizeof(a))

const int N=1e5+5;
int dp[N];
int pre[N];
int a[N];
bool vis[N];
vector<int>ans;
int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    int m,n;
    cin>>m;
    cin>>n;
    for(int i=1;i<=n;i++)cin>>a[i];
    if(m>1e5){
        cout<<0<<endl;
        return 0;
    }
    dp[0]=1;
    mem(pre,-1);
    for(int i=1;i<=n;i++){
        for(int j=m;j>=a[i];j--){
            if(dp[j-a[i]]){
                dp[j]+=dp[j-a[i]];
                if(pre[j]==-1)pre[j]=i;
            }
        }
    }
    if(dp[m]==0)cout<<0<<endl;
    else if(dp[m]==1){
        int t=pre[m];
        int sum=m;
        while(~t){
            vis[t]=true;
            m-=a[t];
            t=pre[m];
        }
        for(int i=1;i<=n;i++)if(!vis[i])ans.pb(i);
        for(int i=0;i<ans.size();i++){
            if(i==ans.size()-1)cout<<ans[i]<<endl;
            else cout<<ans[i]<<" ";
        }
    }
    else cout<<-1<<endl;
    return 0;
}

 

URAL 1244 Gentlemen

标签:style   tor   while   tar   gpo   ack   names   push   its   

原文地址:https://www.cnblogs.com/widsom/p/8378652.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!