ccf 再买菜 搜索 dfs

//递推关系式：(b[n-1]+b[n]+b[n+1])/3=a[n]
//所以b[n+1]=3*a[n]-b[n-1]-b[n]，或b[n+1]=3*a[n]-b[n-1]-b[n]+1，或b[n+1]=3*a[n]-b[n-1]-b[n]+2
//有三种结果是因为3的余数可以是0，1，2
//这里要考虑特殊情况，就是第一天和最后一天是除以2，所以写两种情况就可以了
#include <stdio.h>//因为数据比较小，所以可以直接搜索，枚举每一种可能性
#include <string.h>
#include <algorithm>
#include <iostream>
#define inf 0x3f3f3f3f
using namespace std;
const int maxn = 500;
int t,a[maxn],b[maxn];
bool f[maxn][maxn][maxn];
void dfs(int n, int x, int y)//表示第n天，第n-1天的菜价和第n天的菜价
{
	if (f[n][x][y]) return;
	f[n][x][y] = 1;
	if (n == t - 1)//这里需要好好理解理解
	{
		if ((3 * a[n] - x) / 2 == a[t] || (3 * a[n] - x + 1) / 2 == a[t] || (3 * a[n] - x + 2) / 2 == a[t])
		{
			for (int i = 1; i < t; i++) printf("%d ", b[i]);
			for (int i = 0; i < 3; i++)
			{
				if ((3 * a[n] - x + i) / 2 == a[t])
				{
					b[t] = 3 * a[n] - x - y + i;//为什么只要if成立就可以直接输出呢，是因为最后一个如果满足if条件的话，就可以确定最后一个数了，这里就不再需要去像下面的for循环一样的枚举来确定数
					printf("%d\n", b[t]);
					exit(0);//直接退出，节约时间
				}
			}
		}
		return;
	}
	for (int i = 0; i < 3; i++)
	{
		b[n + 1] = 3 * a[n] - y - x + i;//枚举过程

		if(b[n+1]>=1) dfs(n + 1, y, b[n + 1]);
	}
}

int main()
{
	cin >> t;
	for (int i = 1; i <= t; i++) scanf("%d", &a[i]);
	memset(f, 0, sizeof(f));
	for (int i = 1; i <= 2 * a[1]; i++)
	{
		b[1] = i;b[2] = 2 * a[1] - i;//第一种可能
		dfs(2, b[1], b[2]);
		b[1] = i; b[2] = 2 * a[1] - i + 1;//第二种可能
		dfs(2, b[1], b[2]);
	}
	return 0;
}

　　

试题编号：	201809-4
试题名称：	再卖菜
时间限制：	1.0s
内存限制：	256.0MB
问题描述：	问题描述 　　在一条街上有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的正整数。 　　请注意，以上都是给的第二天菜价的范围，第一天菜价可能会超过此范围。