Cod sursa(job #3228739)

Utilizator andreioneaAndrei Onea andreionea Data 10 mai 2024 20:46:16
Problema Statistici de ordine Scor 40
Compilator rs Status done
Runda Arhiva educationala Marime 0.94 kb
use std::{cmp::Reverse, collections::BinaryHeap, fs::File, io::{BufRead, BufReader, BufWriter, Write}};

fn main() {
    let mut input_file = BufReader::new(File::open("sdo.in").unwrap());
    let mut first_line = String::new();
    let mut second_line = String::new();
    let _ = input_file.read_line(&mut first_line).unwrap();
    let _ = input_file.read_line(&mut second_line).unwrap();
    let (n, k) = first_line.trim().split_once(' ').unwrap();
    let n = n.parse::<usize>().unwrap();
    let k = k.parse::<usize>().unwrap();
    let limit = n - k + 1;
    let mut heap = BinaryHeap::with_capacity(limit);
    for num in second_line.trim().split(' ') {
        let num = num.parse::<u32>().unwrap();
        if heap.len() == limit {
            heap.pop();
        }
        heap.push(Reverse(num));
    }
    let mut output_file = BufWriter::new(File::create("sdo.out").unwrap());
    let _ = writeln!(output_file, "{}", heap.pop().unwrap().0).unwrap();
}