以前一直以为很难,学了一下很简单
#include<bits/stdc++.h>
using namespace std;
using LL = int64_t;
const int maxn=1e5+5;
const int mod=1e9+7;
const int INF=0x3f3f3f3f;
const LL LLINF=1e14;
const double pi=acos(-1.0);
LL ans=0;
int a[maxn],b[maxn];
void merge(int l,int mid,int r) {
int i=l,j=mid+1;
for(int k=l;k<=r;k++) {
if(j>r||(i<=mid&&a[i]<=a[j])) b[k]=a[i++];
else b[k]=a[j++],ans+=mid-i+1;
}
for(int k=l;k<=r;k++) a[k]=b[k];
}
void merge_sort(int l,int r) {
if(l<r) {
merge_sort(l,(l+r)/2);
merge_sort((l+r)/2+1,r);
merge(l,(l+r)/2,r);
}
}
int main()
{
ios::sync_with_stdio(0);
cin.tie(0);
LL n,k;
while(cin>>n>>k){
ans=0;
for(int i=1;i<=n;i++) cin>>a[i];
merge_sort(1,n);
cout<<max(ans-k,0LL)<<endl;
}
return 0;
}
还有用树状数组求逆序对的,复杂度也是O(nlgn),但树状数组是维护区间前缀和,倒转数组,每次求一次[1,t[i]-1]的前缀,但缺陷是求前缀和不能过大, 要不然不能维护,除非离散化一下
#include<bits/stdc++.h>
using namespace std;
using LL = int64_t;
const int maxn=1e5+5;
const int mod=1e9+7;
const int INF=0x3f3f3f3f;
const LL LLINF=1e14;
const double pi=acos(-1.0);
int lowbit(int x) {return x&-x;}
int a[maxn],t[maxn];
int ask(int x) {
LL ans=0;
for(;x;x-=lowbit(x)) ans+=a[x];
return ans;
}
void add(int x,int y) {
for(;x<=1e5;x+=lowbit(x)) a[x]+=y;
}
int main()
{
ios::sync_with_stdio(0);
cin.tie(0);
LL n,k;
while(cin>>n>>k) {
LL ans=0;
memset(a,0,sizeof(a));
for(int i=1;i<=n;i++) cin>>t[i];
for(int i=n;i>=1;i--) {
add(t[i],1);
ans+=ask(t[i]-1);
}
cout<<max(ans-k,0LL)<<endl;
}
return 0;
}