宇宙狗的危机
在瑞神大战宇宙射线中我们了解到了宇宙狗的厉害之处,虽然宇宙狗凶神恶煞,但是宇宙狗有一 个很可爱的女朋友。
最近,他的女朋友得到了一些数,同时,她还很喜欢树,所以她打算把得到的数拼成一颗树。
这一天,她快拼完了,同时她和好友相约假期出去玩。贪吃的宇宙狗不小心把树的树枝都吃掉 了。所以恐惧包围了宇宙狗,他现在要恢复整棵树,但是它只知道这棵树是一颗二叉搜索树,同 时任意树边相连的两个节点的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 a_i ai,输入保证是升序的。
输出描述
每组数据输出一行,如果能够造出来满足题目描述的树,输出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;
}
}