Gym - 100917J J - Judgement

题目大意：
给你一个数n
之后给你一个期望值p
再给你n个数,为数组a
再给你一个期望值q
再给你n个数,为数组b
问你是不是在任何情况下选同样下标的数都是同样超过期望值或者低于期望值的

思路：
听了师兄的题解感觉师兄的思路挺好的
可以转化成01背包
首先把p作为背包体积,a[i]为单件物品的体积，b[i]为价值
然后我们看在dp[p-1]也就是背包体积-1但价值最大的时候dp[p-1]是不是大于等于q的
如果大于等于就说明了有取数组a中元素的小于p但是取数组b的情况大于q的情况
反之把q作为背包体积,b[i]为单件物品体积,a[i]为价值再去判断
再用个bitset储存NO时候的答案就好了

AC代码：

#include <bits/stdc++.h>

using namespace std;
const int N=1e6+10;
int dp[N];
bitset<110>ans[N];
int a[N];
int b[N];
int n,p,q;
void solve()
{
    for(int i=1; i<=n; i++)
    {
        for(int j=p-1; j>=a[i]; j--)//这里不能从a[i]到p-1的原因是dp[j]=dp[j-a[i]]+b[i]会一直叠加下去
        {
            if(dp[j]<dp[j-a[i]]+b[i])
            {
                dp[j]=dp[j-a[i]]+b[i];
                ans[j]=ans[j-a[i]];
                ans[j][i]=1;
            }
        }
    }
    if(dp[p-1]>=q)
    {
        cout<<"NO"<<endl;
        for(int i=1; i<=n; i++)
        {
            if(ans[p-1][i])cout<<1;
            else cout<<0;
        }
        return;
    }
    memset(dp,0,sizeof(dp));
    for(int i=0; i<=p; i++)ans[i].reset();
    for(int i=1; i<=n; i++)
    {
        for(int j=q-1; j>=b[i]; j--)
        {
            if(dp[j]<dp[j-b[i]]+a[i])
            {
                dp[j]=dp[j-b[i]]+a[i];
                ans[j]=ans[j-b[i]];
                ans[j][i]=1;
            }
        }
    }
    if(dp[q-1]>=p)
    {
        cout<<"NO"<<endl;
        for(int i=1; i<=n; i++)
        {
            if(ans[q-1][i])cout<<1;
            else cout<<0;
        }
        return;
    }
    cout<<"YES"<<endl;
    return;
}
int main()
{
    cin>>n>>p;
    for(int i=1; i<=n; i++)cin>>a[i];
    cin>>q;
    for(int i=1; i<=n; i++)cin>>b[i];
    solve();
    return 0;
}