CF 339C Xenia and Weights 背包

地址:http://codeforces.com/contest/339/problem/C

题意:给1-10种重量的砝码,求一个序列,左右依次放,每次放后天平的倾斜会改变且连续两个数不相等。

解法:

比赛时候脑残YY个贪心,然后怒跪,叉姐说了是背包,当时居然毫无感觉。

dp[n][sum][last]表示第n轮放的天平的值是sum,而最后一个放的砝码是last,那么,一个裸背包诞生了,注意在输出序列的时候要判一下是不是出现了连续相等的情况,在这里wa了一次。

//339C. Xenia and Weights
//背包
#include<iostream>
#include<cstring>
#include<vector>
using namespace std;

bool dp[1001][22][11];
bool a[11];
int n;

void dfs(int i,int j,int k){
    if(i==0) return ;
    int t;
    if(j>10) t=j-k;
    else t=j+k;
    for(int l=0;l<11;l++){
        if(dp[i-1][t][l]&&l!=k){    //这里还需要判一下,不能麻痹大意啊。。。
            dfs(i-1,t,l);
            break;
        }
    }
    cout<<k<<" ";
}

int main(){
    int i,j,k,l;
    string s;
    cin>>s;
    for(i=1;i<=10;i++){
        if(s[i-1]=='1')
            a[i]=1;;
    }
    cin>>n;
    dp[0][10][0]=1;
    for(i=1;i<11;i++){
        if(a[i]){
            dp[1][i+10][i]=1;
        }
    }
    for(i=1;i<n;i++){
        for(j=0;j<22;j++){
            if(j==10) continue;
            for(k=0;k<11;k++){
                if(dp[i][j][k]){
                    for(l=1;l<=10;l++){
                        if(!a[l]||l==k) continue;
                        int t;
                        if(j>10){
                            t=j-l;
                            if(t>=10) continue;
                        }
                        if(j<10){
                            t=j+l;
                            if(t<=10) continue;
                        }
                        dp[i+1][t][l]=1;
                    }
                }
            }
        }
    }
    for(i=0;i<22;i++){
        for(j=0;j<11;j++){
            if(dp[n][i][j]){
                break;
            }
        }
        if(j!=11) break;
    }
    if(i==22) cout<<"NO"<<endl;
    else{
        cout<<"YES"<<endl;
        dfs(n,i,j);
        cout<<endl;
    }
    return 0;
}


你可能感兴趣的:(背包)