Javaの解答例：ABC088B - Card Game for Two , AtCoder Beginners Selection

はじめに

自分用の備忘録です。

問題文

以下のリンクを参照ください。

atcoder.jp

自分の解答例

import java.util.ArrayList;
import java.util.Comparator;
import java.util.List;
import java.util.Scanner;

public class Main {

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);

        int N = sc.nextInt();
        List<Integer> A = new ArrayList<Integer>();
        for (int i = 1; i <= N; i++) {
            A.add(sc.nextInt());
        }

        A.sort(Comparator.reverseOrder());

        int aliceScore = 0;
        int bobScore = 0;

        for (int i = 1; i <= A.size(); i++) {
            if (i % 2 == 1) {
                aliceScore += A.get(i - 1);
            } else {
                bobScore += A.get(i - 1);
            }
        }
        int ans = aliceScore - bobScore;
        System.out.println(ans);
    }

}

公式のコードをJavaに変換

import java.util.*;

public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int N = scanner.nextInt();
        Integer[] a = new Integer[N];
        for (int i = 0; i < N; i++) {
            a[i] = scanner.nextInt();
        }

        Arrays.sort(a, Collections.reverseOrder());  // a[0:N] を大きい順にソート
        int Alice = 0;
        int Bob = 0;
        for (int i = 0; i < N; i++) {
            if (i % 2 == 0) { // Alice のターン
                Alice += a[i];
            } else { // Bob のターン
                Bob += a[i];
            }
        }
        System.out.println(Alice - Bob);
    }
}

このJavaのコードは、C++のコードと同じく、ABC 088 B - Card Game for Two の問題を解くためのものです。配列 a を大きい順にソートし、AliceとBobが交互にカードを取っていきます。最終的にAliceとBobが取ったカードの数値の差を出力します。このコードは、標準入力からNとaの値を読み取り、結果を標準出力に出力します。

参考文献

qiita.com