AOJ 1020 Cleaning Robot

問題

Cleaning Robot

解法

dpでやった、dp[i][y][x] := iターン目に(y,x)につく確率。

漸化式は

dp[i+1][次のy座標][次のx座標] += dp[i][今のy座標][今のx座標] * 0.25 if(進めるなら)
dp[i+1][今のy座標][今のx座標] += dp[i][今のy座標][今のx座標] * 0.25

こんな感じでいいのかな

コード

double dp[16][3][3];
int n, dx[] = {0,-1,0,1}, dy[] = {-1,0,1,0};
char cs, ct, cb;
int s, t, b;

int main(){
  while(scanf("%d", &n) && n){
    memset(dp, 0.0, sizeof(dp));
    scanf("\n%c %c %c", &cs, &ct, &cb);
    s = cs - 'A';
    t = ct - 'A';
    b = cb - 'A';
    dp[0][s/3][s%3] = 1.0;

    rep(i, n) rep(y, 3) rep(x, 3){
      if(dp[i][y][x] == 0) continue;
      rep(d, 4){
    int ny = y + dy[d], nx = x + dx[d];
    if(ny < 0 || nx < 0 || ny >= 3 || nx >= 3 || ny*3+nx == b) dp[i+1][y][x] += dp[i][y][x]*0.25;
    else dp[i+1][ny][nx] += dp[i][y][x]*0.25;
      }
    }

    printf("%.8lf\n", dp[n][t/3][t%3]);
  }
  return 0;
}