正题
题目链接:http://www.51nod.com/Challenge/Problem.html#problemId=1821
题目大意
nnn个可重集合,TTT次询问,询问将集合SxS_xSx和一个选出大小为kkk的SyS_ySy的子集合并成的独立集S′S'S′满足能够找到一个最大的zzz使得对于任意的正整数i∈[1,z]i\in [1,z]i∈[1,z]都存在一个S′S'S′的子集和为iii。
1≤n,mi≤1000,1≤T≤100001\leq n,m_i\leq 1000,1\leq T\leq 100001≤n,mi≤1000,1≤T≤10000
解题思路
考虑我们已经拼出了一个zzz,那么我们如果加入一个≤z+1\leq z+1≤z+1的数xxx那么就可以使得上限变成z+xz+xz+x。
所以可以考虑贪心,因为第一个集合没有限制我们就优先选择第一个集合里的数,直到不能选择我们就选择第二个集合里的直到能够继续选择第一个集合里的。
排好序然后开个栈维护就好了。
时间复杂度O(nmlogm+Tm)O(nm\log m+Tm)O(nmlogm+Tm)
code
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const ll N=1100;
ll n,T,m[N],s[N],a[N][N];
signed main()
{scanf("%lld",&n);for(ll i=1;i<=n;i++){scanf("%lld",&m[i]);for(ll j=1;j<=m[i];j++)scanf("%lld",&a[i][j]);sort(a[i]+1,a[i]+1+m[i]);}scanf("%lld",&T);while(T--){ll x,y,k,z=1,top=0,sum=1,flag=0;scanf("%lld%lld%lld",&x,&y,&k);for(ll i=1;i<=m[x];i++){while(z<=m[y]&&a[y][z]<=sum)s[++top]=a[y][z],z++;while(sum<a[x][i]){if(!k||!top){flag=1;break;}sum+=s[top];k--;top--;}if(flag)break;sum+=a[x][i];}while(k&&top)sum+=s[top],top--,k--;printf("%lld\n",sum-1);}return 0;
}