题目
先给一个长度为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<bits/stdc++.h>
using namespace std;
vector<int>a,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<a.size();++i)
{
if(a[i]==last)continue;
if(l+a[i]<=r)continue;
if(dfs(cnt+1,l+a[i],r,a[i]))
{
ans.push_back(a[i]);
return 1;
}
}
}
else
{
for(int i=0;i<a.size();++i)
{
if(a[i]==last)continue;
if(r+a[i]<=l)continue;
if(dfs(cnt+1,l,r+a[i],a[i]))
{
ans.push_back(a[i]);
return 1;
}
}
}
return 0;
}
int main()
{
scanf("%s",s+1);
scanf("%d",&m);
for(int i=1;i<=10;++i)
if(s[i]=='1')a.push_back(i);
if(!dfs(0,0,0,0))puts("NO");
else
{
puts("YES");
for(int i=ans.size()-1;i>=0;--i)
printf("%d%c",ans[i]," \n"[!i]);
}
return 0;
}
dp
#include<bits/stdc++.h>
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;
vector<int>ans;
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]<dp[i-1][k+off][l]+1)
{
//printf("now:%d pre:%d\n",k+j,k);
dp[i][k+j+off][j]=dp[i-1][k+off][l]+1;
pre[i][k+j+off][j]=l;
}
}
}
}
}
else
{
for(int j=1;j<=10;++j)
{
if(!ok[j])continue;
for(int k=0;k<=10;++k)
{
if(k-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]<dp[i-1][k+off][l]+1)
{
//printf("now:%d pre:%d\n",k-j,k);
dp[i][k-j+off][j]=dp[i-1][k+off][l]+1;
pre[i][k-j+off][j]=l;
}
}
}
}
}
}
for(int i=0;i<=20&&!yes;++i)
{
for(int j=1;j<=10&&!yes;++j)
{
if(dp[m][i][j]==m)
{
sum=i;
yes=1;
pos=j;
}
}
}
if(!yes)puts("NO");
else
{
puts("YES");
for(int i=m;i>=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;
}