AIとプログラミング

2 secs 1024 MB

shinnshinn

Problem

Submissions

Test cases

Editorial

解説

まず全てのタスクをAIに任せます。精度の総和は $A$ の総和となります。そこから、 $i$ 番目のタスクを自分で行った場合、精度の総和は $B_i - A_i$ だけ変化します。

よって、 $B_i - A_i$ が正の値のうち大きい順に $K$ 個を $A$ の総和に足せば最大化を達成できます。

実装

xxxxxxxxxx
 
use proconio::*;
​
fn main() {
    input! {
        n: usize,
        k: usize,
        a: [i64; n],
        b: [i64; n],
    }
​
    let mut d = vec![0; n];
    for i in 0..n {
        d[i] = b[i] - a[i];
    }
​
    d.sort();
    d.reverse();
    
    let mut ans: i64 = a.iter().sum();
    for i in 0..k {
        ans += d[i].max(0);
    }
​
    println!("{}", ans);
}
​