题意
在瑞神大战宇宙射线中我们了解到了宇宙狗的厉害之处,虽然宇宙狗凶神恶煞,但是宇宙狗有一 个很可爱的女朋友。
最近,他的女朋友得到了一些数,同时,她还很喜欢树,所以她打算把得到的数拼成一颗树。
这一天,她快拼完了,同时她和好友相约假期出去玩。贪吃的宇宙狗不小心把树的树枝都吃掉 了。所以恐惧包围了宇宙狗,他现在要恢复整棵树,但是它只知道这棵树是一颗二叉搜索树,同 时任意树边相连的两个节点的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
样例1的构造解释:
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/02d5d3c8da0d435055bc62d08e31b431.png)
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/9eac00063f500e0a982a217a96148130.png)
思路:
一开始写题的时候意识到了是一个区间dp的动态规划问题,当时的想法就是枚举每一个点作为根节点,然后这个点的左边和右边又继续这样枚举子树的跟,并在这个过程中进行判断是是否合法。但是动态规划还是难得。不大会。于是就用暴力构造二叉搜索树想来骗点分。前面五个点都只有5个数,但不知到为啥,好像就过了一个点。
正式来到动态规划,现有一个数组f[i][j]记录i到j能不能构造成这样得一个二叉搜索树。然后有f2的数组记录每两个数之间的状态是否符合题意,即是不是gcd是不是大于1,其中f[i][j]的转移方程为f[i][j]=f[i][k]&&f[k][j],即当【i,k】和【k,j】都是一个合法的二叉搜索树的时候,其合并起来的也是一个合法的二叉搜索树。然后开始枚举区间,即从小区间开始,区间的长度分别为1个点到n个点。然后可以根据区间长度枚举这个区间的起点和终点。并且在其中枚举根节点,在此处的时候即用到转移方程来进行更新。同时可以用f2数组来判断更新f[i-1][j]和f[i][k+1]。最终输出来f[1][n]。**注意:**在这样做后发现第六个点错了。后来问了同学,了解只使用一个f数组没法记录下所有状态,即虽然你知道f[i][j]可以是一个合法的二叉搜索树,但是在这里你不知道它作为一个合法的二叉搜索树的时候,他的根节点和左右子树分别是什么,故有信息缺失。所以在这里的时候又增加了l和r数组,即l[i][j]表示以j为跟,i,j为左子树的合法的树。r[i][j]则是表示以i为根,i,j为右子树的合法的树。加入这些信息后就可以了。
代码:
#include <iostream>
#include <string>
#include <cstring>
#include <cstdio>
#include <algorithm>
using namespace std;
int t,n,a[800];
bool f[800][800],f2[800][800];
bool l[800][800],r[800][800];
int gcd(int x, int y) { return y == 0 ? x : gcd(y, x % y); }
int main(){
cin >> t;
while (t--){
memset(f, 0, sizeof(f));
memset(f2, 0, sizeof(f2));
memset(l, 0, sizeof(l));
memset(r, 0, sizeof(r));
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 (i!=j && gcd(a[i], a[j]) > 1){
f2[i][j] = true;
}
}
}
for (int len = 0; len < n; len++){
for (int q = 1; q + len <= n; q++){
int z = q + len;
for (int root = q; root <= z; root++)
if (l[q][root] && r[root][z]){
f[q][z] = 1;
if (f2[q - 1][root]) r[q - 1][z] = 1;
if (f2[root][z + 1]) l[q][z + 1] = 1;
}
}
}
if (f[1][n]) cout << "Yes" << endl;
else cout << "No" << endl;
}
return 0;
}