[BZOJ4430][Nwerc2015]Guessing Camels赌骆驼

[Nwerc2015]Guessing Camels赌骆驼

Description
Jaap, Jan, and Thijs are on a trip to the desert after having attended the ACM ICPC World Finals 2015 in Morocco. The trip included a camel ride, and after returning from the ride, their guide invited them to a big camel race in the evening. The camels they rode will also participate and it is customary to bet on the results of the race.
One of the most interesting bets involves guessing the complete order in which the camels will finish the race. This bet offers the biggest return on your money, since it is also the one that is the hardest to get right.
Jaap, Jan, and Thijs have already placed their bets, but the race will not start until an hour from now, so they are getting bored. They started wondering how many pairs of camels they have put in the same order. If camel cc is before camel d on Jaap’s, Jan’s and Thijs’ bet, it means that all three of them put c and d in the same order. Can you help them to calculate the number of pairs of camels for which this happened?
Jaap，Jan和Thijs在摩洛哥参加完ACMICPC2015全球总决赛后去沙漠进行了一次旅行。旅行包括了骑骆驼。在骑完回来之后，他们的向导邀请他们在晚上去看一场盛大的骆驼赛跑。他们骑的骆驼也会参加，而赌比赛的结果是一个习惯。其中一个最有趣的赌涉及猜测完成比赛的骆驼的完整名单。这个赌可以为你赢得最多的钱，因为它也是最难猜对的。
Jaap，Jan和Thijs已经下了注，但比赛要在一个小时后才开始，所以他们觉得很无聊。他们开始想知道有多少对骆驼他们赌了同样的顺序。如果在他们三人的猜测名单中，骆驼c在骆驼d前，就意味着c和d在他们的名单中有相同的顺序。你能帮他们计算有多少对骆驼是这样的吗？
Input
The input consists of:
one line with an integer n ( 2≤n≤200000), the number of camels;
one line with n integers a1,…,ana1,…,an ( 1≤ai≤n for all i ), Jaap’s bet. Here a1 is the camel in the first position of Jaap’s bet, a2 is the camel in the second position, and so on;
one line with Jan’s bet, in the same format as Jaap’s bet;
one line with Thijs’ bet, in the same format as Jaap’s bet.
The camels are numbered 1,…,n . Each camel appears exactly once in each bet.
输入包括：
第一行是一个整数n（2<=n<=200000），骆驼的数量。
第二行有n个整数a1…an（1<=ai<=n），是Jaap的下赌名单。a1是名单中第一位，a2是第二位，等等。
第三行是Jan的名单，格式同上。
第四行是Thijs的名单，格式同上。
骆驼从1到n编号，每头骆驼在一份名单中只出现一次。
Output
Output the number of pairs of camels that appear in the same order in all 3 bets.
输出在三份名单中同样顺序的骆驼有多少对。
Sample Input
Sample input 1
3
3 2 1
1 2 3
1 2 3
Sample input 2
4
2 3 1 4
2 1 4 3
2 4 3 1
Sample Output
Sample output 1
0
Sample output 2
3

Code

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define hash _hash
#define MS(_) memset(_,0,sizeof(_))
template<typename T> inline void read(T &x){
    x=0; T f=1; char ch=getchar();
    while(!isdigit(ch)) {if(ch=='-') f=-1; ch=getchar();}
    while(isdigit(ch))  {x=x*10+ch-'0'; ch=getchar();}
    x*=f;
}

const int MaxN = 300100;
int n;
ll a[MaxN],b[MaxN],c[MaxN],pos[MaxN],hash[MaxN],BIT[MaxN];

inline int lowbit(int k){return k&-k;}
inline ll qry(int k){ll res=0;for(;k;k-=lowbit(k))res+=BIT[k];return res;}
inline void mdf(int k){for(;k<=n;k+=lowbit(k))BIT[k]++;}
inline ll cal(ll a[],ll b[]){
    for(int i=1;i<=n;i++)hash[a[i]]=i;
    for(int i=1;i<=n;i++)pos[hash[b[i]]]=i;
    MS(BIT);ll res=0;
    for(int i=n;i>=1;i--){res += qry(pos[i]-1);mdf(pos[i]);}
    return res;
}
int main(){
    read(n);
    for(int i=1;i<=n;i++)read(a[i]);
    for(int i=1;i<=n;i++)read(b[i]);
    for(int i=1;i<=n;i++)read(c[i]);
    ll ans = cal(a,b)+cal(a,c)+cal(b,c);    
    printf("%lld\n",1ll*n*(n-1)/2ll-ans/2ll);
    return 0;
}