CCF认证 2018-09 再卖菜

可以用dfs记忆化搜索+剪枝

更稳的方法是用差分约束，因为会有三元一次的不等式，所以把不等式全部改为前缀和的不等式

s[i]代表前i项的和，如a1+a2+a3>=b可改为s3-s0>=b，a1+a2+a3<=c改为s0-s3>=-c

以此类推，就全部转化为s[x]-s[y]>=z的形式，然后按照差分约束建图

用spfa求解，一开始就把所有点入队，求最长路，便可以得到字典序最小解

下面是两种方法代码：

//记忆化搜索+剪枝
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long ll;
const int N=300+10;
int a[N],up[N],down[N]; //保存第二天菜价，以及第一天和邻居菜价的和的上限，下限
int n;
int b[N]; //保存第一天菜价
int dp[N][N][N]; //初始为-1
int dfs(int d)
{
    if(dp[d][b[d-1]][b[d-2]]!=-1) return dp[d][b[d-1]][b[d-2]]; //记忆化
    if(d==n)
    {
        if(b[n-1]+b[n-2]>=down[n-1]&&b[n-1]+b[n-2]<=up[n-1]) //最后检测一次
            return 1;
        return 0;
    }
    int l=down[d-1]-b[d-1]-b[d-2],r=up[d-1]-b[d-1]-b[d-2]; //根据前两个菜价推算这个菜价范围
    if(l<1) l=1; //最少为1
    if(r<l) return dp[d][b[d-1]][b[d-2]]=0; //剪枝
    for(int i=l;i<=r;i++)
    {
        b[d]=i;
        if(dfs(d+1)) return dp[d][b[d-1]][b[d-2]]=1; //可行
    }
    return dp[d][b[d-1]][b[d-2]]=0; //不可行
}
int main()
{
    scanf("%d",&n);
    for(int i=0;i<n;i++)
        scanf("%d",&a[i]);
    //预处理上下限
    up[0]=a[0]*2+1;
    down[0]=a[0]*2;
    up[n-1]=a[n-1]*2+1;
    down[n-1]=a[n-1]*2;
    for(int i=1;i<n-1;i++)
    {
        up[i]=a[i]*3+2;
        down[i]=a[i]*3;
    }
    memset(dp,-1,sizeof(dp));
    bool flag=false;
    for(int i=1;!flag&&i<up[0];i++) //枚举前两个，搜索后面的
        for(int j=1;j<up[1]-i;j++)
        {
            if(i+j<down[0]) continue;
            b[0]=i;
            b[1]=j;
            if(dfs(2)==1) //可行
            {
                flag=true;
                break;
            }
        }
    for(int i=0;i<n-1;i++)
        printf("%d ",b[i]);
    printf("%d\n",b[n-1]);
    return 0;
}

//差分约束
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
typedef long long ll;
const int N=300+10;
const int M=2000+10;
struct Edge
{
    int to,w,nxt;
}edge[M];
int tot,first[N];
void addedge(int u,int v,int w)
{
    edge[tot].to=v;
    edge[tot].w=w;
    edge[tot].nxt=first[u];
    first[u]=tot++;
}
void init()
{
    tot=0;
    memset(first,-1,sizeof(first));
}
int dist[N];
bool vis[N];
void spfa(int n)
{
    queue<int> q;
    for(int i=0;i<=n;i++)
    {
        q.push(i);
        vis[i]=true;
        dist[i]=0;
    }
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        vis[u]=false;
        for(int i=first[u];i!=-1;i=edge[i].nxt)
        {
            int v=edge[i].to;
            if(dist[v]<dist[u]+edge[i].w)
            {
                dist[v]=dist[u]+edge[i].w;
                if(!vis[v])
                {
                    vis[v]=true;
                    q.push(v);
                }
            }
        }
    }
}
int a[N];
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    init();
    
    addedge(0,2,2*a[1]);
    addedge(2,0,-2*a[1]-1);
    for(int i=2;i<n;i++)
    {
        addedge(i-2,i+1,3*a[i]);
        addedge(i+1,i-2,-3*a[i]-2);
    }
    addedge(n-2,n,2*a[n]);
    addedge(n,n-2,-2*a[n]-1);
    
    for(int i=1;i<=n;i++)
        addedge(i-1,i,1);
    spfa(n);
    for(int i=1;i<=n;i++)
        a[i]=dist[i]-dist[i-1];
    for(int i=1;i<n;i++)
        printf("%d ",a[i]);
    printf("%d\n",a[n]);
    return 0;
}