宇宙狗的危机(M4)

题目

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

题目思路

本来以为这是一道树论题目,在考试的时候选择用暴力求解,但是由于一点小失误导致提交上去编译错误。但是,经过仔细分析以及别人博客的提示发现,原来这个题用区间dp这么简单!!
首先,题目中保证数据升序,且树是一颗二叉搜索树,每一个节点的右子树的值都比该节点大,左子树都比该节点小。且可以发现满足要求的树的左右子树同样也是二叉搜索树。因此,r[ ][ ] 和 l[ ][ ] 两个二维数组分别表示左右子树是否满足条件,r[ i ][ j ]表示[ i+1,j ]是否是一个二叉搜索树(可以作为一个右子树),l[ i ][ j ]同理表示[ i , j-1]是一棵二叉搜索树(可以作为一个左子树)。
转移方程:

//向left-1方向转移,当前子树是它的右子树
if(dp[i-1][root]) r[i-1][j]=1;
//向right+1方向转移,当前子树是他的左子树
if(dp[j+1][root]) l[i][j+1]=1;

当序号1和序号n点相连时,证明可以构建一颗二叉搜索树,输出Yes;否则输出No。

代码实现

#include<bits/stdc++.h>
using namespace std;
int t,n;
int a[703],l[703][703],r[703][703],dp[703][703];
int gcd(int a,int b){return b == 0 ? a : gcd(b,a%b);}
int main()
{
	cin>>t;
	while(t--)
	{
		cin>>n;
		memset(dp, 0, sizeof(dp));
	        memset(l, 0, sizeof(l));
	        memset(r, 0, sizeof(r));
		bool flag=false;
		for(int i=1;i<=n;i++) cin>>a[i];
		for(int i=1;i<=n;i++)
		{
			for(int j=i;j<=n;j++)
			{
				if(gcd(a[i],a[j])>1) dp[i][j]=1,dp[j][i]=1;
			}
		}
		for(int i=1;i<=n;i++) l[i][i]=1,r[i][i]=1;
		for(int i=n;i>0;i--)
		{
			for(int j=i;j<=n;j++)
			{
				for(int root=i;root<=j;root++)
				{
					if(l[i][root] && r[root][j])
					{
						if(i==1 && j==n){
							flag=true;
							break;
						}
						if(dp[i-1][root]) r[i-1][j]=1;
						if(dp[j+1][root]) l[i][j+1]=1;
					}
				}
				if(flag) break; 
			}
			if(flag) break; 
		}
		if(flag) cout<<"Yes"<<endl;
		else cout<<"No"<<endl;
	}
	return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值