题目:火柴排队-题目
以下是本人代码:
#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#define mod 99999997
using namespace std;
long n,p[100010]={0},q[100010]={0},r[100010]={0},c[100010]={0},ans=0;
struct ele
{
long val,pos;
}a[100010],b[100010];
bool cmp(ele a,ele b)
{
return a.val<b.val;
}
int lowbit(int i)
{
return i&(-i);
}
void add(int x)
{
for(int i=x;i<=n;i+=lowbit(i))
c[i]++;
}
long sum(int x)
{
long s=0;
for(int i=x;i>0;i-=lowbit(i))
s+=c[i];
return s;
}
int main()
{
scanf("%ld",&n);
for(int i=1;i<=n;i++) {scanf("%ld",&a[i].val);a[i].pos=i;}
for(int i=1;i<=n;i++) {scanf("%ld",&b[i].val);b[i].pos=i;}
sort(a+1,a+n+1,cmp);
for(int i=1;i<=n;i++) p[a[i].pos]=i;
for(int i=1;i<=n;i++) r[p[i]]=i;
sort(b+1,b+n+1,cmp);
for(int i=1;i<=n;i++) q[b[i].pos]=i;
for(int i=1;i<=n;i++)
{
add(r[q[i]]);
ans=(ans+(i-sum(r[q[i]])))%mod;
}
printf("%ld",ans%mod);
return 0;
}