CCF CSP认证201809-4 再卖菜

201809-4 再卖菜

题目

思路

枚举、dfs暴力求解，80分。
设置一个vis数组可以有效剪枝，100分代码见下方。（注意第3、5、6行）

由于有递推公式：
$$b[k+1]=3\ast a[k]-b[k-1]-b[k]+i（i从1到3）$$
对于某特定k，如果b[k]与b[k-1]都确定了，那之后的dfs结果就确定了。
因此可根据k、b[k]、b[k-1]，设置一个三维数组避免重复搜索。

代码如下（80分）

#include<cstdio>
int n,a[305],b[305];
bool dfs(int k,int x){
 	if(x<=0) return false;//每一项都必须为正
 	b[k]=x;
 	if(k==n){
  		if((x+b[n-1])/2==a[n]) return true;
  		return false;
 	}
 	for(int i=0;i<3;i++)//前两项已知的情况下，第三项有三种可能
  		if(dfs(k+1,a[k]*3-b[k-1]-b[k]+i)) return true;
 	return false;
}
int main(){
 	scanf("%d",&n);
 	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
 	for(b[1]=1;b[1]<a[1]+a[2]+a[3];b[1]++){//枚举第一项，第二项有相应的两种情况
  		if(dfs(2,a[1]*2-b[1])) break;
  		if(dfs(2,a[1]*2-b[1]+1)) break;
 	}
 	for(int i=1;i<=n;i++) printf("%d ",b[i]);
 	return 0;
}

AC代码

#include<cstdio>
int n,a[305],b[305];
bool vis[305][305][305]={false};//商店序号和菜价的范围都是1至300
bool dfs(int k,int x){
 	if(x<=0||vis[k][x][b[k-1]]) return false;
 	vis[k][x][b[k-1]]=true;
 	b[k]=x;
 	if(k==n){
  		if((x+b[n-1])/2==a[n]) return true;
  		return false;
 	}
 	for(int i=0;i<3;i++)
  		if(dfs(k+1,a[k]*3-b[k-1]-b[k]+i)) return true;
 	return false;
}
int main(){
 	scanf("%d",&n);
 	for(int i=1;i<=n;i++) scanf("%d",&a[i]);
 	for(b[1]=1;b[1]<a[1]+a[2]+a[3];b[1]++){
  		if(dfs(2,a[1]*2-b[1])) break;
  		if(dfs(2,a[1]*2-b[1]+1)) break;
 	}
 	for(int i=1;i<=n;i++) printf("%d ",b[i]);
 	return 0;
}