问题描述
在一条街上有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的正整数。
请注意,以上都是给的第二天菜价的范围,第一天菜价可能会超过此范围。
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define inf 2147483647
#define maxn 10005
#define maxm 500005
using namespace std;
int n, m, s, num_edge=0;
int dis[maxn], vis[maxn], head[maxm];
int a[maxn];
int sign;
struct Edge {
int next, to, w;
}edge[maxm]; //结构体表示静态邻接表
void init() {
memset(head, -1, sizeof(head));
}
void add(int u, int v, int w) {
edge[sign].to = v;
edge[sign].w = w;
edge[sign].next = head[u];
head[u] = sign++;
}
void spfa() {
queue<int> q; //spfa用队列,这里用了STL的标准队列
for(int i=0; i<=n; i++) {
q.push(i);
dis[i] = 0; //带权图初始化,这点与spfa不同,因为我们要求最长路,所以初始设为0才能更新
vis[i] = 0; //记录点i是否在队列中,同dijkstra算法中的visited数组
}
q.push( s ); dis[s] = 0; vis[s] = 1; //第一个顶点入队,进行标记
while( !q.empty() ) {
int u = q.front(); //取出队首
q.pop(); vis[u] = 0; //出队标记
for( int i = head[u]; ~i; i = edge[i].next ) { //邻接表遍历,不多解释了(也可用vector代替)
int v = edge[i].to;
if( dis[v] < dis[u] + edge[i].w ) { //如果有最长路就更改
dis[v] = dis[u] + edge[i].w;
if( vis[v] == 0 ) { //未入队则入队
vis[v] = 1; //标记入队
q.push(v);
}
}
}
}
}
int main()
{
cin>>n;
init();
for(int i=1;i<=n;i++)
cin>>a[i];
for(int i=0;i<=n-3;i++)
{
add(i,i+3,3*a[i+2]);
add(i+3,i,-(3*a[i+2]+2));
}
add(0,2,a[1]*2);
add(2,0,-(a[1]*2+1));
add(n-2,n,a[n]*2);
add(n,n-2,-(a[n]*2+1));
for(int i=1;i<=n;i++)
{
add(i-1,i,1);
}
spfa();
a[1]=dis[1];
for(int i=2;i<=n;i++)
a[i]=dis[i]-dis[i-1];
for(int i=1;i<=n;i++)
{
if(i!=1)
cout<<" ";
cout<<a[i];
}
}