Codeforces Round #658 (Div. 2) D. Unmerge(dp,01背包)

题目链接

思路:

将每个数和他之后的比他小的数都分为一组,对分好的组进行01背包,看能否将组的总和变为n。

代码:

#include
using namespace std;
#define int long long
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
const int N=1e5+7;
const double eps=1e-8;
const int mod=1e9+7;
const int inf=0x7fffffff;
const double pi=3.1415926535;
signed main()
{
    IOS;
    int t;
    cin>>t;
    while(t--)
    {
        int ans[N],arr[N],dp[N]={0},n,m=0;
        cin>>n;
        for(int i=1;i<=2*n;i++)
        {
            cin>>arr[i];
        }
        int Max=arr[1],num = 1;
        for(int i=2;i<=2*n;i++)
        {
            if(arr[i]>Max)
            {
                Max=arr[i];
                ans[m++]=num;
                num = 0;
            }
            num++;
        }
        for(int i=0;i<m;i++)
        {
            for(int j = n;j >= ans[i];j--)
            {
                dp[j] = max(dp[j],dp[j-ans[i]]+ans[i]);
            }
        }
        cout<<(dp[n]==n?"YES":"NO")<<endl;
    }
    return 0;
}

你可能感兴趣的:(思维,dp,dfs)