题意:给出n个升序排列的数,求是否能用这n个数组成一颗二叉排序树。
思路:
考场上——看标题,BST!看了题面后,我zz的想,这不就是裸的图论题么…然后就一顿敲,交上去一遍AC,心里美滋滋。结果正式测评就在51个点WA了。
其实正解是dp。
令L[i][j]表示以i为根,i的左子树可到j 这样的bst是否存在。
同理R[i][j]表示以i为根,i的右子树可到j 这样的bst是否存在。
然后枚举一个中间变量进行转移。
比如说L的转移:其中g[i][p]为true是为了判断p可与i相连,而L[p][k]即是指p的左儿子可扩展到k。由于是BST,所以i+1一定是i的祖先或在i的右子树中。R[i+1][p]为true指i+1可扩展到p,也就是指i+1是i的祖先。 所以可以看出,p是i的父亲,i+1为i的祖先时,i只有左子树,所以可以利用p的左子树进行转移。
最后,如果存在一个i使得L[i][n]==R[1][i]==1,则说明存在一颗以i为根的bst,否则不存在。
代码:
#include<bits/stdc++.h>
using namespace std;
#define maxn 700
#define rep(i,x,y) for(int i=x;i<=y;i++)
int n;
int a[maxn+5];
int g[maxn+5][maxn+5];
bool L[maxn+5][maxn+5],R[maxn+5][maxn+5];
int GCD(int x,int y) {
if(y==0) return x;
return GCD(y,x%y);
}
int main() {
scanf("%d",&n);
rep(i,1,n) {
scanf("%d",&a[i]);
}
rep(i,1,n) {
rep(j,1,n) {
g[i][j]=GCD(a[i],a[j]);
if(g[i][j]==1) g[i][j]=0;
else g[i][j]=1;
}
}
rep(i,1,n) L[i][i]=R[i][i]=1;
rep(j,1,n-1) rep(i,1,n-j) {
int k=i+j;
rep(p,i+1,k) L[i][k]|=(g[i][p]&L[p][k]&R[i+1][p]);
rep(p,i,k-1) R[i][k]|=(g[p][k]&L[p][k-1]&R[i][p]);
}
rep(i,1,n) {
if(L[i][n]&&R[1][i]) {
printf("Yes");
return 0;
}
}
printf("No");
return 0;
}