题意:
在瑞神大战宇宙射线中我们了解到了宇宙狗的厉害之处,虽然宇宙狗凶神恶煞,但是宇宙狗有一
个很可爱的女朋友。
最近,他的女朋友得到了一些数,同时,她还很喜欢树,所以她打算把得到的数拼成一颗树。
这一天,她快拼完了,同时她和好友相约假期出去玩。贪吃的宇宙狗不小心把树的树枝都吃掉
了。所以恐惧包围了宇宙狗,他现在要恢复整棵树,但是它只知道这棵树是一颗二叉搜索树,同
时任意树边相连的两个节点的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)符合最优子结构,可用dp去做
2)定义:
dp[i][j][0]表示第 i 个结点到第 j 个结点作为某一棵二叉搜索树的左子树是否可行,dp[i][j][1]表示作为右子树是否可行。
3)状态转移:
分别枚举二叉搜索树的起点和终点,以及根节点的位置。这里根据子结构->结果时,需要注意枚举的范围,
终点j从1->n,起点i从j->1,保证递推时子结构已经被更新
反思:
注意对于多次重复运算两个节点的gcd,可先用二维数组记录结果
代码:
#include<iostream>
#include<algorithm>
#include<string>
#include<cstring>
using namespace std;
int gcd(int a,int b)
{
return b == 0 ? a : gcd(b,a%b);
}
int a[710],res[710][710],dp[710][710][2];
int ggcd[710][710];
int n;
int main()
{
int t;cin>>t;
while(t--)
{
cin>>n;
memset(dp,0,sizeof(dp));
memset(res,0,sizeof(res));
memset(ggcd,0,sizeof(ggcd));
for(int i=1;i<=n;i++)
{
cin>>a[i];
dp[i][i][0]=dp[i][i][1]=1;
}
for(int i=1;i<=n;i++)
{
for(int j=i+1;j<=n;j++)
{
if(gcd(a[i],a[j])>1)
ggcd[i][j]=ggcd[j][i]=1;
}
}
for(int j=1;j<=n;j++)
{
for(int i=j;i>=1;i--)
{
for(int k=i;k<=j;k++)
{
if(dp[k][i][0]&&dp[k][j][1])
{
res[i][j]=1;
if(ggcd[i-1][k]) dp[i-1][j][1]=1;
if(ggcd[j+1][k]) dp[j+1][i][0]=1;
}
}
}
}
if(t!=0)
{
if(res[1][n])
cout<<"Yes"<<endl;
else cout<<"No"<<endl;
}
else
{
if(res[1][n])
cout<<"Yes";
else cout<<"No";
}
}
return 0;
}