CCF201809-4 再卖菜（记忆化搜索 / 差分约束）

问题描述

  在一条街上有n个卖菜的商店，按1至n的顺序排成一排，这些商店都卖一种蔬菜。
  第一天，每个商店都自己定了一个正整数的价格。店主们希望自己的菜价和其他商店的一致，第二天，每一家商店都会根据他自己和相邻商店的价格调整自己的价格。具体的，每家商店都会将第二天的菜价设置为自己和相邻商店第一天菜价的平均值（用去尾法取整）。
  注意，编号为1的商店只有一个相邻的商店2，编号为n的商店只有一个相邻的商店n-1，其他编号为i的商店有两个相邻的商店i-1和i+1。
  给定第二天各个商店的菜价，可能存在不同的符合要求的第一天的菜价，请找到符合要求的第一天菜价中字典序最小的一种。
  字典序大小的定义：对于两个不同的价格序列(a1, a2, …, an)和(b1, b2, b3, …, bn)，若存在i (i>=1), 使得ai<bi，且对于所有j<i，aj=bj，则认为第一个序列的字典序小于第二个序列。

输入格式

输入的第一行包含一个整数n，表示商店的数量。
第二行包含n个正整数，依次表示每个商店第二天的菜价。

输出格式

输出一行，包含n个正整数，依次表示每个商店第一天的菜价。

样例输入

8
2 2 1 3 4 9 10 13

样例输出

2 2 2 1 6 5 16 10

数据规模和约定

对于30%的评测用例，2<=n<=5，第二天每个商店的菜价为不超过10的正整数；
对于60%的评测用例，2<=n<=20，第二天每个商店的菜价为不超过100的正整数；
对于所有评测用例，2<=n<=300，第二天每个商店的菜价为不超过100的正整数。
请注意，以上都是给的第二天菜价的范围，第一天菜价可能会超过此范围。

思路

• 前几周课上讲的一道DFS剪枝的例题，首先根据题意推算出如下公式：
  设第一天商店 $i$ 的定价为 $a[i]$，第二天商店 $i$ 的定价为 $b[i]$
  $a[1]=1到200$
  $a[2]=2\ast b[1]-a[1]+i,(i=0,1)$
  $a[x]=3\ast b[x-1]-a[x-2]-a[x-1]+i,(i=0,1,2)$
  $a[n]=2\ast b[n]-a[n-1]+i,(i=0,1)$
  因为第二天每个商店的菜价为不超过100的正整数，$a[1]$的取值范围为1到200，又因为第二天的价格计算方法用的是去尾法取整，所以可以用余数来往后推算第一天每一个商店的定价，这里要引入记忆化搜索来减少重复的搜索过程，具体做法是设置一个$vis$数组，用$vis[x][a[x-1]][a[x-2]]$来这一状态是否已被访问，若已被访问过则直接剪去。
• 补充：今天课上又学习了差分约束求解，再来练习一下。
  设 $s[i]$ 为数组a的前缀和，$s[0]=0$，由第二天价格与第一天价格的关系可得以下不等式组：
  $3\ast b[i]\le s[i+1]-s[i-2]\le 3\ast b[i]+2，i>=2$
  $2\ast b[1]\le s[2]-s[0]\le 2\ast b[1]+1$
  $2\ast b[n]\le s[n]-s[n-2]\le 2b[n]+1$
  保证最终的结果为正数则可得：$s[i]-s[i-1]\ge 1$
  又因为题目要求字典序最小所以构造 $\ge$ 的不等式组，跑最长路求解。

代码实现

#include <bits/stdc++.h>
using namespace std;

int n,a[310],b[310];
bool vis[310][310][310];

void dfs(int x){
	if(x==1){
		for(int i=1;i<=200;i++){
			a[x]=i;
			dfs(x+1);
		}
	}
	else if(x==2){
		for(int i=0;i<2;i++){
			a[x]=2*b[1]-a[1]+i;
			if(a[x]>0)
				dfs(x+1);
		}
	}
	else if(x==n){
		for(int i=0;i<2;i++){
			a[n]=2*b[n]-a[n-1]+i;
			if(((a[n]+a[n-1]+a[n-2])/3)==b[n-1]&&a[n]>0){
				for(int j=1;j<=n;j++){
					printf("%d ",a[j]);
				}
				printf("\n");
				exit(0);				
			}
		}
		return;
	}
	else{
		if(vis[x][a[x-1]][a[x-2]])
			return;
		vis[x][a[x-1]][a[x-2]]=1;
		for(int i=0;i<3;i++){
			a[x]=3*b[x-1]-a[x-2]-a[x-1]+i;
			if(a[x]>0)
				dfs(x+1);
		}	
	}
} 

int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		scanf("%d",&b[i]);
	}
	memset(vis,0,sizeof(vis));
	dfs(1);
	return 0;
}

#include <bits/stdc++.h>
using namespace std;
const int maxn=310;
const int maxm=50000;
const int inf=1e9;
int n,tot,a[maxn],b[maxn],dis[maxn],cnt[maxn],head[maxn];
bool inq[maxn];

struct Edge{
	int u,v,w,next;
}edge[maxm];

void init(int nn){
	tot=0;
	for(int i=0;i<=nn;i++){
		head[i]=-1;
		dis[i]=-inf;
		inq[i]=0;
		cnt[i]=0;
	}
}

void add(int x,int y,int w){
	edge[++tot].u=x;
	edge[tot].v=y;
	edge[tot].w=w;
	edge[tot].next=head[x];
	head[x]=tot;
}

void spfa(int s){
	queue<int> q;
	q.push(s);
	inq[s]=1;
	dis[s]=0;
	while(!q.empty()){
		int x=q.front();
		q.pop();
		inq[x]=0;
		for(int i=head[x];i!=-1;i=edge[i].next){
			int y=edge[i].v,w=edge[i].w;
			if(dis[y]<dis[x]+w){
				dis[y]=dis[x]+w;
				if(!inq[y]){
					q.push(y);
					inq[y]=1;
				}
			}
		}
	}
}

int main()
{
	scanf("%d",&n);
	init(n);
	for(int i=1;i<=n;i++){
		scanf("%d",&b[i]);
	}	
	add(0,2,2*b[1]);
	add(2,0,-(2*b[1]+1));
	add(n-2,n,2*b[n]);
	add(n,n-2,-(2*b[n]+1));
	add(0,1,1);
	add(n-1,n,1);
	for(int i=2;i<n;i++){
		add(i-1,i,1);
		add(i-2,i+1,3*b[i]);
		add(i+1,i-2,-(3*b[i]+2));
	}
	spfa(0);
	for(int i=1;i<=n;i++)
		printf("%d ",dis[i]-dis[i-1]);
}