（模拟赛）另类逆序对

最新推荐文章于 2022-10-10 15:19:00 发布

zwbnnhz

最新推荐文章于 2022-10-10 15:19:00 发布

阅读量336

点赞数

CC 4.0 BY-SA版权

分类专栏：数据结构

本文链接：https://blog.csdn.net/zybnnhz/article/details/77435710

数据结构专栏收录该内容

4 篇文章

订阅专栏

逆序对是一个非常经典的问题，对于一个序列 a，如果有 ai<aj并且 i>j，则我们称 (i,j)为一个逆序对。
但是逆序对虽然经典，但却一个非常困难的问题，因为一个序列的逆序对数可能太多了。所以为简化问题，我们给定一个长度为N的序列 a和一个参数k，我们希望知道有多少个 (L,R)满足 1≤L<R≤N，且 a1,a2......aL,aR......aN-1,aN的逆序对个数不超过 k。
【输入格式】
第一行两个整数N,k。
接下来一行 N个整数代表序列。
【输出格式】
一行个整数代表答案。
【样例输入1】
3 1

【样例输出1】
3
【样例输入2】
5 2
1 3 2 7
【样例输出2】
6
【数据范围与规定】
对于 100%的数据， 1≤N≤10^5,1≤.ai≤10^9,K≤10^18，有部分。

嗯真的感觉这是一道好题

这题感觉就是NOIP2011选择客栈的变式

对于当前枚举的L,R若满足条件，则对于当前的L，当R取R~N时都满足条件

对于当前的枚举的L,R若不满足条件，则对于当前的L，当R取L+1~R时都不能满足条件

于是就转变成了一个滑动窗口问题，双指针维护，用树状数组算出左右指针移动对当前总值的贡献

一定要离散化！！！！

#include<cstdio>
#include<iostream>
#include<algorithm>
#define For(i,j,k) for(i=j;i<=k;++i)
#define Forr(i,j,k) for(i=j;i>=k;--i)
using namespace std;
const int N=1000000+5;
inline int read(){
	char c=getchar();int f=1;
	while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}
	int x=c-'0';
	c=getchar();
	while(c<='9'&&c>='0'){x=x*10+c-'0';c=getchar();}
	return f*x;
	}
int a[N],sub_a[N],c[N],c_re[N];
int n,k,ans;
int sum[N],sum2[N];
int lowbit(int x){
	return x&(-x);
	}
int query(int x){
	int ret=0;
	while(x>0){
		ret+=sum[x];
		x-=lowbit(x);
		}
	return ret;
	}
void add(int x,int d){
	while(x<=n){
		sum[x]+=d;x+=lowbit(x);                      
		}
	}
int query2(int x){
	int ret=0;
	while(x>0){
		ret+=sum2[x];
		x-=lowbit(x);
		}
	return ret;
	}
void add2(int x,int d){
	while(x<=n){
		sum2[x]+=d;x+=lowbit(x);                      
		}
	}
int main(){
	int i,j;
	n=read();k=read();
	For(i,1,n){a[i]=read();sub_a[i]=a[i];}
	sort(sub_a+1,sub_a+1+n);
	int size=unique(sub_a+1,sub_a+1+n)-(sub_a+1);
	For(i,1,n)a[i]=lower_bound(sub_a+1,sub_a+1+size,a[i])-(sub_a+1)+1;
	int tot=0;
	j=n+1;
	For(i,1,n-1){
		if(i==j){
			tot-=query(n)-query(a[j]);
			tot-=query2(a[j]-1);
			add2(a[j],-1);
			j++;
		}
		tot+=query(n)-query(a[i]);
		tot+=query2(a[i]-1);
		add(a[i],1);
		while(tot<=k&&j>i+1){
			j--;
			tot+=query(n)-query(a[j]);
			tot+=query2(a[j]-1);
			add2(a[j],1);
			}
		while(tot>k&&j<=n){
			tot-=query(n)-query(a[j]);
			tot-=query2(a[j]-1);
			add2(a[j],-1);
			j++;
			}
			if(j==n+1)break;
			ans+=n-j+1;
	}
	cout<<ans;
	return 0;
	}