[CF961E] Tufurama

Description:

有一天Polycarp决定重看他最喜爱的电视剧《Tufurama》。当他搜索“在线全高清免费观看Tufurama第3季第7集”却只得到第7季第3集的结果时，他很惊讶。这让Polycarp感到疑惑——如果有天他决定重看整个系列却无法找到正确的剧集观看，那该怎么办呢？Polycarp现在想统计一下他被迫用不同方案搜索同一剧集的次数。

电视连续剧有\(n\) 季（从\(1\) 到\(n\) 编号），第\(i\) 季有\(a_i\) 集（从\(1\) 到\(a_i\) 编号）。Polycarp认为如果有一对\(x\) 和\(y\) （\(x\<y\) ），使第\(x\) 季第\(y\) 集、第\(y\) 季第\(x\) 集存在，那么其中一个搜索就会包含错误的内容。请帮助Polycarp统计这样的数对的数量吧！

Hint:

主席树SB题,随便搞个前缀和,离散化都不用

#include <map>

#include <set>

#include <stack>

#include <cmath>

#include <queue>

#include <cstdio>

#include <cstring>

#include <cstdlib>

#include <iostream>

#include <algorithm>

using namespace std;

typedef long long ll;

const int mxn=2e5+5;

int n,m,tot,cnt,hd[mxn];

inline int read() {

	char c=getchar(); int x=0,f=1;

	while(c>'9'||c<'0') {if(c=='-') f=-1;c=getchar();}

	while(c<='9'&&c>='0') {x=(x<<3)+(x<<1)+(c&15);c=getchar();}

	return x*f;

}

inline void chkmax(int &x,int y) {if(x<y) x=y;}

inline void chkmin(int &x,int y) {if(x>y) x=y;}

struct ed {

	int to,nxt;

}t[mxn<<1];

inline void add(int u,int v) {

	t[++cnt]=(ed) {v,hd[u]}; hd[u]=cnt;

}

int a[mxn],rt[mxn<<6],ls[mxn<<6],rs[mxn<<6],sz[mxn<<6];

ll ans;

void update(int las,int &p,int l,int r,int pos) {

	if(!p) p=++tot; sz[p]=sz[las]+1;

	if(l==r) return ; int mid=(l+r)>>1;

	if(pos<=mid) update(ls[las],ls[p],l,mid,pos),rs[p]=rs[las];

	else update(rs[las],rs[p],mid+1,r,pos),ls[p]=ls[las];

}

int query(int p,int l,int r,int pos) {

	if(l==r) return sz[p];

	int mid=(l+r)>>1;

	if(pos<=mid) return sz[rs[p]]+query(ls[p],l,mid,pos);

	else return query(rs[p],mid+1,r,pos);

}

int main()

{

	n=read();

	for(int i=1;i<=n;++i) {

		a[i]=read();

		update(rt[i-1],rt[i],1,n,a[i]);

		ans+=query(rt[min(i-1,a[i])],1,n,i);

	}

	printf("%lld",ans);

    return 0;

}