■0203 A New Plan of Aizu Ski Resort
dp[j][i]=(i,j)に到達するまでの滑り方のパターン数としてDP.
dp[0][i]=
0 - (i,0)に障害物がある
1 - (i,0)に障害物がない
あとは,説明通りに更新していく.dp[n]まで用意しておいて,dp[n-1]とdp[n]の総和が答え.
import java.util.*;
import java.lang.*;
import java.math.*;
import java.io.*;
import static java.lang.Math.*;
import static java.util.Arrays.*;
public class Main{
Scanner sc=new Scanner(System.in);
int INF=1<<28;
double EPS=1e-9;
int m, n;
int[][] a;
void run(){
for(;;){
m=sc.nextInt();
n=sc.nextInt();
if((m|n)==0){
break;
}
a=new int[n][m];
for(int j=0; j<n; j++){
for(int i=0; i<m; i++){
a[j][i]=sc.nextInt();
}
}
solve();
}
}
void solve(){
long[][] dp=new long[n+1][m];
for(int i=0; i<m; i++){
if(a[0][i]==0){
dp[0][i]=1;
}
}
for(int j=0; j<n-1; j++){
for(int i=0; i<m; i++){
switch(a[j][i]){
case 0:
if(i-1>=0&&a[j+1][i-1]==0)
dp[j+1][i-1]+=dp[j][i];
if(a[j+1][i]!=1)
dp[j+1][i]+=dp[j][i];
if(i+1<m&&a[j+1][i+1]==0)
dp[j+1][i+1]+=dp[j][i];
break;
case 2:
if(j>=n-2||a[j+2][i]!=1)
dp[j+2][i]+=dp[j][i];
break;
}
}
}
long ans=0;
for(int j=n-1; j<n+1; j++){
for(int i=0; i<m; i++){
ans+=dp[j][i];
}
}
println(ans+"");
}
void debug(Object... os){
System.err.println(Arrays.deepToString(os));
}
void print(String s){
System.out.print(s);
}
void println(String s){
System.out.println(s);
}
public static void main(String[] args){
// System.setOut(new PrintStream(new BufferedOutputStream(System.out)));
new Main().run();
}
}
0 件のコメント:
コメントを投稿