Practice #2
Coddyの「ヒープと優先度付きキュー - データ構造シリーズ #7」コースのレッスン 11/14。
チャレンジ
簡単整数配列 arr と整数 k を受け取り、大きい方からk個の値を昇順にソートして返す関数 kLargest を作成してください。
もし k が配列のサイズより大きい場合は、すべての値をソートして返してください。k が0または負の場合は、空の配列を返してください。
必ず MinHeap クラスを使用してください(minheap.<ext> で提供されています)。結果を計算するために、sort、slice、または標準ライブラリのヒープライブラリなどの言語組み込み機能を使用しないでください。
自分で試してみよう
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <stdbool.h>
#include "solution.h"
int main() {
char line[8192];
char tline[64];
if (!fgets(line, sizeof(line), stdin)) line[0] = '\0';
if (!fgets(tline, sizeof(tline), stdin)) tline[0] = '\0';
int arr[4096];
int n = 0;
char* tok = strtok(line, " \t\r\n");
while (tok) { arr[n++] = atoi(tok); tok = strtok(NULL, " \t\r\n"); }
int k = atoi(tline);
int rs = 0;
int* r = kLargest(arr, n, k, &rs);
for (int idx = 0; idx < rs; idx++) { if (idx > 0) printf(" "); printf("%d", r[idx]); }
printf("\n");
return 0;
}