week16 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);}

Input

输入第一行一个t，表示数据组数。
对于每组数据，第一行输入一个n，表示数的个数
接下来一行有n个数$a_i$，输入保证是升序的。

Output

每组数据输出一行，如果能够造出来满足题目描述的树，输出Yes，否则输出No。
无行末空格。

Sample Input 1

1
6
3 6 9 18 36 108

Sample Output 1

Yes

Sample Input 2

2
2
7 17
9
4 8 10 12 15 18 33 44 81

Sample Output 2

No
Yes

Hint

样例1可构造如下图

数据组成
给出的数为上限。

我的思路：
这道题是要求构建一棵二叉搜索树，要求左右节点的最小公因数大于1。这道题可以利用动态规划的思想来解决。我们设置三个数组，分别记录两个数是否可连接dp[i][j]（即他们最小公约数大于1）、L[i][j]（从j（大）可以延伸到i（小））、R[i][j])(从i(小)可以延伸到j（大）)。首先，我们根据输入的数，初始化数组dp，再将L[i][i]、R[i][i]初始化为1,，其余初始化为0。我们枚举长度，即从长为1开始，知道长度为n,其中在每次列举长度时，再在其内部枚举左端点，并根据左端点和长度找到右端点，在左右端点之间进行选值，从而进行L,R数组的更新。更新的策略为如果区间内的数字k，可以满足向左可以延伸到区间最左端，向右可以延伸到区间最右端，则可以判断l-1和r+1是否与k相关联，如果满足，则对应延伸区间可以进行更新（即若l-1满足关联关系，则r向左可延伸到l-1,即L[l-1][r];同理，若r+1满足关联关系，则R[l][r+1]=1）。通过如此更新，不断用小区间去合并成大区间，可以得到最后的L、R关系数组。最后，只要对于输入的数进行枚举，如果可以找到一个数，这个数向左可以延伸到最左端，向右可以延伸到最右端，则可以建立满足题目要求的二叉搜索树。反之，则不能找到。

我的总结：
这道题考查的是动态规划。动态规划对于相关数组关系的递推要求比较高，一定要考虑清楚初始状态，已经递推的过程中相关数组的更新。

我的代码：

#include<cstdio>
#include<cstring>
using namespace std;
int t,n,a[705];
bool L[705][705],R[705][705],dp[705][705];
int gcd(int a,int b){return b == 0 ? a : gcd(b,a%b);}
int main()
{
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d",&n);
        memset(L,0,sizeof(L));
        memset(R,0,sizeof(R));
        memset(dp,0,sizeof(dp));
        for(int i=1;i<=n;i++) 
		{
			scanf("%d",&a[i]);
			L[i][i]=R[i][i]=1;
		}
        for(int i=1;i<=n;i++)
		{
            for(int j=i+1;j<=n;j++)
			{
                if(gcd(a[i],a[j])!=1) dp[i][j]=dp[j][i]=1;
            }
        }
        for(int ll=1;ll<=n;ll++)
		{
            for(int l=1;l+ll-1<=n;l++)
			{
                int r=l+ll-1;
                for(int k=l;k<=r;k++)
				{
                    if(L[l][k]&&R[k][r])
					{
                        if(dp[l-1][k]) R[l-1][r]=1;
                        if(dp[r+1][k]) L[l][r+1]=1;
                    }
                }
            }
        }
        bool flag=false;
        for(int i=1;i<=n;i++)
		{
            if(L[1][i]&&R[i][n])
			{
                flag=true;
                break;
            }
        }
        if(flag) printf("Yes");
        else printf("No");
        if(t!=0) printf("\n");
    }
	return 0;
}