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

点击打开链接

题意:有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;
}