Description
从m个物品中选出n个,m个物品每个有重量w[i],价值c[i],要求选出的第i个物品的重量不小于a[i],价值不小于b[i],问是否有合法方案
Input
第一行一整数n表示要选出物品数量,之后n行每行两个整数a[i]和b[i]表示对选出的第i个物品的限制,之后输入一整数m表示所有物品数,之后m行每行两个整数w[i]和c[i]表示第i件物品的重量和价值(1<=n,m<=2e5,1<=a[i],b[i],w[i],c[i]<=1e9)
Output
如果存在合法方案则输出n个整数表示选出的第i个物品的编号,否则输出-1
Sample Input
3
3 4
5 1
2 6
4
5 7
4 5
3 3
5 2
Sample Output
2 4 1
Solution
对n个限制按b[i]降序排,对m件物品按c[i]降序排,游标法,每次把c[i]大于等于当前限制b[i]的物品拿出来,找一个重量大于等于a[i]的最小值作为这个限制选出的物品,用set维护当前可以用的物品,每次二分查找重量大于等于a[i]的物品,时间复杂度O(mlogm+nlogn)
Code
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
#define maxn 222222
int n,m,ans[maxn];
struct node
{
int w,c,id;
bool operator<(const node&b)const
{
return w>b.w;
}
}a[maxn],b[maxn];
typedef pair<int,int>P;
set<P>s;
set<P>::iterator it;
int main()
{
while(~scanf("%d",&n))
{
for(int i=1;i<=n;i++)scanf("%d%d",&a[i].w,&a[i].c),a[i].id=i;
scanf("%d",&m);
for(int i=1;i<=m;i++)scanf("%d%d",&b[i].w,&b[i].c),b[i].id=i;
sort(a+1,a+n+1);
sort(b+1,b+m+1);
s.clear();
int i=1,j=1,gg=0;
while(i<=n)
{
while(j<=m&&b[j].w>=a[i].w)s.insert(P(b[j].c,b[j].id)),j++;
it=s.lower_bound(P(a[i].c,-1));
if(it==s.end())
{
gg=1;
break;
}
ans[a[i].id]=(*it).second;
s.erase(it);
i++;
}
if(gg)printf("-1\n");
else
{
for(int i=1;i<=n;i++)printf("%d%c",ans[i],i==n?'\n':' ');
}
}
return 0;
}