C - 宇宙狗的危机

题意

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

一开始写题的时候意识到了是一个区间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;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值