15 数据流的第 K 大数值

15 数据流的第 K 大数值

作者: Turbo时间限制: 1S章节: 课程设计

问题描述 :

设计一个找到数据流中第 k 大元素的类（class）。注意是排序后的第 k 大元素，不是第 k 个不同的元素。

请实现 KthLargest 类：

    KthLargest(int k, int[] nums) 使用整数 k 和整数流 nums 初始化对象。

    int add(int val) 将 val 插入数据流 nums 后，返回当前数据流中第 k 大的元素。

示例：

输入：

3 4

4 5 8 2

add 3

add 5

add 10

add 9

add 4

输出：

4

5

5

8

8

解释：

KthLargest kthLargest = new KthLargest(3, [4, 5, 8, 2]);

kthLargest.add(3);   // return 4

kthLargest.add(5);   // return 5

kthLargest.add(10);  // return 5

kthLargest.add(9);   // return 8

kthLargest.add(4);   // return 8

输入说明 :

输入若干行：

首行输入两个整数k和n，k代表题中的k，n代表用来初始化对象的nums数组的长度。

第二行输入n个整数代表用来初始化对象的数组nums的元素。

后面若干行输入 add 指令，每个add后面跟一个整数表示val。

提示：

    1 <= k <= 10^4

    0 <= n <= 10^4

    -10^4 <= nums[i] <= 10^4

    -10^4 <= val <= 10^4

    最多调用 add 方法 10^4 次

    题目数据保证，在查找第 k 大元素时，数组中至少有 k 个元素

输出说明 :

输出若干行，每行一个整数，为add指令的返回值。

输入范例 :

2 4
3 5 7 6
add 5
add 4
add 8

输出范例 :

6
6
7
 

该段代码来源于我组 CC同学 必须表示感谢

、

#include<iostream>
#include<algorithm>
using namespace std;
class KthLargest
{
    int k;
    int* arr;
    int size;
public:
    KthLargest();

    KthLargest(int k, int nums[], int n);

    KthLargest(const KthLargest& K);

    KthLargest operator=(const KthLargest& K);

    ~KthLargest();

    int add(int val);
};

KthLargest::KthLargest()
{
    k = 0;
    arr = new int[100001];
    size = 0;
}
KthLargest::KthLargest(int k, int nums[], int n)
{
    this->k = k;
    arr = new int[100001];
    size = n;
    for (int i = 0; i < size; i++)
    {
        arr[i] = nums[i];
    }
}

KthLargest::KthLargest(const KthLargest& K)
{
    k = K.k;
    arr = new int[100001];
    size = K.size;
    for (int i = 0; i < size; i++)
    {
        arr[i] = K.arr[i];
    }
}

KthLargest KthLargest::operator=(const KthLargest& K)
{
    k = K.k;
    arr = new int[100001];
    size = K.size;
    for (int i = 0; i < size; i++)
    {
        arr[i] = K.arr[i];
    }
    return *this;
}

KthLargest::~KthLargest()
{
    delete[]arr;
}


int KthLargest::add(int val)
{
    int i;
    for (i = 0; i < size; i++)
    {
        if (val < arr[i])
            break;
    }
    for (int j = size; j > i; j--)
    {
        arr[j] = arr[j - 1];
    }
    arr[i] = val; size++;
    return arr[size - k];
}

int nums[10005];

int main()
{
    int k, n, i, val, theK; cin >> k >> n;
    for (i = 0; i < n; i++)
    {
        cin >> nums[i];
    }
    sort(nums, nums + n);
    string str;
    KthLargest MyK(k, nums, n);
    cin.get();
    while (cin >> str)
    {
        cin >> val;
        cin.get();
        theK = MyK.add(val);
        cout << theK << endl;
    }
    return 0;
}

 代码二

#include<iostream>
#include<algorithm>
using namespace std;
bool cmp(int x,int y)
{
	return x > y;
}
struct student
{
	int k;
	int num[100001] = {0};
	
	
	student(int k, int nums[],int n)
	{
		this->k = k;
		for (int i = 0; i < n; i++)
		{
			num[i] = nums[i];
		}
		
	}
	int add(int val,int n)
	{
		num[n] = val;
		n++;
		sort(num, num + n, cmp);
		return num[k];

	}

};

int n;

int main()
{
	int k = 0;
	cin >> k >> n;
	int arr[100001] = {0};
	for (int i = 0; i < n; i++)
	{
		cin >> arr[i];
	}
	student shit(k,arr,n);
	string temp;
	for (; cin >> temp;)
	{
		if (temp == "add")
		{

			int vital=0;
			cin >> vital;
			int temp1=shit.add(vital,n);
			cout << temp1+1 << endl;

		}
	}
	return 0;
}