分治法4-逆序对的数量

题目4：逆序对的数量

给定一个长度为n的整数数列，以及一个整数k，请用快速选择算法求出数列从小到大排序后的第k个数。

输入格式

第一行包含两个整数n和k。

第二行包含n个整数，表示整数数列。

输出格式

输出一个整数，表示数列的第k小数。

输入样例：

6
2 3 4 5 6 1

输出样例：

5

代码：

#include<iostream>
#include<vector>
using namespace std;

int func(vector<int>& vec, int start)
{
    
    if (start >= vec.size() - 1)
    {
        return 0;
    }
    //把原问题分解成两个规模不同的子问题A和子问题B
    //对子问题A进行求解
    int sum = 0;
    for (int i = start + 1; i < vec.size(); i++)
    {
        if (vec[start] > vec[i])
        {
            sum++;
        }
    }
    //对子问题B进行递归分解
    int B = func(vec, start + 1);
    //合并子问题A和子问题B的解得到原问题的解
    return sum + B;
}


int main()
{
    int n;
    cin >> n;
    vector<int> vec;
    for (int i = 0; i < n; i++)
    {
        int tmp;
        cin >> tmp;
        vec.push_back(tmp);
    }
    cout << func(vec, 0) << endl;
    
}