Description
Solution
题目要求的是最大值与最小值的差尽量小,那么就是指尽量使得最大值小,最小值大
考虑把 a i a_i ai和 b i b_i bi混在一起排序,那么结果一定是某个子序列的左右两端的差,其余的删去
用个双指针 l , r l,r l,r扫一下,代表着删去区间 [ 1 , l ] [1,l] [1,l]和 [ r , n ] [r,n] [r,n]注意两点:
- 删去的 a a a的个数不能超过 m m m
- 不要同时删去一张卡牌的 a a a和 b b b
第1点用个变量动态记录一下就好,第2点开个标记数组记录哪张卡牌的某个面被删去了
时间复杂度 O ( n log n ) O(n\log n) O(nlogn),瓶颈是排序,当然可以优化到 O ( n ) O(n) O(n)
Code
#include<cstdio>
#include<algorithm>
#define inf 2147000000
#define N 1000005
using namespace std;
struct node
{
int val,flag,num;
}a[N<<1];
int n,m,l,r,now,ans;
bool bj[N];
bool cmp(node x,node y) {return x.val<y.val;}
int main()
{
freopen("card.in","r",stdin);
freopen("card.out","w",stdout);
scanf("%d%d",&n,&m);
for (int i=1;i<=n;++i)
{
scanf("%d",&a[i].val);
a[i].flag=1;
a[i].num=i;
}
for (int i=1;i<=n;++i)
{
scanf("%d",&a[i+n].val);
a[i+n].flag=0;
a[i+n].num=i;
}
sort(a+1,a+2*n+1,cmp);
l=0;r=2*n+1;
while (!bj[a[l+1].num]&&now+a[l+1].flag<=m) ++l,now+=a[l].flag,bj[a[l].num]=true;
while (!bj[a[r-1].num]&&now+a[r-1].flag<=m) --r,now+=a[r].flag,bj[a[r].num]=true;
ans=inf;
while (l>=0)
{
ans=min(ans,a[r-1].val-a[l+1].val);
bj[a[l].num]=true;
now-=a[l].flag;
--l;
while (!bj[a[r-1].num]&&now+a[r-1].flag<=m) --r,now+=a[r].flag,bj[a[r].num]=true;
}
printf("%d\n",ans);
return 0;
}