CSP 201809-4 再卖菜

dfs的时候注意记忆化搜索，不然会T掉只有80分。
记忆化搜索的时候保存一个数组 $f_{i,j,k}$ 保存填第 $i$ 天时，前两天所填的数字有没有被搜索过

#include 
#include 

const int MAXN = 300 + 5;

int n, cnt;
int a[MAXN], b[MAXN];
bool f[MAXN][MAXN][MAXN];

bool check() {
    for (int i = 1; i <= n; i++) {
        int tmp;
        if (i == 1) tmp = (a[i] + a[i + 1]) / 2;
        else if (i == n) tmp = (a[i] + a[i - 1]) / 2;
        else tmp = (a[i] + a[i + 1] + a[i - 1]) / 3;
        if (tmp != b[i]) return false;
    }
    return true;
}

void dfs(int cur) {
    if (cur > n) {
        if (check()) {
            for (int i = 1; i <= n; i++)
                printf("%d ", a[i]);
            exit(0);
        }
        else return;
    }

    if (cur > 2) {
        if (f[cur][a[cur - 2]][a[cur - 1]]) return;
        f[cur][a[cur - 2]][a[cur - 1]] = true;
    }

    if (cur == 2) {
        for (int i = 2 * b[1] - a[1]; (a[1] + i) / 2 == b[1]; i++) {
            if (i > 0) {
                a[cur] = i;
                dfs(cur + 1);
            }
        }
    }
    else if (cur == n) {
        for (int i = 2 * b[n] - a[n - 1]; (a[n - 1] + i) / 2 == b[n]; i++) {
            if (i > 0) {
                a[cur] = i;
                dfs(cur + 1);
            }
        }
    }
    else {
        for (int i = 3 * b[cur - 1] - a[cur - 2] - a[cur - 1]; (a[cur - 2] + a[cur - 1] + i) / 3 == b[cur - 1]; i++) {
            if (i > 0) {
                a[cur] = i;
                dfs(cur + 1);
            }
        }
    }
}

int main()
{
    scanf("%d", &n);
    for (int i = 1; i <= n; i++) 
        scanf("%d", &b[i]);

    for (int i = 1; ; i++) {
        a[1] = i;
        dfs(2);
    }
    return 0;
}