CSP模测4 T3 宇宙狗的危机

宇宙狗的危机

在瑞神大战宇宙射线中我们了解到了宇宙狗的厉害之处，虽然宇宙狗凶神恶煞，但是宇宙狗有一 个很可爱的女朋友。
最近，他的女朋友得到了一些数，同时，她还很喜欢树，所以她打算把得到的数拼成一颗树。
这一天，她快拼完了，同时她和好友相约假期出去玩。贪吃的宇宙狗不小心把树的树枝都吃掉 了。所以恐惧包围了宇宙狗，他现在要恢复整棵树，但是它只知道这棵树是一颗二叉搜索树，同 时任意树边相连的两个节点的gcd(greatest common divisor)都超过1。
但是宇宙狗只会发射宇宙射线，他来请求你的帮助，问你能否帮他解决这个问题。
补充知识：
GCD：最大公约数，两个或多个整数共有约数中最大的一个 ，例如8和6的最大公约数是2。
一个简短的用辗转相除法求gcd的例子：

int gcd(int a,int b){return b == 0 ? a : gcd(b,a%b);}

输入描述

输入第一行一个t，表示数据组数。 对于每组数据，第一行输入一个n，表示数的个数 接下来一行有n个数$a_i$，输入保证是升序的。

输出描述

每组数据输出一行，如果能够造出来满足题目描述的树，输出Yes，否则输出No。 无行末空格。

样例输入1

1 
6 
3 6 9 18 36 108

样例输出1

Yes

样例输入2

2 
2 
7 17 
9 
4 8 10 12 15 18 33 44 81

样例输出2

No 
Yes

问题分析

给出n个数，问能否构成一棵二叉搜索树.

L[i][j]代表[i,j-1]是否可作为j的左子树。
R[i][j]表示[i+1,j]是否可以作为i的右子树。
转移条件为：
if (f[l-1][root])R[l-1][r]=1;
if (f[r+1][root])L[l][r+1]=1;
当[l,r]为[1,n]时说明可以构建符合题意的树，如果枚举结束都到不了[1,n]说明不可以构建符合题意的树。

f数组记录两点是否可以相连，即gcd是否大于1.初始化时注意将 L[i][i]和R[i][i]都赋值为1，当点i的左子树或者右子树为空时一定是符合题意的.

代码实现

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int N=710;

int gcd(int a,int b){return b == 0 ? a : gcd(b,a%b);}


int n,a[N],f[N][N],L[N][N],R[N][N];
int bj=0;
int main()
{
    int t;
    ios::sync_with_stdio(false);
    cin>>t;
	while(t--)
	{	
	    bj=0;
		memset(L,0,sizeof(L));
 	 	memset(R,0,sizeof(R));
  		memset(f,0,sizeof(f));
	  	cin>>n;
  		for(int i=1;i<=n;i++) 
  		{
  			cin>>a[i];
			L[i][i]=R[i][i]=1;
		}
		  
  		for(int i=1;i<=n;i++) 
  		{
  			for(int j=1;j<=n;j++)
   	        	if(gcd(a[i],a[j])>1) 
				    f[i][j]=f[j][i]=1;
		  }
  		
  		for(int l=n;l>=1;l--)
  		{
  			for(int r=l;r<=n;r++)
  			{
  				for(int k=l;k<=r;k++)
  				{
  					if(L[l][k]&&R[k][r])
                 	{
	            		if(l==1&&r==n) {bj=1;}
	            		if(f[l-1][k]) R[l-1][r]=1;
	             		if(f[k][r+1]) L[l][r+1]=1;
                	}
				}
			}
		}
		
      	if(bj==1)
		    cout<<"Yes"<<endl;
      	else 
  		    cout<<"No"<<endl;
	}
}