【ACWing】170. 加成序列

题目地址：

https://www.acwing.com/problem/content/172/

满足如下条件的序列$X$（序列中元素被标号为$1,2,3,\dots ,m$）被称为“加成序列”：
1、$X[1]=1$
2、$X[m]=n$
3、$X[1]<X[2]<\dots <X[m-1]<X[m]$
对于每个$k$（$2\le k\le m$）都存在两个整数$i$和$j$（$1\le i,j\le k-1$，$i$和$j$可相等），使得$X[k]=X[i]+X[j]$。你的任务是：给定一个整数$n$，找出符合上述条件的长度$m$最小的“加成序列“。如果有多个满足要求的答案，只需要找出任意一个可行解。

输入格式：
输入包含多组测试用例。每组测试用例占据一行，包含一个整数$n$。当输入为单行的$0$时，表示输入结束。

输出格式：
对于每个测试用例，输出一个满足需求的整数序列，数字之间用空格隔开。每个输出占一行。

数据范围：
$1\le n\le 100$

思路是迭代加深搜索。直接枚举每个$X[k]$是由哪两个$X[i]$和$X[j]$的和算出来的。我们发现，如果直接暴力DFS的话，很容易搜到非常深的节点（因为可以与$X[1]=1$做和操作，总能搜到$n$），但是这样搜索的话，大部分时间都会被浪费在较深的节点上，而非常有可能答案是在很浅的节点上的。所以迭代加深就是个很好的做法。每次DFS的时候都先预设一个最大深度，如果搜到了比最大深度更深的位置了就不继续搜了，而是直接假设无解返回。每次将最大深度加$1$，直到搜到答案为止。代码如下：

#include <iostream>
using namespace std;

const int N = 110;
int n;
int a[N];

bool dfs(int depth, int max_depth) {
    if (depth > max_depth) return false;

    if (a[depth - 1] == n) return true;

	// st[i]是记录是否枚举过a[depth] = i的情况
    bool st[N] = {0};
    // 从大到小枚举a[depth]，更加容易接近答案
    for (int i = depth - 1; i >= 0; i--) 
        for (int j = i; j >= 0; j--) {
            int s = a[i] + a[j];
            if (s > n || s <= a[depth - 1] || st[s]) continue;

            st[s] = true;
            a[depth] = s;
            if (dfs(depth + 1, max_depth)) return true;
        }

    return false;
}

int main() {
    a[0] = 1;
    while (cin >> n, n) {
    	// 根节点深度是0，接下来从深度1开始搜，只要没搜到答案，就增加最大深度，直到搜到答案为止
        int max_depth = 1;
        while (!dfs(1, max_depth)) 
            max_depth++;
        
        for (int i = 0; i < max_depth; i++) 
            cout << a[i] << ' ';
        
        cout << endl;
    }

    return 0;
}

时间复杂度指数级，空间$O(n)$。