読者です 読者をやめる 読者になる 読者になる

丸三角四角

IT業界にしがみつく新人SEが立派なプログラマになろうともがく奮闘記

【Java】車両入替

今回の問題は、車両がどの順番で出庫されるかを出力する問題です。

車のナンバーが入力として与えられ、0が入力されたら後から入庫した車を出庫していきます。どの順番で車が出庫されたかを求める問題です。この問題はキューを使えば簡単に求めることができます。以下コードです。

import static java.lang.Integer.parseInt;
import static java.lang.System.out;
import java.util.Deque;
import java.util.LinkedList;
import volume0.BaseExe;

public class SwitchingRailroadCars extends BaseExe {

    public static void main(String[] args) {
        new SwitchingRailroadCars().exeSysIn();
    }

    @Override
    protected void execute() {
        Deque<Integer> inCars = new LinkedList<>();
        while (judgeInData()) {
            setCars(parseInt(getInData()), inCars);
        }
    }

    private void setCars(int carNumber, Deque<Integer> inCars) {
        if (carNumber == 0) {
            out.println(inCars.poll());
        } else {
            inCars.addFirst(carNumber);
        }
    }
}

0が入力されたらキューの先頭にある車のナンバーを表示しています。0以外はキューの先頭に追加しています。テストコードは以下になります。

import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import volume0.BaseTest;

public class SwitchingRailroadCarsTest extends BaseTest {

    private SwitchingRailroadCars src;

    @Before
    public void setUp() throws Exception {
        super.setUp();
        src = new SwitchingRailroadCars();
    }

    @After
    public void tearDown() throws Exception {
        super.tearDown();
    }

    @Test
    public void test0001() {
        src.exeFileIn("./data/volume0_0013/in.txt");
        assertOutList("./data/volume0_0013/out.txt");
    }
}

テストドライバ内で使用している「in.txt」・「out.txt」の内容は以下になります。

in.txt
1
6
0
8
10
0
0
0
out.txt
6
10
8
1

今回の問題は以下にあります。

車両入れ替え | Aizu Online Judge

アルゴリスムで困った時は、以下の本を参考にしてます。Javaに置き換えるのが少々難解ですが、、、

固有処理の基底クラスは以下の記事に記載していますので、参考にしてください。

dadainu.hateblo.jp

テストドライバで継承しているクラスは以下の記事に記載していますので、参考にしてみてください。

dadainu.hateblo.jp