Codeforces Round #505 (rated, Div. 1 + Div. 2) D. Recovering BST(区间dp/bitset优化)

题意:给定n个点,若是两个点的点权的gcd==1,那么这两个点之间可以连一条边,问是否可以构建出二叉搜索树.

区间dp,记忆化搜索,dp[i][j][fa]表示[i,j]区间的数可以构建出合法的二叉搜索树,并连接到父节点fa上

要注意的是这种做法所需的空间卡的比较紧,bool都卡不过去,我们需要用bitset优化内存

#include<bits/stdc++.h>
using namespace std;
bitset<710> vis[710][710],dp[710][710];
int a[710];
int solve(int l,int r,int fa)
{
    if(vis[l][r][fa])return dp[l][r][fa];
    vis[l][r][fa]=1;
    if(l==r&&__gcd(a[l],a[fa])!=1||l>r) return dp[l][r][fa]=1;
    for(int i=l; i<=r; i++)
    {
        if(__gcd(a[i],a[fa])!=1)
        {
            if(solve(l,i-1,i)&&solve(i+1,r,i)) return dp[l][r][fa]=1;
        }
    }
    return dp[l][r][fa]=0;
}
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1; i<=n; i++) scanf("%d",&a[i]);
    if(solve(1,n,0)) printf("Yes\n");
    else printf("No\n");
}