题目描述
在瑞神大战宇宙射线中我们了解到了宇宙狗的厉害之处,虽然宇宙狗凶神恶煞,但是宇宙狗有一
个很可爱的女朋友。
最近,他的女朋友得到了一些数,同时,她还很喜欢树,所以她打算把得到的数拼成一颗树。
这一天,她快拼完了,同时她和好友相约假期出去玩。贪吃的宇宙狗不小心把树的树枝都吃掉
了。所以恐惧包围了宇宙狗,他现在要恢复整棵树,但是它只知道这棵树是一颗二叉搜索树,同
时任意树边相连的两个节点的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
样例解释v
样例1可构造如下图
数据组成
给出的数为上限。
数据点数 | t | n | a i a_i ai |
---|---|---|---|
1,2,3 | 5 | 15 | 1 0 9 10^9 109 |
4,5,6 | 5 | 35 | 1 0 9 10^9 109 |
7,8,9,10 | 5 | 700 | 1 0 9 10^9 109 |
思路:
给出一些数,若要求两数gcd>1
才能连边,询问他们是否能构成一棵二叉搜索树。
dp;
L[i][j]
表示区间[i,j-1]
作为j
的左儿子是否合法
R[i][j]
表示区间[i+1,j]
作为j
的右儿子是否合法
设k为区间[l,r]
的根,
如果L[l][k]=1且r[k][r]=1
,则该状态合法,可以转移;
同时l==1
且r==n
则说明存在一颗以K为根的BST。
如果k
可以与l-1
相连,说明以l-1
为根,[l,r]
为右儿子的状态合法。
如果k
可以与r+1
相连,说明以r+1
为根,[l,r]
为左儿子的状态合法。
代码:
#include <algorithm>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn = 710;
int n, a[maxn], f[maxn][maxn], L[maxn][maxn], R[maxn][maxn];
int gcd(int a, int b)
{
return b == 0 ? a : gcd(b, a % b);
}
void clear()
{
memset(L, 0, sizeof(L));
memset(R, 0, sizeof(R));
memset(f, 0, sizeof(f));
}
void solve()
{
scanf("%d", &n);
for (int i = 1; i <= n; i++)
{
scanf("%d", &a[i]);
L[i][i] = R[i][i] = 1; //初始化
}
for (int i = 1; i <= n; i++)
{
for (int j = i; 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)
{ //结果
printf("Yes\n");
return;
}
if (f[l - 1][k]) //可以连接 则更新转移状态
R[l - 1][r] = 1;
if (f[k][r + 1]) //可以连接 则更新转移状态
L[l][r + 1] = 1;
}
}
}
}
printf("No\n");
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
clear();
solve();
}
return 0;
}