CF 961E Tufurama 离散化+树状数组

题意:n个个电视剧,第i个电视剧有a[i]集. 可以用二元组(x,y)来表示某集电视剧.
若(x,y),(y,x)这两集都存在,则称(x,y),(y,x)为一个合法对.
n<=2e5,a[i]<=1e9 问总共有多少个合法对数?

当x=i 则y要满足:i<y<=a[i] && a[y]>=i
容易用树状数组求出后缀i中 a[y]>=i的个数.

因为要在点a[i]+1时计算 后缀a[y]>=i的个数 所以要记录v[p][i]:表示在p点要计算>=i的个数.

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e5+5;
ll n,a[N],c[N];
vector<int> b,v[N];//在第i点需要查询的数为v[i][j]
map<ll,ll> mp[N];//mp[i][j] 后缀i中a[y]>=j的个数 
int lowbit(int x)
{
	return x&-x;
}
void update(int pos,int val)
{
	for(int i=pos;i<N;i+=lowbit(i))
		c[i]+=val;
}
ll sum(int pos)
{
	ll res=0;
	for(int i=pos;i>0;i-=lowbit(i))
		res+=c[i];
	return res;
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>a[i],b.push_back(a[i]);
		v[min(n+1,a[i]+1)].push_back(i);//在a[i]+1点需要查询<=i的个数 
	} 
	sort(b.begin(),b.end());
	b.erase(unique(b.begin(),b.end()),b.end());
	ll res=0;
	for(int i=n;i>=1;i--)
	{
		ll p1=lower_bound(b.begin(),b.end(),i)-b.begin();
		ll v1=n-i-sum(p1);
		ll v2=mp[min(n+1,a[i]+1)][i];
		
		ll x=lower_bound(b.begin(),b.end(),a[i])-b.begin()+1;
		update(x,1);
		for(int j=0;j<v[i].size();j++)
			mp[i][v[i][j]]=n-i+1-sum(lower_bound(b.begin(),b.end(),v[i][j])-b.begin());
		if(a[i]>i)
			res+=v1-v2;
		//cout<<i<<' '<<v1<<' '<<v2<<'\n';
	}
	cout<<res<<'\n';
	return 0;
}