URAL1244. Gentlemen(背包）

链接

以前做的题 VJ太水了 数组里面的数可能会小于0 当时没判断

#include <iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<stdlib.h>
using namespace std;
int dp[100010],w[110],path[110],o,f[110],n,ff[110];
int flag,tw;
void dfs(int sw,int v,int u)
{
    int j,i;
    if(flag) return ;
    path[v] = u;
    if(sw==0)
    {
        flag  =1;
        for(i = 1; i <= v ; i++)
        ff[path[i]] = 1;
        return ;
    }
    if(sw<0) return ;
    for(j = 1; j <= n ; j++)
    {
        if(sw-w[j]<0)
        continue;
        if(!f[j]&&dp[sw-w[j]])
        {
            f[j] = 1;
            dfs(sw-w[j],v+1,j);
            f[j] = 0;
        }
    }
}
int main()
{
    int i,j;
    cin>>tw;
    cin>>n;
    for(i = 1; i <= n ; i++)
    cin>>w[i];
    dp[0] = 1;
    for(i = 1 ; i <= n ;i++)
        for(j = tw ; j>=w[i] ; j--)
        {
            dp[j] += dp[j-w[i]];
        }
    int sw = tw;
    if(dp[tw]==0)
    cout<<"0
";
    else if(dp[tw]>1)
    cout<<"-1
"<<endl;
    else
    {
        for(i = 1; i <= n ; i++)
        {
            if(sw-w[i]<0)
            continue;
            if(dp[sw-w[i]])
            {
                f[i] = 1;
                dfs(sw-w[i],1,i);
                f[i] = 0;
            }
            if(flag) break;
        }
        int kk=0;
        for(i = 1; i <= n ; i++)
        {
            if(!ff[i])
            {
                if(kk)
                printf(" ");
                printf("%d",i);
                kk++;
            }
        }
        puts("");
    }
    return 0;
}