题解-CF429C Guess the Tree

题面

CF429C Guess the Tree

给一个长度为 (n) 的数组 (a_i)，问是否有一棵树，每个节点要么是叶子要么至少有两个儿子，而且 (i) 号点的子树大小是 (a_i)。

数据范围：(1le nle 24)。

---

题解

发现 (n) 很小，想到可以状压。

设叶子节点有 (ln) 个，所以中间节点有 (mn=n-ln) 个。

由于“每个节点要么是叶子要么至少有两个儿子”，所以 (lngelceilfrac n2 ceil)，(mnle n-lceilfrac n2 ceil le 11)。

所以可以先特判 (2mnge n) 的情况答案为 NO。

然后剩下的可以 dp，设 (f_{t,s,i})：

(t) 表示是一棵森林还是一个子树（为了对付“每个节点要么是叶子要么至少有两个儿子”，如果是子树 (t=0)，否则 (t=1)）。

(s) 表示包含的中间节点集合，(s< 2^{mn}le 2048)。

(i) 表示包含的叶子节点个树，因为叶子节点都是一样的，所以这样可以优化状压。

值表示是否存在这样的森林，如果存在 (=1)，否则 (=0)。

考虑怎么转移：

1. 一棵森林（子树）和另一棵森林（子树）合并成新的森林。

2. 一棵森林上面加一个 (a=) 森林大小 (+1) 的节点成为一棵子树（怎么求一棵森林的大小？其实就是 ({ m popcount}(s)+i) 啦）。

然后就剩初始化的问题了，因为 (i) 这维就像一个背包，而且因为 (s) 这一维保证不会有重点，所以可以在 dp 中用无限背包的方式，这样就只需要初始化 (f_{0,0,1}=1) 了。

时间复杂度 (Theta(ln^2 3^{mn}))，细节看代码。

---

代码

#include <bits/stdc++.h>
using namespace std;

//Start
typedef long long ll;
typedef double db;
#define mp(a,b) make_pair((a),(b))
#define x first
#define y second
#define bg begin()
#define ed end()
#define sz(a) int((a).size())
#define pb(a) push_back(a)
#define R(i,a,b) for(int i=(a),i##E=(b);i<i##E;i++)
#define L(i,a,b) for(int i=(b)-1,i##E=(a)-1;i>i##E;i--)
const int iinf=0x3f3f3f3f;
const ll linf=0x3f3f3f3f3f3f3f3f;

//Data
const int N=24,mN=11;
int n,sn,ln,mn,a[N],b[1<<mN];
int f[2][1<<mN][N+1]; // 森林还是子树，中间节点集，叶子节点数
bool get(int s,int i){ // 返回 f[1][s][i] 的值
    for(int su=s;su;su=s&(su-1))if(!(su&(s^su)))R(j,0,i+1)
        if((f[0][su][j]||f[1][su][j])&&(f[0][s^su][i-j]||f[1][s^su][i-j])) return true;
    R(j,0,i+1)if((f[0][0][j]||f[1][0][j])&&(f[0][s][i-j]||f[1][s][i-j])) return true; // 因为 su!=0，补上循环中缺少的 su=0
    return false;
}

//Main
int main(){
    ios::sync_with_stdio(0);
    cin.tie(0),cout.tie(0);
    cin>>n; R(i,0,n) cin>>a[i]; sort(a,a+n,greater<int>());
    R(i,0,n)if(a[i]>1) mn++; ln=n-mn,sn=1<<mn,sort(a,a+mn); 
    if(mn*2>=n) cout<<"NO
",exit(0);
    R(i,1,sn) b[i]=b[i>>1]+(i&1);
    f[0][0][1]=true;
    R(s,0,sn)R(i,0,ln+1)if(get(s,i)){
        f[1][s][i]=true;
        R(t,0,mn)if(!(s&(1<<t))&&a[t]==b[s]+i+1) // 加新的 a= 森林大小+1 的节点形成子树
            f[0][s^(1<<t)][i]=true;
    }
    if(f[0][sn-1][ln]) cout<<"YES
";
    else cout<<"NO
";
    return 0;
}

---

祝大家学习愉快！