CodeForces - 1381B Unmerge（01背包）

题面：

题意：

给定一共长度为 $2\times n$ 的数组（每个数都不相同）。问能不能将其划分成两个长度为 $n$ 的数组。满足：每次取这两个数组开头的较小的数放入队列，直至取完，队列中的数列与给出的 $2\times n$ 的数组相同。

思路：

根据条件，我们可以将原数组划分为许多段，每一段满足第一个数大于该段中的其他数，且让这个段尽量长（手玩一下就知道原因了 ），然后 $01$ 背包，容量为 $n$ ，如果能装满就 $YES$ ，否则 $NO$ 。

#include<bits/stdc++.h>
#define ll long long
#define ms(x,a) memset(x,a,sizeof(x))
#define inf 0x3f3f3f3f
using namespace std;
const int maxn = 4e3 + 10;
int a[maxn],sz[maxn],dp[maxn];

int main() {
	int t;
	scanf("%d",&t);
	for(int ca = 1;ca <= t; ++ca){
		int n;
		scanf("%d",&n);
		for(int i = 1;i <= (n << 1); ++i){
			scanf("%d",&a[i]);
		} 
		int cnt = 0;
		for(int i = 1;i <= (n << 1);){
			int j = i + 1,now = a[i];
			while(a[j] < now && j <= (n << 1)){
				++j;
			}
			sz[++cnt] = j - i;
			i = j;
		}
		ms(dp,0);
		for(int i = 1;i <= cnt; ++i){
			for(int j = n;j >= sz[i]; --j){
				dp[j] = max({dp[j],dp[j - sz[i]] + sz[i]});
			}
		}
		if(dp[n] == n){
			puts("YES");
		}
		else {
			puts("NO");
		}
	}
	return 0;
}

/*
1
2
3 1 2 4

1
4
6 1 3 7 4 5 8 2

*/