思路:
树形结构,一般我们会想到用递归的方法,具有子结构特征,可以转换为dp的形式。
从整体上看,给定一个区间(从小区间扩到大区间),二叉搜索树需要遍历区间每个数当做根,因此内部有一个从左端到右端的循环。
子结构:右子树和左子树
当右子树(k+1~ r)任意组合 && 左子树(l~ k-1)任意组合能够找到一种情况合法的时候,这个大树 (l ~ r)就存在一种情况合法,就可以扩展子树。
因此定义状态:
dpL[k][l] :以 k为根, l~ k-1作为左子树能否存在一种组合合法(true)
dpR[k][r] : 以k为 根 , k+1~r 作为右子树 是否存在一种组合合法
初始状态:dpL[i][i] 、dpR[i][i]为true
状态转移:如果l -1可以与k相连(最大公约数大于1),且dpL[k][l] 与dpR[k][r]为true即整个大树(l~r)合法。于是dpR[l-1][r] 找到了一种排列合法,设置为true.
同理:如果r+1可以与k相连,且dpL[k][l] 与dpR[k][r]为true,则dpL[r+1][l]设置为true.
可行性:
因为k是l ~ r的遍历,因此可以一个区间包括所有情况。
因为是从小区间到大区间遍历,区间左端点从n开始逆序遍历,右端点顺序遍历。
每次用到dpL[k][l]和dpR[k][r]的时候,前面就已经求出来了
最后检查区间[1,n],以区间内某个数为根,检查它的左右子树是否合法,存在一个合法,则整体合法(和子树合法性检查一样,都是k遍历区间)。
代码
#include <iostream>
#include <cstring>
using namespace std;
int n;
int a[710];
bool flag[710][710];//判断各点直接能否直接相连
bool dpL[710][710];//定义状态dpL[k][l]以 k为根, l~ k-1作为左子树 符合条件
bool dpR[710][710];//状态dpR[k][r] 以k为 根 , k+1~r 作为右子树 符合条件
//扩展状态: flag[l-1][k] 成立且 前面以k为根的左右子树都已经形成了,则可扩展为 dpR[l-1][r]符合要求
//扩展状态: flag[k][r+1]成立 且 前面以k为根左右子树符和要求 则 dpL[r+1][l]符和要求
bool dp_solve()//由小区间扩展 为大区间 作为区间左右端点 每个点都能被扩展一格 由小及大 能遍历所有区间
{
for(int l = n; l >= 1; l--)//逆序决定了由小及大 外面遍历r正序 ,里层遍历l逆序??
{
for(int r = 1; r <= n; r++)
{
for(int k = l; k <= r; k++)
{
if(dpL[k][l] && dpR[k][r])//dp子结构在用到之前 一定已经求出来了
{
if(flag[l-1][k])
dpR[l-1][r] = 1;
if(flag[r+1][k])
dpL[r+1][l] = 1;
}
}
}
}
//寻找区间长度为n的区间 (以某个当做根) 是否左右子树都成立 -->有1个就可以
for(int i = 1; i <= n; i++)
{
if(dpL[i][1] && dpR[i][n])
return true;
}
return false;
}
int gcd(int a, int b)
{
return b == 0 ? a : gcd(b, a%b);
}
int main()
{
int T;
cin >> T;
while(T--)
{
scanf("%d", &n);
memset(dpL, 0, sizeof(dpL));
memset(dpR, 0, sizeof(dpR));
memset(flag, 0, sizeof(flag));
for(int i = 1; i <= n; i++)
{
scanf("%d", &a[i]);//输入升序
dpL[i][i] = 1;
dpR[i][i] = 1;
}
for(int i = 1; i <= n; i++)
{
for(int j = i; j <= n; j++)
{
if(gcd(a[i], a[j]) > 1)
flag[i][j] = flag[j][i] = 1;
else flag[i][j] = flag[j][i] = 0;
}
}
if(dp_solve())
cout <<"Yes" <<endl;
else cout << "No" <<endl;
}
return 0;
}
总结:
区间dp完全没有看出来,模测的时候竟然以为是和图论有关T4的题,要构造树的结构,把所有二叉搜索树全部构造出来,再遍历二叉搜索树检查,这就太难了,现在想了想也不会。
对了,检查两点是否可连,把求最大公约数的结构保存在一个二维数组中,否则可能会超时。
参考博客:
https://blog.csdn.net/weixin_44034698/article/details/106545010