【BZOJ4769】超级贞鱼
Description
马达加斯加贞鱼是一种神奇的双脚贞鱼,它们把自己的智慧写在脚上——每只贞鱼的左脚和右脚上各有一个数。有一天,K只贞鱼兴致来潮,排成一列,从左到右第i只贞鱼会在右脚写Ai,左脚上写上i;第二年,这K只贞鱼以右脚的数为第一关键字、左脚的数为第二关键字,从小到大排成一列。然后,它们决定重编号,从左到右第i只贞鱼会在右脚上写上左脚的数,在左脚上写i;第三年,它们按第二年的方法重排列、重编号......N年后,对于从左到右第i和第j贞鱼,若i<j且第i只贞鱼右脚上的数比第j只贞鱼右脚上的数大,则称它们为一对“超级贞鱼”。问一共有多少对“超级贞鱼”?
Input
共3行,第一行一个正整数K,表示有K只贞鱼。
第二行K个正整数,第i个数表示Ai。
第三行一个非负整数N,表示年数。
K≤2×10^6, Ai≤10^9,N≤10^18
Output
一个整数,表示“超级贞鱼”的对数。
Sample Input
6
5 2 6 3 1 7
0
5 2 6 3 1 7
0
Sample Output
7
题解:以前刷的水题重测了一发就TLE了~
容易发现,无论过了多少年,原序列的逆序对数都是不变的,所以直接求逆序对即可。
但是用排序+树状数组会被卡常数,所以要用归并排序求逆序对。这个会cdq分治的都应该会吧?
在solve(l,r)时,先solve(l,mid)和(mid+1,r),这样两边就都排好序了,就只需要统计左边和右边产生的逆序对数,用两个指针在左右两边扫一下并统计答案即可。
#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
typedef long long ll;
inline char nc(){
static char buf[100000],*p1=buf,*p2=buf;
return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline int rd(){
char ch=nc();int sum=0;
while(!(ch>='0'&&ch<='9'))ch=nc();
while(ch>='0'&&ch<='9')sum=sum*10+ch-48,ch=nc();
return sum;
}
int n,v[2000010],p[2000010];
ll ans;
void solve(int l,int r)
{
if(l==r) return ;
int mid=(l+r)>>1,i=l,h1=l,h2=mid+1;
solve(l,mid),solve(mid+1,r);
for(i=l;i<=r;i++)
{
if(h1<=mid&&(h2>r||v[h1]<=v[h2])) ans+=h2-mid-1,p[i]=v[h1++];
else p[i]=v[h2++];
}
for(i=l;i<=r;i++) v[i]=p[i];
}
int main()
{
n=rd();
int i;
for(i=1;i<=n;i++) v[i]=rd();
solve(1,n);
printf("%lld",ans);
return 0;
}