逆序对 - 归并排序求逆序对

 

#include <bits/stdc++.h>
using namespace std;

class Solution
{
public :
    int InversePairs( vector < int > data)
    {
        ans = 0;
        merge_sort( data , 0 , data . size() - 1);
        return ans;
    }

    vector < int > merge( vector < int > & vl , vector < int > & vr)
    {
        int n1 = vl . size (), n2 = vr . size();
        vector < int > res( n1 + n2);
        int i = 0 , j = 0 , cnt = 0;
        while ( i < n1 && j < n2)
        {
            if ( vl [ i ]   <= vr [ j ])
                res [ cnt ++ ] = vl [ i ++ ];
            else
            {
                ans += n1 - i; /* 统计答案 */
                res [ cnt ++ ] = vr [ j ++ ];
            }
        }
        while ( i < n1)
            res [ cnt ++ ] = vl [ i ++ ];
        while ( j < n2)
            res [ cnt ++ ] = vr [ j ++ ];
        return res;
    }

    vector < int > merge_sort( vector < int > & nums , int l , int r)
    {
        vector < int > vl , vr;
        if ( l < r)
        {
            int mid = ( l + r) >> 1;
            vl = merge_sort( nums , l , mid);
            vr = merge_sort( nums , mid + 1 , r);
            return merge( vl , vr);
        }
        else
        {
            vector < int > ve( r - l + 1);
            int cnt = 0;
            for ( int i = l; i <= r; ++ i)
                ve [ cnt ++ ] = nums [ i ];
            return ve;
        }
    }
private :
    int ans;
};

int main( int argc , char const * argv [])
{
    int n;
    while ( cin >> n)
    {
        vector < int > nums(n);
        for ( int i = 0; i < n; ++ i)
            cin >> nums [ i ];
        Solution solution;
        int ans = solution . inversePairs( nums);
        cout << ans << endl;
    }
    return 0;
}