CCF201809-4 再卖菜（100分）DFS

Posted 2021-09-09 海岛Blog

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

问题链接：CCF201809-4 再卖菜
问题简述：（略）
问题分析：用DFS来解决，需要剪枝来避免重复搜索。
程序说明：（略）
参考链接：（略）
题记：原始代码来自仙客传奇团队。

100分的C++语言程序如下：

/* CCF201809-4 再卖菜 */

#include <bits/stdc++.h>

using namespace std;

const int P = 100 + 5;
const int N = 300 + 1;
int n, a[N], b[N], vis[N][P][P];

void dfs(int d)
{
    if (d == n + 1) {
        if (b[n] != (a[n-1] + a[n]) / 2) return;
        cout << a[1];
        for (int i = 2; i <= n; i++)
            cout << " " << a[i];
        cout << endl;
        exit(0);
    } else {
        for (a[d] = 3 * b[d - 1] - a[d - 1] - a[d - 2];
             a[d] <= 3 * b[d - 1] + 2 - a[d - 1] - a[d - 2]; a[d]++)
            if (a[d] > 0 && vis[d][a[d]][a[d - 1]] == 0) {
                vis[d][a[d]][a[d - 1]] = 1;
                dfs(d + 1);
            }
    }
}

int main()
{
    std::ios::sync_with_stdio(false);
    std::cin.tie(NULL);
    cout.tie(NULL);

    cin >> n;
    for (int i = 1; i <= n; i++) cin >> b[i];

    memset(vis, 0, sizeof vis);

    for(a[1] = 1; a[1] <= 2 * b[1]; a[1]++)
        for(a[2] = 2 * b[1] - a[1]; a[2] <= 2 * b[1] + 1 - a[1]; a[2]++)
            if (a[2] > 0) dfs(3);

    return 0;
}