Menu
Coddy logo textTech

Practice #2

Coddyの「ヒープと優先度付きキュー - データ構造シリーズ #7」コースのレッスン 11/14。

challenge icon

チャレンジ

簡単

整数配列 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;
}

ヒープと優先度付きキュー - データ構造シリーズ #7のすべてのレッスン