AtCoder Beginner Contest 231 F 离散化 树状数组

题目

有N个种礼物，你必须选择两个礼物（可以是相同的种类）分别送给两个人。
A和B都对礼物每个礼物有期望值ai bi。
他们都不希望对方的礼物的期望值大于送到自己的礼物（期望值是相对自己而言）。
问在N平方种方法中有几种分配方法能让满足他们的要求。

题解思路

正向反向考虑应该都行。
我这里用的是正向考虑。
我们将礼物的期望值带编号排序，这样我们往前遍历的过程中，之前的点就可以满足一个人的要求了，这样问题就变成了，前N个物品，找出有几个大于等于bi的。
可以想到利用树状数组的存编号来继续快速前缀和判断（类似求逆序对），但是树状数组只能存原始数值，加上这题只需两个值的相对数组关系，很容易想到离散化了。
但是到这里，还是有个bug，就是两个礼物的值完全相同的时候。这下就需要特判了。
总之这题感觉就是在凑答案。

AC代码

#include <bits/stdc++.h>
//#include <unordered_map>
//priority_queue
#define PII pair<int,int>
#define ll long long

using namespace std;

const  int  INF =  0x3f3f3f3f;
const  int  N =  200100;

PII a[N] ; 
PII b[N] ; 
unordered_map <int , int > mp ;
unordered_map <int , int > mp2 ; 
map <PII , int > vis ; 
int tee[N] ;  
int n ;
int lowbit(int x )
{
    return x&(-x) ; 
}
int sum(int x )
{
    int ans = 0 ; 
    for (int i = x ; i ; i-=lowbit(i) )
        ans += tee[i] ; 
    return ans ; 
}
void add(int x )
{
    for (int i = x ; i <= n ; i+=lowbit(i) )
        tee[i] += 1 ; 
}
bool cmp(PII x , PII y )
{
    if ( x.first == y.first )
        return mp[x.second] > mp[y.second] ; 
    return x.first < y.first ; 
}
int main()
{
    ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
    long long ans = 0 ;
    cin >> n ;
    for (int i = 1 ; i <= n ; i++ )
    {
        int t1 ;
        cin >> t1 ; 
        a[i].first = t1 ;
        a[i].second = i ; 
    }
    for (int i = 1 ; i <= n ; i++ )
    {
        int t1 ;
        cin >> t1 ; 
        b[i].first = t1 ;
        b[i].second = i ; 
    }
    sort(b+1,b+1+n) ; 
    int cnt = 1 ; 
    for (int i = 1 ; i <= n ; i++ )
    {
        if (!mp2[b[i].first])
        {
            mp[b[i].second] = cnt  ;
            mp2[b[i].first] = cnt  ; 
            cnt ++ ; 
        }else
            mp[b[i].second] = mp2[b[i].first] ; 
    }
    sort(a+1,a+1+n,cmp) ;  
    /*
    for (int i = 1 ; i <= n ; i++ )
        cout << a[i].first << " " ;
    cout << "\n" ; 
    for (int i = 1 ; i <= n ; i++ )
        cout << mp[a[i].second] << " " ;
    cout << "\n" ; 
    */
    for (int i = 1 ; i <= n ; i++ )
    {
        add(mp[a[i].second]);
        int s = sum(n) - sum(mp[a[i].second]-1); 
        ans += s ; 
        //cout << s << " " << ans << "\n" ; 
        vis[{a[i].first,mp[a[i].second]}]++ ;
        if ( vis[{a[i].first,mp[a[i].second]}] )  
            ans += vis[{a[i].first,mp[a[i].second]}] - 1 ; 
    }
    cout << ans  << "\n" ; 
    return 0 ;
}