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.Scanner;
public class Main{
	public static void main(String[] args){
		int N,K;
		Scanner sc = new Scanner(System.in);
		N=sc.nextInt();//何日間の予定を考えるか
		K=sc.nextInt();//何日間の予定が決まっているか
		int pasta[][]=new int[6][N];
		int pastaplan[]=new int[N];
		for(int i=0;i<K;i++){
			int day=sc.nextInt();//何日の予定が決まっているか
			pastaplan[day-1]=sc.nextInt();//その日のパスタは何か
		}
		if(pastaplan[0]==0){//もし１日目の予定が決まっていなければ
			pasta[1][0]=1;pasta[3][0]=1;pasta[5][0]=1;
			
		}else if(pastaplan[0]==1){
			pasta[1][0]=1;
		}else if(pastaplan[0]==2){
			pasta[3][0]=1;
		}else if(pastaplan[0]==3){
			pasta[5][0]=1;
		}
		for(int j=1;j<N;j++){
			if(pastaplan[j]==0){//j日の予定がきまってなければ
				pasta[0][j]=pasta[1][j-1];//２日連続の時には前日の１日目の値がそのままくる。
				pasta[2][j]=pasta[3][j-1];
				pasta[4][j]=pasta[5][j-1];
				pasta[1][j]=pasta[2][j-1]%10000+pasta[3][j-1]%10000+pasta[4][j-1]%10000+pasta[5][j-1]%10000;
				pasta[3][j]=pasta[0][j-1]%10000+pasta[1][j-1]%10000+pasta[4][j-1]%10000+pasta[5][j-1]%10000;
				pasta[5][j]=pasta[0][j-1]%10000+pasta[1][j-1]%10000+pasta[2][j-1]%10000+pasta[3][j-1]%10000;
			}else if(pastaplan[j]==1){//j日の予定がトマトだったら
				pasta[0][j]=pasta[1][j-1];
				pasta[1][j]=pasta[2][j-1]+pasta[3][j-1]+pasta[4][j-1]+pasta[5][j-1];
			}else if(pastaplan[j]==2){
				pasta[2][j]=pasta[3][j-1];
				pasta[3][j]=pasta[0][j-1]+pasta[1][j-1]+pasta[4][j-1]+pasta[5][j-1];
			}else{
				pasta[4][j]=pasta[5][j-1];
				pasta[5][j]=pasta[0][j-1]+pasta[1][j-1]+pasta[2][j-1]+pasta[3][j-1];
			}
		}
		int sum=pasta[0][N-1]%10000+pasta[1][N-1]%10000+pasta[2][N-1]%10000+pasta[3][N-1]%10000+pasta[4][N-1]%10000+pasta[5][N-1]%10000;
		System.out.println(sum%10000);
	}
}