Kattis Problem - The Dragon of Loowater

Kattis Problem - The Dragon of Loowater

原题链接

题目类型：贪心、排序

题意

你的王国里有一条 n 个头的恶龙，你希望雇一些骑士把它杀掉（即砍掉所有的头）。村里有 m 个骑士可以雇佣，一个能力值为 x 的骑士可以砍掉恶龙一个直径不超过 x 的头，且需要支付 x 个金币。如何雇佣骑士才能砍掉恶龙所有的头，且需要支付的金币最少？注意，一个骑士只能砍一个头（且不能被雇佣两次）。

分析

将龙的每一个头的直径和骑士的身高从小到大排序，然后一个一个的砍（能砍的就砍，不能砍的就下一个）。最后判断能否将所有的龙头都砍完。

代码

public static void solve(int n, int m) throws IOException {
    int[] a = new int[n];
    int[] b = new int[m];
    
    for (int i = 0; i < n; i++) a[i] = nextInt();
    for (int i = 0; i < m; i++) b[i] = nextInt();
    
    if (m < n) {
        pw.println("Loowater is doomed!");
        return ;
    }
    
    Arrays.sort(a);
    Arrays.sort(b);
    
    int sum = 0, cnt = 0;
    for (int i = 0; i < m && cnt < n; i++) {
        if (b[i] >= a[cnt]) {
            sum += b[i];
            cnt++;
        }
    }
    
    pw.println(cnt == n ? sum : "Loowater is doomed!");
}