Description
给出一个长度为n的序列,你每次可以反转两个值,在反转不超过k次的情况下问反转后序列最小的逆序对数
Input
多组输入,每组用例第一行为两个整数n和k分别表示序列长度和最大反转次数,第二行n个整数表示这个序列,以文件尾结束输入
Output
对于每组用例,输出反转次数不超过k次后序列的最小逆序对数
Sample Input
3 1
2 2 1
3 0
2 2 1
Sample Output
1
2
Solution
显然要想让反转后序列的逆序对数最小,那么每次反转都要把一逆序对反转,所以只要用树状数组求出逆序对ans之后与k比较即可,如果ans<=k,那么所有的逆序对都可以被反转,答案是0,如果ans>k,则答案为ans-k,注意因为序列元素较大所以要离散化
Code
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
#define maxn 111111
typedef long long ll;
struct node
{
ll val;
int pos;
}a[maxn];
ll b[maxn];
int n;
ll k;
ll cmp(node a,node b)
{
if(a.val!=b.val)//值不同则按数值升序排
return a.val<b.val;
return a.pos<b.pos;//值相同则按位置升序排
}
int getsum(int x)//求前缀和
{
int sum=0;
while(x>0)
{
sum+=b[x];
x-=x&-x;
}
return sum;
}
void update(int x,int v)//单点更新
{
while(x<=maxn)
{
b[x]+=v;
x+=x&-x;
}
}
int main()
{
while(scanf("%d%I64d",&n,&k)!=EOF)
{
memset(b,0,sizeof(b));
for(int i=0;i<n;i++)
{
scanf("%I64d",&a[i].val);
a[i].pos=i+1;//记录位置
}
sort(a,a+n,cmp);//排序
ll ans=0;
for(int i=n-1;i>=0;i--)//倒着插点求逆序对
{
ans+=getsum(a[i].pos);
update(a[i].pos,1);
}
if(k>=ans)
printf("0\n");
else
printf("%I64d\n",ans-k);
}
return 0;
}