Codeforces 339C Xenia and Weights DP+打印路径

来源：互联网发布：盗取qq密码软件编辑：程序博客网时间：2024/04/28 06:04

点击打开链接

题意:有weight 1~10砝码无限个,每次放的砝码不能相同&&当前放完砝码的一边重量大,问能否放m个砝码若能则输出解(m<=1e3)

错误解法greedy:每次使大的一边减去小的一边尽量小,容易举出反例:2,3,6,10 greedy只能算到9 ans为1000

记balance为 sum(l)-sum(r) 依题意每次操作后,balance变号.

dp[i][j][k] 放第i个砝码砝码重量为k时 balance为j 是否存在,答案只要找到dp[m][j][k]是否存在即可

记录前一个状态决策,递归打印解即可.

#include <bits/stdc++.h>using namespace std;const int N=1e3+20;char s[N];int m,dp[N][11][11];//dp[i][j][k] 放第i个砝码 砝码重量为k时 balance为j 是否存在 //dp[i][j][k]= dp[i-1][k-j][t]==0?0:1 t!=k int path[N][11][11]; void print(int i,int j,int k){if(i!=1)print(i-1,k-j,path[i][j][k]);cout<<k<<' ';}int main(){scanf("%s%d",s+1,&m);memset(dp,0,sizeof(dp));for(int i=1;i<=10;i++){if(s[i]=='1'){dp[1][i][i]=1;}}//O(1e6)for(int i=2;i<=m;i++){for(int j=1;j<10;j++){for(int k=j+1;k<=10;k++){if(s[k]=='1'){for(int t=1;t<=10;t++){if(s[t]=='1'&&t!=k){if(dp[i-1][k-j][t]){dp[i][j][k]=1;path[i][j][k]=t;//记录前一个状态 }}}}}}}bool flag=false;for(int j=1;j<=10;j++){for(int k=1;k<=10;k++){if(dp[m][j][k]){puts("YES");print(m,j,k);cout<<endl;flag=true;break;}}if(flag)break;}if(flag==false)puts("NO");return 0;}

0 0