时间限制 | 空间限制 |
---|---|
5S | 256MB |
题目描述
在瑞神大战宇宙射线中我们了解到了宇宙狗的厉害之处,虽然宇宙狗凶神恶煞,但是宇宙狗有一个很可爱的女朋友。
最近,他的女朋友得到了一些数,同时,她还很喜欢树,所以她打算把得到的数拼成一颗树。
这一天,她快拼完了,同时她和好友相约假期出去玩。贪吃的宇宙狗不小心把树的树枝都吃掉
了。所以恐惧包围了宇宙狗,他现在要恢复整棵树,但是它只知道这棵树是一颗二叉搜索树,同时任意树边相连的两个节点的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可构造如下图
数据组成
给出的数为上限。
数据点数 | 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 |
思路
测试的时候没有注意思考,觉得这就是一个二叉搜索树问题,写了好久,最后也没有写出来,之后与同学讨论了之后,发现,这原来时区间dp,当时就惊了。测试的时候竟然没有想到。
将数据读入后,对每两个数计算一次最大公约数,保存到二维数组g中,然后L[i][j]表示ij-1可作为j的左子树,R[i][j]表示i+1j可作为i的右子树,dp[i][j]表示i~j满足建树条件
转移过程:
if (L[l][k] && R[k][r])满足,则
- dp[l][r]=1;
- if (g[k][l - 1] > 1) R[l - 1][r] = 1;
- if (g[k][r + 1] > 1) L[l][r + 1] = 1;
最后根据dp[1][n]的值来判断是否可以建成树
代码
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn = 710;
int a[maxn], dp[maxn][maxn], g[maxn][maxn], L[maxn][maxn], R[maxn][maxn];
int gcd(int a, int b) { return b == 0 ? a : gcd(b, a % b); }
int main()
{
int t;
scanf("%d", &t);
while (t--)
{
memset(a, 0, sizeof(a));
memset(dp, 0, sizeof(dp));
memset(g, 0, sizeof(g));
memset(L, 0, sizeof(L));
memset(R, 0, sizeof(R));
int n;
scanf("%d", &n);
for (int i = 1; i <= n; i++)
scanf("%d", &a[i]);
for (int i = 1; i <= n; i++)
{
for (int j = 1; j <= n; j++)
g[i][j] = g[j][i] = gcd(a[i], a[j]);
L[i][i] = R[i][i] = 1;
}
for (int i = 1; i <= n; i++)
{
for (int l = 1; l <= n - i + 1; l++)
{
int r = l + i - 1;
for (int k = l; k <= r; k++)
{
if (L[l][k] && R[k][r])
{
dp[l][r] = 1;
if (g[k][l - 1] > 1)
R[l - 1][r] = 1;
if (g[k][r + 1] > 1)
L[l][r + 1] = 1;
}
}
}
}
if (dp[1][n] == 1)
printf("Yes\n");
else
printf("No\n");
}
return 0;
}