Problem:
パスタ (Pasta)

問題

あなたはパスタが大好物であり，毎日，晩御飯にパスタを作って食べている．あなたはトマトソース，クリームソース，バジルソースの 3 種類のパスタを作ることができる．

N 日間の晩御飯の予定を考えることにした．それぞれの日に 3 種類のパスタから 1 種類を選ぶ．ただし，同じパスタが続くと飽きてしまうので，3 日以上連続して同じパスタを選んではいけない．また，N 日のうちの K 日分のパスタはすでに決めてある．

入力として N の値と，K 日分のパスタの情報が与えられたとき，条件をみたす予定が何通りあるかを 10000 で割った余りを求めるプログラムを作成せよ．

入力

入力は K + 1 行からなる．

1 行目には 2 つの整数 N, K (3 ≦ N ≦ 100，1 ≦ K ≦ N) が空白を区切りとして書かれている．

1 + i 行目 (1 ≦ i ≦ K) には 2 つの整数 Ai, Bi (1 ≦ Ai ≦ N，1 ≦ Bi ≦ 3) が空白を区切りとして書かれている．これは，Ai 日目のパスタはすでに決まっており，Bi = 1 のときはトマトソースであり，Bi = 2 のときはクリームソースであり，Bi = 3 のときはバジルソースであることを表す．Ai (1 ≦ i ≦ K) は全て異なる．与えられる入力データにおいて，条件をみたす予定は 1 通り以上あることが保証されている．

出力

条件をみたす予定が何通りあるかを 10000 で割った余りを 1 行で出力せよ．

入出力例

入力例 1

5 3
3 1
1 1
4 2

出力例 1

6

入出力例 1 において，あなたは 5 日間の予定を考える．1 日目と 3 日目はトマトソースであり，4 日目はクリームソースである．また，3 日以上連続して同じパスタを選んではいけない．これらの条件をみたす予定は 6 通りある．

 1日目 

 2日目 

 3日目 

 4日目 

 5日目 

予定 1   

1

2

1

2

1

予定 2   

1

2

1

2

2

予定 3   

1

2

1

2

3

予定 4   

1

3

1

2

1

予定 5   

1

3

1

2

2

予定 6   

1

3

1

2

3

この表では，1 はトマトソースを，2 はクリームソースを，3 はバジルソースを表す．

入力例 2

20 5
10 2
4 3
12 1
13 2
9 1

出力例 2

2640

入出力例 2 において，条件をみたす予定は全部で 4112640 通りある．それを 10000 で割った余りである 2640 を出力する．

問題文と自動審判に使われるデータは、情報オリンピック日本委員会が作成し公開している問題文と採点用テストデータです。

Solution in JAVA (judge verdict: Accepted)

import java.util.*;
public class Main {

	public static void main(String[] args) {
		// TODO Auto-generated method stub
		// 0 = tomato2, 1 = tomato1, 2 = cream2, 3 = cream1,
		// 4 = basil2, 5 = basil1 ??¨??????
		Scanner sc = new Scanner(System.in);
		int N = sc.nextInt();
		int K = sc.nextInt();
		int ans = 0;
		int[] prog = new int[N];
		int[][] table = new int[6][N];
		for(int i = 0; i < K; i++) {
			int A = sc.nextInt() - 1;
			int B = sc.nextInt();
			prog[A] = B;
		}
		switch(prog[0]) {
			case 0:
				table[1][0] = 1;
				table[3][0] = 1;
				table[5][0] = 1;
				break;
			case 1: //tomato1
				table[1][0] = 1;
				break;
			case 2: //cream1
				table[3][0] = 1;
				break;
			case 3: //basil
				table[5][0] = 1;
				break;
		}
		for(int i = 1; i < N; i++) {
			switch(prog[i]) {
				case 0:
					table[0][i] = (table[1][i - 1]); //tomato2
					table[1][i] = (table[2][i - 1] + table[3][i - 1] + table[4][i - 1] + table[5][i - 1]) % 10000; //tomato1
					table[2][i] = (table[3][i - 1]); //cream2
					table[3][i] = (table[0][i - 1] + table[1][i - 1] + table[4][i - 1] + table[5][i - 1]) % 10000; //cream1
					table[4][i] = (table[5][i - 1]); //basil2
					table[5][i] = (table[0][i - 1] + table[1][i - 1] + table[2][i - 1] + table[3][i - 1]) % 10000; //basil1
					break;
				case 1:
					table[0][i] = (table[1][i - 1]); //tomato2
					table[1][i] = (table[2][i - 1] + table[3][i - 1] + table[4][i - 1] + table[5][i - 1]) % 10000; //tomato1
					break;
				case 2:
					table[2][i] = (table[3][i - 1]); //cream2
					table[3][i] = (table[0][i - 1] + table[1][i - 1] + table[4][i - 1] + table[5][i - 1]) % 10000; //cream1
					break;
				case 3:
					table[4][i] = (table[5][i - 1]); //basil2
					table[5][i] = (table[0][i - 1] + table[1][i - 1] + table[2][i - 1] + table[3][i - 1]) % 10000; //basil1
					break;
			}
		}
		for(int i = 0; i < 6; i++) ans += table[i][N - 1];
		System.out.println(ans % 10000);
	}

}