結果
問題 |
No.1013 〇マス進む
|
ユーザー |
|
提出日時 | 2020-03-20 22:01:38 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 99 ms / 2,000 ms |
コード長 | 1,043 bytes |
コンパイル時間 | 1,519 ms |
コンパイル使用メモリ | 174,688 KB |
実行使用メモリ | 42,196 KB |
最終ジャッジ日時 | 2024-12-15 05:50:32 |
合計ジャッジ時間 | 6,309 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 62 |
ソースコード
#include <bits/stdc++.h> using namespace std::literals::string_literals; using i64 = std::int_fast64_t; using std::cout; using std::cerr; using std::endl; using std::cin; template<typename T> std::vector<T> make_v(size_t a){return std::vector<T>(a);} template<typename T,typename... Ts> auto make_v(size_t a,Ts... ts){ return std::vector<decltype(make_v<T>(ts...))>(a,make_v<T>(ts...)); } int main() { int n, k; scanf("%d%d", &n, &k); std::vector<int> p(n); for(int i = 0; i < n; i++) scanf("%d", &p[i]); auto dp = make_v<int>(32, n); auto ep = make_v<i64>(32, n); for(int i = 0; i < n; i++) dp[0][i] = (i + p[i]) % n; for(int i = 0; i < n; i++) ep[0][i] = p[i]; for(int i = 0; i < 31; i++) { for(int j = 0; j < n; j++) { dp[i + 1][j] = dp[i][dp[i][j]]; ep[i + 1][j] = ep[i][j] + ep[i][dp[i][j]]; } } for(int i = 0; i < n; i++) { i64 ans = i + 1; int now = i; for(int j = 31; j >= 0; j--) { if(!(k >> j & 1)) continue; ans += ep[j][now]; now = dp[j][now]; } printf("%lld\n", ans); } return 0; }