再卖菜(CCF201809-4)（差分约束）

试题编号：	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的正整数。 　　请注意，以上都是给的第二天菜价的范围，第一天菜价可能会超过此范围。

 

题解：这是一道差分约束题，首先可以列出一组式子

a1<=(x1+x2)/2<a1+1

a2<=(x1+x2+x3)/3<a2+1

....

an<=(x(n-1)+xn)/2<an+1

然后展开看一下

2*a1<=(x1+x2)<2*a1+2

3*a2<=(x1+x2+x3)<3*a2+3

......

2*an<=(x(n-1)+xn)<2*an+2

因为x都是整数么，所以可以将<变为<=

2*a1<=(x1+x2)<=2*a1+1

3*a2<=(x1+x2+x3)<=3*a2+2

......

2*an<=(x(n-1)+xn)<=2*an+1

但是呢？差分约束不是都两个么 ，这个属于三个，那么考虑s=x0+x1+x2+....xn，之后就又有了一组式子，这个x0必然为0

2*a1<=s2-s0<=2*a1+1

3*a2<=s3-s0<=3*a2+2

3*a3<=s4-s1<=3*a3+2

.......

2*an<=sn-s(n-2)<=2*an+1

之后就是差分约束不是方向统一么，那么就统一用>=吧，之后式子拆解成两个

s2-s0>=2*a1  s0-s2>=-2*a1-1

s3-s0>=3*a3  s0-s3>=-3*a2-2

.....

sn-s(n-2)>=2*an  s(n-2)-sn>=-2*an-1

也就是sj-si>=x，建立从i到j有一条x的边，之后记住差分约束就是<=求解最短路，>=求解最长路，然后这个题拿spfa跑最长路，最后还有一点要注意的就是各个x都要是正整数，也就是还得需要建立一个约束条件，si-s(i-1)=xi，因为xi>=1,所以si-s(i-1)>=1，这样这个题就可以解了，其实还是蛮简单的，只怪当时没多练几个差分约束。

附上代码：

#include<bits/stdc++.h>

using namespace std;

const int maxm=1e3+50;
const int maxn=5e2+50;

struct edge{
    int v,w,next;
};
edge edges[maxm];
int head[maxn],tot;

void add_edges(int u,int v,int w)
{
    edges[tot].v=v;
    edges[tot].w=w;
    edges[tot].next=head[u];
    head[u]=tot++;
}

int n;
bool vis[maxn];
int cnt[maxn];
int dist[maxn];

bool spfa()
{
    queue<int>q;
    for(int i=0;i<n;i++){
        q.push(i);
        vis[i]=false;dist[i]=0;cnt[i]=1;
    }
    while(!q.empty()){
        int u=q.front();
        q.pop();
        vis[u]=0;
        for(int i=head[u];~i;i=edges[i].next){
            int v=edges[i].v;
            if(dist[v]<dist[u]+edges[i].w){
                dist[v]=dist[u]+edges[i].w;
                if(!vis[v]){
                    vis[v]=1;
                    q.push(v);
                    if(++cnt[v]>n){
                        return false;
                    }
                }
            }
        }
    }
    return true;
}

int a[maxn];

int main()
{
    memset(head,-1,sizeof(head));
    tot=0;
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
    }
    for(int i=0;i<n-2;i++){
        add_edges(i+3,i,-(a[i+2]*3+2));
        add_edges(i,i+3,a[i+2]*3);
    }
    add_edges(2,0,-(a[1]*2+1));add_edges(0,2,a[1]*2);
    add_edges(n,n-2,-(a[n]*2+1));add_edges(n-2,n,a[n]*2);
    for(int i=1;i<=n;i++){
        add_edges(i-1,i,1);
    }
    if(spfa()){
        a[1]=dist[1];
        for(int i=2;i<n+1;i++){
            a[i]=dist[i]-dist[i-1];
        }
        printf("%d",a[1]);
        for(int i=2;i<n+1;i++){
            printf(" %d",a[i]);
        }
    }
    return 0;
}