M4-宇宙狗的危机

题目

分析

考虑区间dp。
left [ i ] [ root ]代表以 root 为根节点，从左到 i 是否可以作为左子树；
right [ root ] [ j ]表示以root 为根节点，从右到 j 是否可以作为右子树；
f2 [ i ] [ j ]表示的是a[ i ] a[ j ]的gcd值是否>1;
f1 [ i ] [ j ]表示的是i 到 j 能否构成二叉搜索树；
状态表示好之后，如何转移呢？
1、明显，当left[i][root]==1和right[root][j]==1时，f1[ i ][ j ]=1；以root为根。
即：

if (lft[k][root] && rig[root][k+i]){
		f1[k][k+i] = 1;

2、在f1 [ i ][ j ]是true的情况下：每次向左转移一个：
if ( f2 [ i-1 ][ root ] ) right [ i-1 ] [ j ] = 1;
如果a[k-1],a[root]的gcd>1，那么代表a[i-1]可以和a[root]相连，也就是i-1的孩子是root，那么i-1到j之间可以构成一个二叉搜索树作为左子树。
同理每在向右转移一个。
即：

if (f2[k-1][root]) rig[k-1][k+i] = 1;
if (f2[root][k+i+1]) lft[k][k+i+1] = 1;

代码

#include <iostream>
#include <string>
#include <cstring>
#include <algorithm>
using namespace std;
int t,n,a[800];
bool f1[800][800],f2[800][800],lft[800][800],rig[800][800];
void ini(){//初始化
	memset(f1, 0, sizeof(f1));
	memset(f2, 0, sizeof(f2));
	memset(lft, 0, sizeof(lft));
	memset(rig, 0, sizeof(rig));
}
//计算gcd的函数
int gcd(int x, int y) { return y == 0 ? x : gcd(y, x % y); }

int main(){
	cin >> t;
	while (t--){//t组数据
		ini();
		cin >> n;//n个数
		for(int i=1;i<=n;i++){
			cin >> a[i];//二叉搜索树的结点值
			lft[i][i] =1; rig[i][i] = 1;//初始化为true
		}
		//初始化f2
		for(int i=1;i<=n;i++){
			for(int j=1;j<=n;j++){
				if (i!=j && gcd(a[i],a[j])>1){
					f2[i][j]=true;//i j之间是可以相连的
				}
			}
		}
		for (int i=0;i<n;i++){
			for (int k=1;k<=n-i;k++){
				for (int root=k;root<=k+i;root++)
					if (lft[k][root] && rig[root][k+i]){
						f1[k][k+i] = 1;
						if (f2[k-1][root]) rig[k-1][k+i] = 1;
						if (f2[root][k+i+1]) lft[k][k+i+1] = 1;
					}
			}
		}
		if (f1[1][n]) cout<<"Yes"<<endl;
		else cout<<"No"<<endl;
	}
	return 0;
}