问题描述
在一条街上有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的正整数。
请注意,以上都是给的第二天菜价的范围,第一天菜价可能会超过此范围。
分析:对第一天单独进行遍历,对之后的情况进行搜索,最后对最后一天的价格进行合理化判断。
第一天和最后一天的浮动范围都是[0,1],中间浮动范围是[0,2]。考虑到中间过程为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, 即分开为0,1,2三种情况继续搜索。最后一天判断
3*a[n-1]-b[n-2]+i)/2==a[n] (i = 0,1,2)是否成立。
#include<iostream>
#include<cstdlib>
using namespace std;
int n;///商店个数
int vis[400][400][400];///
int a[400]={0};///输入菜价
int b[400]={0};///前一天菜价
void dfs(int x, int y, int z)///第x天,第x-1天菜价,第x天菜价(x,b[x-1], b[x])
{
if(vis[x][y][z] == 1)return ;
vis[x][y][z] = 1;
if(x == n-1)
{
///对最后一天菜价进行处理判断
if((3*a[x]-y)/2==a[n]||(3*a[x]-y+1)/2==a[n]||(3*a[x]-y+2)/2==a[n])
{
for(int i=1;i<=x;++i)
cout<<b[i]<<" ";
for(int i=0;i<3;++i)
{
if((3*a[x]-y+i)/2==a[n])
{
cout<<3*a[x]-y-z+i;
exit(0); //直接结束程序,这样用时更短
}
}
}
return;
}
for(int i = 0; i < 3; i++)///中间部分分[0,2]三种情况
{
b[x+1] = 3 * a[x] - y - z + i;
///当得到的值大于0时才是可行的
if(b[x+1]>=1)dfs(x+1,z,b[x+1]);
}
}
int main()
{
cin>>n;
for(int i = 1; i <= n; i++)cin>>a[i];
///初始分加一和不加一两种情况
for(int i = 1; i <= 2*a[1]; i++)
{
b[1]=i;
b[2]=2*a[1]-i;
dfs(2,i,b[2]);
b[1]=i;
b[2]=2*a[1]-i+1;
dfs(2,i,b[2]);
}
return 0;
}
/*
8
2 2 1 3 4 9 10 13
*/