H - Nine Packs Kattis - ninepacks

题目链接
题意就是在a数组中找出x个数，b数组中找出y个数，且x个数之和等于y个数之和。
本想着用暴力的写法，但是出来之后不对，因为此题并不是连续的，还可以跳着取数。
下面是WA的代码：

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<vector>
using namespace std;
const int N=1e6+10;
struct node
{
    int data;
    int count;
};
int a[N],b[N];
vector<node> vec1,vec2;
int main()
{
    int n;
    cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i];
    int m;
    cin>>m;
    for(int i=1;i<=m;i++) cin>>b[i];
    for(int i=1;i<=n;i++)
    {
        int sum=0;
        int coun=0;
        for(int j=i;j>=1;j--)
        {
            sum+=a[j];
            coun++;
            vec1.push_back({sum,coun});
        }
    }
    for(int i=1;i<=m;i++)
    {
        int sum=0;
        int coun=0;
        for(int j=i;j>=1;j--)
        {
            sum+=b[j];
            coun++;
            vec2.push_back({sum,coun});
        }
    }
    int minnum=0x3f3f3f3f;
    for(int i=0;i<vec1.size();i++)
    {
        for(int j=0;j<vec2.size();j++)
        {
            if(vec1[i].data==vec2[j].data)
            {
                if(minnum>vec1[i].count+vec2[j].count)
                {
                    minnum=vec1[i].count+vec2[j].count;
                }
            }
        }
    }
    if(minnum==0x3f3f3f3f) printf("impossible\n");
    else printf("%d\n",minnum);
    return 0;
}

下面是AC代码：

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
const int N=1e7;
int dp1[N],dp2[N];

int main()
{
    int n;
    memset(dp1,0x3f,sizeof(dp1));
    cin>>n;
    dp1[0]=0;
    for(int i=1;i<=n;i++)
    {
        int x;
        cin>>x;
        for(int j=1100000;j>=x;j--)
        {
            dp1[j]=min(dp1[j-x]+1,dp1[j]);
        }
    }
    memset(dp2,0x3f,sizeof(dp2));
    int m;
    cin>>m;
    dp2[0]=0;
    for(int i=1;i<=m;i++)
    {
        int x;
        cin>>x;
        for(int j=1100000;j>=x;j--)
        {
            dp2[j]=min(dp2[j-x]+1,dp2[j]);
        }
    }
    int ans=0x3f3f3f3f;
    for(int i=1;i<=1100000;i++)
    {
        ans=min(ans,dp1[i]+dp2[i]);
    }
    if(ans==0x3f3f3f3f) printf("impossible\n");
    else printf("%d\n",ans);
    return 0;
}
Comment in Markdown
Preview:

这里注意取数一定要大于1e6，因为100个1000就是1e6。