Practice #3
Coddyの「ヒープと優先度付きキュー - データ構造シリーズ #7」コースのレッスン 12/14。
チャレンジ
簡単整数配列 arr と正の整数 k を受け取り、k番目に小さい値(1から始まるインデックス)を返す関数 kthSmallest を作成してください。つまり、k = 1 は最小値を返し、k = 2 は2番目に小さい値を返します。
もし k が範囲外(1未満、または配列の長さより大きい)の場合は、-1 を返してください。
必ず 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 r = kthSmallest(arr, n, k);
printf("%d\n", r);
return 0;
}