[Java] 藍橋杯ADV-194 演算法提高 盾神與積木遊戲

柳婼發表於2018-10-10

問題描述
最近的m天盾神都去幼兒園陪小朋友們玩去了~
每個小朋友都拿到了一些積木,他們各自需要不同數量的積木來拼一些他們想要的東西。但是有的小朋友拿得多,有的小朋友拿得少,有些小朋友需要拿到其他小朋友的積木才能完成他的大作。如果某個小朋友完成了他的作品,那麼他就會把自己的作品推倒,而無私地把他的所有積木都奉獻出來;但是,如果他還沒有完成自己的作品,他是不會把積木讓出去的喲~
盾神看到這麼和諧的小朋友們感到非常開心,於是想幫助他們所有人都完成他們各自的作品。盾神現在在想,這個理想有沒有可能實現呢?於是把這個問題交給了他最信賴的你。

輸入格式
第一行為一個數m。
接下來有m組資料。每一組的第一行為n,表示這天有n個小朋友。接下來的n行每行兩個數,分別表示他現在擁有的積木數和他一共需要的積木數。

輸出格式
輸出m行,如果第i天能順利完成所有作品,輸出YES,否則輸出NO。
樣例輸入
2
2
2 2
1 3
3
1 5
3 3
0 4
樣例輸出
YES
NO
資料規模和約定
1<=n<=10000,1<=m<=10。

package adv194;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;

public class Main {

    public static void main(String[] args) throws IOException {
        BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(System.in));
        int m = Integer.parseInt(bufferedReader.readLine());
        for (int i = 0; i < m; i++) {
            int n = Integer.parseInt(bufferedReader.readLine());
            Child[] childs = new Child[n];
            for (int j = 0; j < n; j++) {
                String[] t = bufferedReader.readLine().split(" ");
                childs[j] = new Child(Integer.parseInt(t[0]), Integer.parseInt(t[1]));
            }

            System.out.println(bankAlgorithm(childs));
        }
        bufferedReader.close();
    }

    private static String bankAlgorithm(Child[] childs) {
        Arrays.sort(childs);
        int currentNeed = 0;
        for (int i = 0; i < childs.length; i++) {
            if (childs[i].need <= 0) {
                currentNeed += childs[i].have;
            } else { // childs[i].need > 0
                if (currentNeed >= childs[i].need) {
                    currentNeed += childs[i].have;
                } else {
                    return "NO";
                }
            }
        }
        return "YES";
    }
}

class Child implements Comparable<Child> {
    int have;
    int total;
    int need;

    public Child(int have, int total) {
        this.have = have;
        this.total = total;
        this.need = total - have;
    }

    @Override
    public int compareTo(Child child) {
        return need - child.need;
    }

}

 

相關文章