csp模测4th_T4 - 宇宙狗的危机

T4 - 宇宙狗的危机

---

题目描述

题目思路

这个题目在模测的时候拿到以为真的要写二叉搜索树结构，结果直接就放弃了。之后才了解到这实际就是一个区间dp的问题。这里我们采用动态规划的思想。二维数组L和R分别是用来判断a[j]/a[i]能否向左或者向右连通到a[i]/a[j]。g数组保存a[i]和a[j]的连通情况，用于后面更新L和R数组。dp数组就是判断a[i]和a[j]之间的数是否满足要求。在dp的过程中，我遍历区间长度，确定以每一个点为左端点之后的所有点是否符合要求。如果可以向左可以和左端点连通，向右可以和右端点连通，就符合要求将该区间的dp值设为1。最后判断dp[1][n]即可。

代码实现

#include
#include
#include
#include
using namespace std;
const int N = 710;

int n,t,a[N],dp[N][N],L[N][N],R[N][N],g[N][N];

int gcd(int a,int b){return b == 0 ? a : gcd(b, a % b);}

void init(){
memset(dp,0,sizeof(dp));
memset(L,0,sizeof(L));
memset(R,0,sizeof®);
}

int main(){
cin >> t;
while(t–){
init();
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 = i + 1 ; j <= n; j++)
{
if(gcd(a[i], a[j]) > 1)
g[i][j] = g[j][i] = 1;
else if(gcd(a[i], a[j]) == 1)
g[i][j] = g[j][i] = 0;
}
}

	for(int i = 1; i <= n; i++)
	{
		for(int j = 1; j <= n - i + 1; j++)
		{
			int r = j + i - 1;
			for(int k = 0; k < i; k++)
			{
				if(L[j][j + k] == 1 && R[j + k][r] == 1){
					dp[j][r] = 1;
					if(g[j - 1][j + k]){
						R[j - 1][r] = 1;
					}
					if(g[j + k][r + 1]){
						L[j][r + 1] = 1;
					}
				}	
			}
		}
	}
	if(!dp[1][n]) cout << "No" << endl;
	else cout << "Yes" << endl;
}

}