■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 件のコメント:
コメントを投稿