剑指offer----数组中的逆序对

题目描述
在数组中的两个数字,如果前面一个数字大于后面的数字,则这两个数字组成一个逆序对。输入一个数组,求出这个数组中的逆序对的总数P。并将P对1000000007取模的结果输出。 即输出P%1000000007。

class Solution {
public:
//合并两个已排序子数组,并统计逆序对,返回合并后排好序的数组
    vector<int> merge_count(const vector<int>& vec1,const vector<int>& vec2,long long &count)
    {
        vector<int> res(vec1.size()+vec2.size(),0);
        int i=vec1.size()-1,j=vec2.size()-1;
        int index=vec1.size()+vec2.size()-1;
        while(i>=0&&j>=0)
        {
            if(vec1[i]>vec2[j])
            {
                count+=j+1;
                res[index--]=vec1[i];
                --i;
            }
            else
            {
                res[index--]=vec2[j];
                --j;
            }
        }
        if(i<0)
        {
            while(j>=0)
            {
                res[index--]=vec2[j--];
            }
        }
        else
        {
            while(i>=0)
            {
                res[index--]=vec1[i--];
            }
        }
        count%=1000000007;
        return res;
    }
    int InversePairs(vector<int> data) {
        int mark_size=1;
        long long count=0;
        //用来存放需要合并的数组,牺牲了大量空间,但更为容易理解
        vector<vector<int>> vec;
        while(mark_size<data.size())
        {
            if(mark_size==1)
            {
                int i=0;
                for(i=0;i+1<data.size();i+=2)
                {
                    vec.push_back(merge_count(vector<int>(1,data[i]),\
                    vector<int>(1,data[i+1]),count));
                }
                if(i<data.size())
                {
                    vec.push_back(vector<int>(1,data[i]));
                }
            }
            else
            {
                vector<vector<int>> mark_vec;
                int i=0;
                for(;i+1<vec.size();i+=2)
                {
                    mark_vec.push_back(merge_count(vec[i],vec[i+1],count));
                }
                if(i<vec.size())
                {
                    mark_vec.push_back(vec[i]);
                }
                vec=mark_vec;
            }
            mark_size*=2;
        }
        return count;
    }
};

猜你喜欢

转载自blog.csdn.net/xiaocongcxc/article/details/82766013