Codeforces Round #197 (Div. 2) C. Xenia and Weights(暴力dfs/dp)

题目

先给一个长度为10的字符串,第i(1-based)位为0/1,代表重量为i的砝码是否为无穷个

两人轮流放初始平衡的天平上放砝码,一人放左盘一人放右盘,不能和上一个人放的砝码相同,

且本次放完,必须天平往自己这边倾斜,

问是否存在放砝码方案,输出方案

思路来源

https://www.cnblogs.com/dwtfukgv/p/5653550.html

题解

自己写了90行dp,一看题解是40行dfs,补一下吧

dp[i][j][k],代表第i轮先手-后手差为j且这一轮选的为k时,两人放的最大轮数为dp[i][j][k]

pre[i][j][k],代表第i轮先手-后手差为j且这一轮选的为k时,达到最大轮数时上一轮选的砝码为pre[i][j][k]

dfs,第i轮先手选了lsum,后手选了rsum,最后一个选的人选的是last时是否能放m轮,确定一组解即可

要用最简洁的方法coding啊

代码

dfs

#include
using namespace std;
vectora,ans;
char s[11];
int m;
bool dfs(int cnt,int l,int r,int last)
{
	if(cnt==m)return 1;
	if(cnt%2==0)
	{
		for(int i=0;i=0;--i)
		printf("%d%c",ans[i]," \n"[!i]);
	}
	return 0;
}

dp

#include
using namespace std;
typedef long long ll;
const int off=10;
int sum,now,pos;
int n,m,dp[1002][21][11],pre[1002][21][11];
//dp[i][j][k]:i-th round x-y==j[-10,10] choose k last time
char s[11];
bool ok[11],yes;
vectorans;
int main()
{
	scanf("%s",s+1);
	scanf("%d",&m);
	for(int i=1;i<=10;++i)
	if(s[i]=='1')ok[i]=1;
	memset(dp,-1,sizeof dp);
	dp[0][off][0]=0;//x-y==0
	for(int i=1;i<=m;++i)
	{
		if(i&1)
		{
			for(int j=1;j<=10;++j)//j kilos this time
			{
				if(!ok[j])continue;
				for(int k=-10;k<=0;++k)//x-y==k
				{
					if(k+j<=0)continue;
					for(int l=0;l<=10;++l)
					{
						if(j==l)continue;
						if(i!=1&&l==0)continue;
						if(dp[i-1][k+off][l]==-1)continue;
						if(dp[i][k+j+off][j]=0)continue;
					for(int l=1;l<=10;++l)
					{
						if(j==l)continue;
						if(dp[i-1][k+off][l]==-1)continue;
						if(dp[i][k-j+off][j]=1;--i)
	 	{
		 	int sg=i&1?-1:1;
		 	ans.push_back(pos);
		 	now=pos;
	 		pos=pre[i][sum][pos];
	 		sum+=sg*now;
		}
		for(int i=ans.size()-1;i>=0;--i)
		printf("%d%c",ans[i]," \n"[i==0]);
	 }
	return 0;
} 

 

你可能感兴趣的:(dfs,dp,搜索(bfs/dfs)/回溯)