Codeforces Round #658 (Div. 1) B. Unmerge

B. Unmerge

题目传送门：

Unmerge

思路：

我们如果一个数后面跟着一串比它小的数，那么这些数一定来自同一个数列。然后我们就可以把长度为2*n的数列用这种判断方法分成很多段，然后用01背包的思想，看能不能取其中的几段数列从而构成一段长度为n的数列。

AC Code

#include<bits/stdc++.h>
using namespace std;
const int N=5000;
int p[N];
int w[N],tot=0,f[N];//每个子串的长度，子串的数量
int main()
{
    
    
    int t;
    scanf("%d",&t);
    while(t--)
    {
    
    
        int n;
        scanf("%d",&n);
        for(int i=1;i<=2*n;i++)
            scanf("%d",&p[i]);
        memset(f,0,sizeof(f));
        tot=0;
        int idx=1,num=1;
        while(idx<=2*n&&num<=2*n)
        {
    
    
            int k=1;
            while(p[num+1]<p[idx]&&num+1<=2*n)
            {
    
    
                num++;
                k++;
            }
            w[++tot]=k;
            num++;
            idx=num;
        }
        //01背包,相当于有tot件物品，能不能刚好装满n的空间
        for(int i=1;i<=tot;i++)
        {
    
    
            for(int j=n;j>=w[i];j--)
                f[j]=max(f[j],f[j-w[i]]+w[i]);
        }
        if(f[n]==n) printf("YES\n");
        else printf("NO\n");
    }
    //system("pause");
    return 0;
}