P1966 [NOIP2013 提高组] 火柴排队做题笔记

这题和 P5677 一样，是从树状数组题单里翻出来的，由于开始看时感觉题解代码写的不是很清晰，就先放进了做题计划里，后来几次看这道题，但由于第一次看题可能留下了一些心理阴影以及时间不多，一直没切掉。直到先去做了用树状数组求逆序对，才感觉这道题变得简单了不少。思路：用一个数组储存输入数组的标号，按对应数的大小对标号数组进行排序。然后把其中一个标号数组为关键字对另一个标号数组进行排序，被排序的标号数组排序后的逆序对数量即为答案。

#include <bits/stdc++.h>
  using namespace std;
#define int long long
const int MOD=1e8-3;
int n,a[100005],b[100005],c[100005],d[100005],t[100005],s[100005],ans;
bool cmp1(int u,int v)
{
//	if (a[u]==a[v]) return u>v;
	return a[u]<a[v];
}
bool cmp2(int u,int v)
{
//	if (b[u]==b[v]) return u>v;
	return b[u]<b[v];
}
void Add(int t)
{
	for (int j=t;j<=n;j+=j&(-j))
	{
		s[j]++;
		s[j]%=MOD;
	}
}
int getsum(int t)
{
	int ans=0;
	for (int j=t;j;j-=j&(-j))
	{
		ans+=s[j];
		ans%=MOD;
	}
	return ans;
}
signed main()
{
	ios::sync_with_stdio(0); cout.tie(nullptr);
	cin>>n;
	for (int i=1;i<=n;i++)
	{
		cin>>a[i];
		c[i]=i;
	}
	for (int i=1;i<=n;i++)
	{
		cin>>b[i];
		d[i]=i;
	}
	sort(c+1,c+n+1,cmp1);
	sort(d+1,d+n+1,cmp2);
	for (int i=1;i<=n;i++)
	{
		t[c[i]]=d[i];
	}
	for (int i=n;i>=1;i--)
	{
		ans+=getsum(t[i]-1);
		Add(t[i]);
		ans%=MOD;
	}
	cout<<ans<<endl;
	return 0;
}