Solution -「CF 1025D」Recovering BST

$\mathcal{Description}$

  Link.

  给定序列 $\{a_n\}$，问是否存在一棵二叉搜索树，使得其中序遍历为 $\{a_n\}$，且相邻接的两点不互素。

  $n\le700$。

$\mathcal{Solution}$

  显然的 $\mathcal O(n^4)$ DP：$f(l,r,i)$ 表示区间 $[l,r]$ 是否能构成以 $i$ 为根的树。

  一个重要的性质：若区间 $[l,r]$ 构成二叉搜索树的一棵完整的子树，则其父亲是 $l-1$ 或 $r+1$。证明显然。

  那么状态可以优化，令 $f(l,r,0/1)$ 表示区间 $[l,r]$ 能否作为 $l-1/r+1$ 的子树，转移：

$$f(l,r,0)=\bigvee_{k=l}^rf(l,k-1,1)\land f(k+1,j,0)\land \gcd(a_k,a_{l-1})\not=1\\f(l,r,1)=\bigvee_{k=l}^rf(l,k-1,1)\land f(k+1,j,0)\land \gcd(a_k,a_{r+1})\not=1$$

  当 $l=1$ 或 $r=n$，认为逻辑与的最后一项为真即可。复杂度 $\mathcal O(n^3)$。

$\mathcal{Code}$

#include <cstdio>

const int MAXN = 700;
int n, a[MAXN + 5];
bool f[MAXN + 5][MAXN + 5][2];

inline int gcd ( const int a, const int b ) { return b ? gcd ( b, a % b ) : a; }

inline bool toLeft ( const int fid, const int id ) {
	return ! fid || ( a[fid] < a[id] && gcd ( a[fid], a[id] ) ^ 1 );
}

inline bool toRight ( const int fid, const int id ) {
	return fid > n || ( a[id] < a[fid] && gcd ( a[id], a[fid] ) ^ 1 );
}

int main () {
	scanf ( "%d", &n );
	for ( int i = 1; i <= n; ++ i ) scanf ( "%d", &a[i] );
	for ( int i = 1; i <= n; ++ i ) {
		f[i][i][0] = toLeft ( i - 1, i );
		f[i][i][1] = toRight ( i + 1, i );
		f[i][i - 1][0] = f[i][i - 1][1] = true;
	}
	f[n + 1][n][0] = f[n + 1][n][1] = true;
	for ( int len = 2; len <= n; ++ len ) {
		for ( int i = 1, j; ( j = i + len - 1 ) <= n; ++ i ) {
			bool &curl = f[i][j][0], &curr = f[i][j][1];
			for ( int k = i; k <= j && ( ! curl || ! curr ); ++ k ) {
				if ( f[i][k - 1][1] && f[k + 1][j][0] ) {
					if ( toLeft ( i - 1, k ) ) curl = true;
					if ( toRight ( j + 1, k ) ) curr = true;
				}
			}
		}
	}
	puts ( f[1][n][0] || f[1][n][1] ? "Yes" : "No" );
	return 0;
}