【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」の内容は以下になります。