結果

問題 No.463 魔法使いのすごろく🎲
ユーザー r_dream0r_dream0
提出日時 2017-02-20 21:16:20
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,746 bytes
コンパイル時間 1,053 ms
コンパイル使用メモリ 83,300 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-12-30 10:54:37
合計ジャッジ時間 2,419 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 21 WA * 15
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <iomanip>
#include <vector>
#include <algorithm>
using namespace std;

int n, m;
vector<double> c;

int main() {
  cin >> n >> m;
  c = vector<double>(n);
  for(int i = 1; i < n - 1; i++) {
    cin >> c[i];
  }
  // 後ろからの期待値
  // x_i マス iに居る時の期待値
  vector<vector<double>> M(n, vector<double>(n + 1));
  // x_n = 0
  M[n-1][n-1] = 1;
  M[n-1][n] = 0;
  for(int i = 0; i < n - 1; i++) {
    for(int j = 1; j <= m; j++) {
      int to = i + j;
      if(to >= n) to = n - (to - n) - 1;
      M[i][to] = -1.0 / m;
    }
    M[i][i] = 1;
    M[i][n] = c[i];
  }
  // for(int i = 0; i < n; i++) {
  //   for(int j = 0; j <= n; j++) {
  //     cout 
  //       << M[i][j] << " ";
  //   }
  //   cout << endl;
  // }
  for(int i = 0; i < n; i++) {
    int pivot = i;
    for(int j = i; j < n; j++) {
      if(abs(M[j][i]) > abs(M[pivot][i])) pivot = j;
    }
    swap(M[pivot], M[i]);
    for(int j = i + 1; j <= n; j++) {
      M[i][j] /= M[i][i];
    }
    M[i][i] = 1;
    for(int j = 0; j < n; j++) {
      if(i == j) continue;
      double div = M[j][i] / M[i][i];
      for(int k = 0; k <= n; k++) {
        M[j][k] -= div * M[i][k];
      }
    }
  }
  vector<double> de(n), dp(n);
  for(int i = 0; i < n; i++) de[i] = M[i][n];
  for(int i = n - 1; i >= 0; i--) {
    if(i + m >= n) dp[i] = 0;
    else {
      double t1 = 0;
      for(int j = 1; j <= m; j++) {
        t1 += dp[j + i] + c[i + j];
      }
      dp[i] = t1 / m;
      for(int j = 1; j <= m; j++) {
        dp[i] = min(dp[i], c[i + j] + de[i + j]);
      }
    }
  }
  cout << fixed << setprecision(10) << min(de[0], dp[0]) << endl;
  //for(int i = 0; i< n; i++) {
  //  cout << dp[i] << "," << de[i] << endl;
  //}
}
0