程序设计思维与实践 CSP-M4 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$，输入保证是升序的。
输出描述
每组数据输出一行，如果能够造出来满足题目描述的树，输出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

思路

这题我当时在做的时候想尽各种方法找树的构建成功的关键与数字的关系，但是找了好久都没有找到一个可以使用的准确简洁的关系，后来经同学提点才直到这题原来用区间dp做…这是真的想不到要这样做。区间dp的思路为：设dp[i][j]表示原序列 i - j 是否能构造成一棵树，设 l[i][j] 为以i为根，原序列 j - i 是否可以构造成i的左子树，r[i][j]为以i为根，原序列 i - j 是否可以构造成i的右子树，于是就有
dp[i][j] |= ( l[k][i] && r[k][j] ）
同时，若dp[i][j]==1
l[j+1][i]∣=gcd(a[j+1],a[k]),(i≤k≤j)（k为dp[i][j]构造出来的树的根节点）
r[i−1][j]∣=gcd(a[i−1],a[k]),(i≤k≤j)
很容易理解，在 i - j 可构建为以k为根的树的条件下，只要a[j+1]或者a[i-1]能够与k相连，即有大于1的最大公约数，即可确保l[j+1][i]或r[i-1][j]为1。
初始化的话因为一个节点可以作为以自己为根节点的左子树右子树及树，因此对所有dp[i][i]，l[i][i]，r[i][i]均设为1即可。

代码

#include <iostream>
#include <stdio.h>
#include <string.h>
using namespace std;
int gcd(int a,int b)
{
	return (b== 0 ? a : gcd(b,a%b));	
}
int l[750][750];int r[750][750];int dp[750][750];int gc[750][750];
int s[750];
int n;
void init()
{
	memset(l,0,sizeof(l));
    memset(r,0,sizeof(r));
    memset(dp,0,sizeof(dp));
    for(int i=1;i<=730;i++)
    {
    	l[i][i]=1;
    	r[i][i]=1;
    	dp[i][i]=1;
	}
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=n;j++)
		{
			gc[i][j]=gcd(s[i],s[j])>1 ? 1 : 0;
		}
	}
}
int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d",&n);
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&s[i]);
		}
		init();
		for(int len=1;len<=n;len++)
		{
			for(int i=1;i<=n-len+1;i++)
			{
				int j=i+len-1;
				for(int k=i;k<=j;k++)
				{
					if(l[k][i]&&r[k][j])
					{
						dp[i][j]=1;
						l[j+1][i]|=gc[k][j+1];
						r[i-1][j]|=gc[k][i-1]; 
					}
				}
			}
		}
		if(dp[1][n])
			printf("Yes\n");
		else
			printf("No\n");
	}
}