結果
| 問題 | No.1013 〇マス進む |
| ユーザー |
 marurunn11
|
| 提出日時 | 2022-02-27 11:57:51 |
| 言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
| 結果 |
TLE
|
| 実行時間 | - |
| コード長 | 22,126 bytes |
| コンパイル時間 | 4,036 ms |
| コンパイル使用メモリ | 245,700 KB |
| 実行使用メモリ | 8,864 KB |
| 最終ジャッジ日時 | 2023-09-18 22:14:43 |
| 合計ジャッジ時間 | 23,334 ms |
|
ジャッジサーバーID （参考情報） |
judge14 / judge15 |
（要ログイン）
テストケース
テストケース表示| 入力 | 結果 | 実行時間 実行使用メモリ |
|---|---|---|
| testcase_00 | AC | 2 ms
4,380 KB |
| testcase_01 | AC | 1 ms
4,380 KB |
| testcase_02 | AC | 1 ms
4,380 KB |
| testcase_03 | AC | 3 ms
4,380 KB |
| testcase_04 | AC | 2 ms
4,376 KB |
| testcase_05 | AC | 3 ms
4,376 KB |
| testcase_06 | AC | 4 ms
4,380 KB |
| testcase_07 | AC | 4 ms
4,376 KB |
| testcase_08 | AC | 4 ms
4,380 KB |
| testcase_09 | AC | 5 ms
4,380 KB |
| testcase_10 | AC | 4 ms
4,376 KB |
| testcase_11 | AC | 3 ms
4,380 KB |
| testcase_12 | AC | 5 ms
4,380 KB |
| testcase_13 | AC | 62 ms
4,380 KB |
| testcase_14 | AC | 1,055 ms
4,376 KB |
| testcase_15 | TLE | - |
| testcase_16 | AC | 1,862 ms
4,376 KB |
| testcase_17 | TLE | - |
| testcase_18 | AC | 1,944 ms
4,380 KB |
| testcase_19 | AC | 543 ms
4,380 KB |
| testcase_20 | TLE | - |
| testcase_21 | -- | - |
| testcase_22 | -- | - |
| testcase_23 | -- | - |
| testcase_24 | -- | - |
| testcase_25 | -- | - |
| testcase_26 | -- | - |
| testcase_27 | -- | - |
| testcase_28 | -- | - |
| testcase_29 | -- | - |
| testcase_30 | -- | - |
| testcase_31 | -- | - |
| testcase_32 | -- | - |
| testcase_33 | -- | - |
| testcase_34 | -- | - |
| testcase_35 | -- | - |
| testcase_36 | -- | - |
| testcase_37 | -- | - |
| testcase_38 | -- | - |
| testcase_39 | -- | - |
| testcase_40 | -- | - |
| testcase_41 | -- | - |
| testcase_42 | -- | - |
| testcase_43 | -- | - |
| testcase_44 | -- | - |
| testcase_45 | -- | - |
| testcase_46 | -- | - |
| testcase_47 | -- | - |
| testcase_48 | -- | - |
| testcase_49 | -- | - |
| testcase_50 | -- | - |
| testcase_51 | -- | - |
| testcase_52 | -- | - |
| testcase_53 | -- | - |
| testcase_54 | -- | - |
| testcase_55 | -- | - |
| testcase_56 | -- | - |
| testcase_57 | -- | - |
| testcase_58 | -- | - |
| testcase_59 | -- | - |
| testcase_60 | -- | - |
| testcase_61 | -- | - |
| testcase_62 | -- | - |
| testcase_63 | -- | - |
| testcase_64 | -- | - |
ソースコード
#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#include "bits/stdc++.h"
#ifdef _MSC_VER
#include <intrin.h> //gcc上ではこれがあると動かない。__popcnt, umul128 等用のincludeファイル。
#define __builtin_popcount __popcnt
#define __builtin_popcountll __popcnt64
// 1 の位から何個 0 が連なっているか。(0 入れると壊れる)
inline unsigned int __builtin_ctz(unsigned int x) { unsigned long r; _BitScanForward(&r, x); return r; }
inline unsigned int __builtin_ctzll(unsigned long long x) { unsigned long r; _BitScanForward64(&r, x); return r; }
// 2進での leading 0 の個数。(0 入れると壊れる)
inline unsigned int __builtin_clz(unsigned x) { return (unsigned int)__lzcnt(x); }
inline unsigned int __builtin_clzll(unsigned x) { return (unsigned int)__lzcnt64(x); }
#pragma warning(disable : 4996)
#pragma intrinsic(_umul128)
#endif
//#include <atcoder/all>
//using namespace atcoder;
using namespace std;
//---------- 多倍長関連 ----------
//#include <boost/multiprecision/cpp_int.hpp>
//#include <boost/multiprecision/cpp_dec_float.hpp>
//using namespace boost::multiprecision;
typedef long long ll;
typedef long double ld;
//#define int long long
#define LL128 boost::multiprecision::int128_t
#define LL boost::multiprecision::cpp_int
#define LD100 boost::multiprecision::cpp_dec_float_100
#define LD50 boost::multiprecision::cpp_dec_float_50
#define rep(i, n) for(long long i = 0; i < (n); ++i)
#define REP(i, s, n) for(long long i = (s); i < (n); ++i)
#define rrep(i, n) for(long long i = (n) - 1; i >= 0; --i)
#define sqrt(d) pow((ld) (d), 0.50)
#define PII pair<int, int>
#define MP make_pair
#define PB push_back
#define ALL(v) v.begin(), v.end()
const int INF2 = std::numeric_limits<int>::max() / 2 - 100000000;
const long long INF = std::numeric_limits<long long>::max() / 2 - 100000000;
const ld pi = acos(-1);
//constexpr int MOD = 1000000007; //1e9 + 7
constexpr int MOD = 998244353; // 7 * 17 * 2^23 + 1
//---------- chmax, min 関連 ----------
template<class T> inline void chmax(T& a, T b) {
if (a < b) a = b;
}
template<class T> inline void chmin(T& a, T b) {
if (a > b) a = b;
}
//---------- 整数の根号関連 ----------
//res * res <= n なる最大の整数 res を返す。
template<typename T = long long>
T floor_sqrt(T n) {
assert(n >= 0);
T res = max((T)floor(sqrt(n)) - (T)2, (T)0);
while ((res + 1) * (res + 1) <= n) res++;
return res;
}
//res * res >= n なる最小の整数 res を返す。
template<typename T = long long>
T ceil_sqrt(T n) {
T res = floor_sqrt(n);
if (res * res == n) return res;
else return res + 1;
}
//---------- gcd, lcm ----------
template<typename T = long long>
T my_gcd(T a, T b) {
if (b == (T)0) return a;
return my_gcd<T>(b, a % b);
}
template<typename T = long long>
T my_lcm(T a, T b) {
return a / my_gcd<T>(a, b) * b;
}
// ax + by = gcd(a, b) を解く。返り値は、gcd(a, b)。
//但し、a, b が負である場合は、返り値が正であることは保障されない。
long long my_gcd_ext(long long a, long long b, long long& x, long long& y) {
if (b == 0) {
x = 1; y = 0;
return a;
}
long long tempo = my_gcd_ext(b, a % b, y, x);
//bx' + ry' = gcd(a, b) → (qb + r)x + by = gcd(a, b) に戻さないといけない。// (r = a % b)
//b(x' - qy') + (bq + r)y' = gcd(a, b) と同値変形できるから、
// x = y', y = x' - qy'
y -= (a / b) * x;
return tempo;
}
//中国式剰余の定理 (CRT)
// x = base1 (mod m1) かつ x = base2 (mod m2) を解く。
// リターン値を (r, m) とすると解は x = r (mod m) で、m = lcm(m1, m2)
// 解なしの場合は (0, -1) をリターン
pair<long long, long long> CRT(long long base1, long long m1, long long base2, long long m2) {
long long p, q;
long long gcd0 = my_gcd_ext(m1, m2, p, q);
if ((base2 - base1) % gcd0 != 0) return make_pair(0, -1);
long long lcm0 = m1 * (m2 / gcd0); // 括弧がないとオーバーフローのリスクがある。
p *= (base2 - base1) / gcd0;
p %= (m2 / gcd0);
q *= (base2 - base1) / gcd0;
q %= (m1 / gcd0);
long long r = (base1 + m1 * p) % lcm0;
if (r < 0) r += lcm0;
return make_pair(r, lcm0);
}
//M を法として、a の逆元を返す。但し gcd(a, M) = 1。
long long my_invmod(long long a, long long M) {
long long x = 0, y = 0;
long long memo = my_gcd_ext(a, M, x, y);
assert(memo == 1LL);
x %= M;
if (x < 0) x += M;
return x;
}
//繰り返し2乗法
//N^aの、Mで割った余りを求める。
template<typename T = long long>
T my_pow(T N, T a, T M) {
T tempo;
if (a == 0) {
return 1;
}
else {
if (a % 2 == 0) {
tempo = my_pow(N, a / 2, M);
return (tempo * tempo) % M;
}
else {
tempo = my_pow(N, a - 1, M);
return (tempo * N) % M;
}
}
}
// 繰り返し2乗法
// T = modint でも動く。
template<typename T = long long>
T my_pow(T N, long long a) {
T tempo;
if (a == 0) {
return 1;
}
else {
if (a % 2 == 0) {
tempo = my_pow(N, a / 2);
return (tempo * tempo);
}
else {
tempo = my_pow(N, a - 1);
return (tempo * N);
}
}
}
// base を底としたときの、n の i桁目を、v.at(i) に入れる。
vector<signed> ll_to_vector(signed base, long long n) {
long long tempo = n;
long long tempo2 = n; //桁数を求めるときに使う
signed n_digit = 1;
while (tempo2 >= base) {
tempo2 /= base;
n_digit++;
}
vector<signed> v(n_digit, 0); // v のサイズを適切に調整。
long long denominator = my_pow<long long>((long long)base, (long long)(n_digit - 1));
for (signed i = 0; i < n_digit; i++) {
v.at(i) = tempo / denominator;
tempo -= v.at(i) * denominator;
denominator /= base;
}
return v;
}
// M 桁に足りない場合、0 を追加して強制的に M 桁にする。
vector<signed> ll_to_vector(signed base, long long n, int M) {
vector<signed> v = ll_to_vector(base, n);
//assert((int)v.size() <= M);
if ((int)v.size() >= M) return v;
else {
int diff = M - v.size();
vector<signed> res(diff, 0);
for (int i = 0; i < (int)v.size(); i++) res.emplace_back(v.at(i));
return res;
}
}
//エラトステネスの篩で、prime で ないところに false を入れる。O(n loglog n)
// T = int (defalt, sieve が ll で間に合うことはないので。)
// vector<char> に替えるとむしろ遅くなる。
template<typename T = int>
vector<bool> sieve_bool(T N) {
vector<bool> res(N + 1, true);
res.at(0) = false;
res.at(1) = false;
for (T i = 2; 2 * i <= N; i++) {
res.at(2 * i) = false;
}
for (T i = 3; i * i <= N; i += 2) {
//ここからは奇数のみ探索。i の倍数に false を入れる。
if (res.at(i)) {
T j = i * i; // i^2 未満の i の倍数には、すでに false が入っているはず。
while (j <= N) {
res.at(j) = false;
j += 2 * i;
}
}
}
return res;
};
// n + 1 の サイズの vector を返す。res.at(i) には、i の 1 以外で最小の約数を入れる。
// res.at(i) == i で、i != 0, 1 なら i は素数。
// 2e8 なら、2.3 ~ 2.4 sec 程度で終わる。sieve_bool は 0.7 sec なので、3 倍強遅い。ll にすると、3.2 sec に伸びてしまう。
// T = int (defalt, sieve が ll で間に合うことはないので。)
template<typename T = int>
vector<T> sieve(T n) {
n++; // n まで判定する。配列サイズは +1。
vector<T> res(n, 0);
for (T i = 1; i < n; i++) {
if (i % 2 == 0) res.at(i) = 2; // 偶数をあらかじめ処理。
else res.at(i) = i; // 奇数には自分自身を入れる。
}
for (T i = 3; i * i < n; i += 2) {
//ここからは奇数のみ探索。i の倍数に i を入れる。
if (res.at(i) == i) {
T j = i * i; // i^2 未満の i の倍数には、すでに最小の約数が入っているはず。
while (j < n) {
if (res.at(j) == j) res.at(j) = i;
j += 2 * i;
}
}
}
return res;
};
//O (sqrt(n)) で素数判定する用。
bool is_prime(long long N) {
if (N == 1 || N == 0) return false;
if (N == 2 || N == 3) return true;
if (N % 2 == 0) return false;
if (N % 3 == 0) return false;
for (long long i = 1; (6 * i + 1) * (6 * i + 1) <= N; ++i) {
if (N % (6 * i + 1) == 0) return false;
}
for (long long i = 0; (6 * i + 5) * (6 * i + 5) <= N; ++i) {
if (N % (6 * i + 5) == 0) return false;
}
return true;
}
// 素因分解アルゴリズム (O(sqrt(N)) → O(N^0.25) のρ法も持っている。
// T = long long (defalt)
template<typename T = long long>
map<T, T> PrimeFactor(T N) {
map<T, T> res;
T i = 2;
while (i * i <= N) {
while (N % i == 0) {
res[i]++;
N /= i;
}
i += 1 + (i % 2); //i == 2 の場合だけ +1, その他の場合は +2
}
if (N > 1) res[N]++; //sqrt((元の N)) より大きな素因数は高々1つしかない。
return res;
}
//関数 sieve で得た、vector min_factor を持ってるときに、素因数分解を高速で行うための関数。
// T = int (defalt, sieve が ll で間に合うことはないので。)
template<typename T = int>
map<T, T> PrimeFactor2(T target, vector<T>& min_factor) {
map<T, T> res;
if (min_factor.empty() || (T)min_factor.size() - 1 < target) min_factor = sieve<T>(target);
while (target > 1) {
res[min_factor[target]]++;
target /= min_factor[target];
}
return res;
}
//約数全列挙を O(sqrt(N)) で行うための関数。
vector<long long> count_dividers(long long target) {
vector <long long> dividers, tempo;
long long i = 1;
while (i * i < target + 1) {
if (target % i == 0) {
dividers.push_back(i);
if (i < target / i) tempo.push_back(target / i); // if節がないと、平方数の時、sqrt(target) がダブルカウントされる。
}
i++;
}
for (long long j = 0; j < (long long)tempo.size(); j++) {
dividers.push_back(tempo.at(tempo.size() - 1 - j));
}
return dividers;
}
//関数 sieve で得た、vector min_factor を持ってるときに、約数全列挙を高速で行うための関数。
// T = int (defalt, sieve が ll で間に合うことはないので。)
template<typename T = int>
vector<T> count_dividers2(T target, vector<T>& min_factor, bool is_sort = false) {
vector<T> dividers = { 1 };
map<T, T> memo = PrimeFactor2<T>(target, min_factor);
for (auto&& iter = memo.begin(); iter != memo.end(); iter++) {
vector <T> tempo = dividers;
for (T k = 0; k < (T)tempo.size(); k++) {
T times = 1;
for (T j = 1; j <= (iter->second); j++) {
times *= iter->first;
dividers.push_back(tempo[k] * times);
}
}
}
if (is_sort) sort(dividers.begin(), dividers.end()); //sortしないと小さい順に並ばないが、必要ないなら消しても良い。
return dividers;
}
class UnionFind {
public:
vector<int> parent;
vector<int> rank;
vector<int> v_size;
UnionFind(int N) : parent(N), rank(N, 0), v_size(N, 1) {
rep(i, N) {
parent[i] = i;
}
}
int root(int x) {
if (parent[x] == x) return x;
return parent[x] = root(parent[x]); //経路圧縮
}
void unite(int x, int y) {
int rx = root(x);
int ry = root(y);
if (rx == ry) return; //xの根とyの根が同じなので、何もしない。
if (rank[rx] < rank[ry]) {
parent[rx] = ry;
v_size[ry] += v_size[rx];
}
else {
parent[ry] = rx;
v_size[rx] += v_size[ry];
if (rank[rx] == rank[ry]) rank[rx]++;
}
}
bool same(int x, int y) {
return (root(x) == root(y));
}
int count_tree() {
int N = parent.size();
int res = 0;
rep(i, N) {
if (root(i) == i) res++;
}
return res;
}
int size(int x) {
return v_size[root(x)];
}
};
// 幾何。二点間距離。
ld calc_dist(int x1, int y1, int x2, int y2) {
int tempo = (x1 - x2) * (x1 - x2) + (y1 - y2) * (y1 - y2);
ld res = sqrt((ld)tempo);
return res;
}
//ランレングス圧縮
vector<pair<int, char>> RunLength(string S) {
int N = S.size();
vector<pair<int, char>> memo;
if (N == 1) {
memo.push_back(MP(1, S.at(0)));
return memo;
}
int tempo = 1;
for (int i = 1; i < N; i++) {
if (i != N - 1) {
if (S.at(i) == S.at(i - 1)) tempo++;
else {
memo.push_back(MP(tempo, S.at(i - 1)));
tempo = 1;
}
}
else {
if (S.at(i) == S.at(i - 1)) {
tempo++;
memo.push_back(MP(tempo, S.at(i - 1)));
}
else {
memo.push_back(MP(tempo, S.at(i - 1)));
memo.push_back(MP(1, S.at(i)));
}
}
}
return memo;
}
void printf_ld(ld res) {
printf("%.12Lf\n", res);
//cout << std::fixed << std::setprecision(12) << res << endl;
}
template<typename T = long long>
void print_vec(vector<T> v) {
int N = v.size();
rep(i, N) {
if (i != N - 1) cout << v.at(i) << " ";
else cout << v.at(i) << endl;
}
}
template<typename T = long long>
void print_vec(deque<T> v) {
int N = v.size();
rep(i, N) {
if (i != N - 1) cout << v.at(i) << " ";
else cout << v.at(i) << endl;
}
}
//mint 構造体。自動で mod を取る。
//m は定数である必要があるので入力を用いることはできない。
template<int m, typename T> class mint {
public:
T val;
//---------- コンストラクタ ----------
constexpr mint(T v = 0) noexcept : val(v% m) {
if (val < 0) val += m;
}
//------------------------------ 二項演算子のオーバーロード ------------------------------
constexpr mint& operator += (const mint& r) noexcept {
val += r.val;
if (val >= m) val -= m;
return *this;
}
constexpr mint& operator -= (const mint& r) noexcept {
val -= r.val;
if (val < 0) val += m;
return *this;
}
constexpr mint& operator *= (const mint& r) noexcept {
val = val * r.val % m;
return *this;
}
constexpr mint& operator /= (const mint& r) noexcept {
//a * u + b * v = 1 を互除法で解く。但し、gcd(a, m) == 1 でなければならない。
T a = r.val, b = m, u = 1, v = 0;
while (b) {
T q = a / b;
a -= q * b; swap(a, b); //互除法。余りをとって swap。
u -= q * v; swap(u, v);
}
val = val * u % m;
if (val < 0) val += m;
return *this;
}
constexpr mint operator + (const mint& r) const noexcept { return mint(*this) += r; }
constexpr mint operator - (const mint& r) const noexcept { return mint(*this) -= r; }
constexpr mint operator * (const mint& r) const noexcept { return mint(*this) *= r; }
constexpr mint operator / (const mint& r) const noexcept { return mint(*this) /= r; }
constexpr bool operator == (const mint& r) const noexcept {
return this->val == r.val;
}
constexpr bool operator != (const mint& r) const noexcept {
return this->val != r.val;
}
//------------------------------ 単項演算子のオーバーロード ------------------------------
//---------- 前置インクリメントのオーバーロード ----------
constexpr mint operator ++() noexcept { this->val++; if (this->val >= m) this->val -= m; return mint(*this); }
constexpr mint operator --() noexcept { this->val--; if (this->val < 0) this->val += m; return mint(*this); }
//---------- 後置インクリメントのオーバーロード ----------
constexpr mint operator++(signed) noexcept { mint temp(val); ++val; if (val >= m) val -= m; return temp; }
constexpr mint operator--(signed) noexcept { mint temp(val); --val; if (val < 0) val += m; return temp; }
constexpr mint operator -() const noexcept { return mint(-val); }
//---------- 入出力のオーバーロード ----------
friend constexpr ostream& operator << (ostream& os, const mint<m, T>& x) noexcept {
return os << x.val;
}
friend istream& operator >> (istream& is, mint<m, T>& x) noexcept {
T init_val;
is >> init_val;
x = mint<m, T>(init_val);
return is;
}
//---------- 繰り返し二乗法 ----------
constexpr mint<m, T> modpow(const mint<m, T>& a, T n) noexcept {
if (n == 0) return 1;
auto t = modpow(a, n / 2);
t = t * t;
if (n & 1) t = t * a;
return t;
}
//---------- 逆元 ----------
constexpr mint<m, T> inverse() noexcept {
mint<m, T> e(1);
return e / (*this);
}
};
using modint = mint<MOD, long long>;
vector<modint> dp_fac;
vector<modint> dp_fac_inv;
// 階乗。x ! まで計算する。結果は dp (デフォルトで dp_fac<modint>) に保存する。
// long long にするためには、第二引数に vector<long long> を指定する必要がある。20 ! = 2.43e18 まで long long に入る。
template<typename T = modint>
T factorial(int x, vector<T>& dp = dp_fac) {
assert(x >= 0);
//既に計算済み
if ((int)dp.size() > x) {
return dp.at(x);
}
int n = dp.size();
//dp サイズを x + 1 に伸ばす。
for (int i = n; i < x + 1; i++) {
if (i == 0) dp.push_back((T)1);
else dp.push_back(dp.back() * i);
}
return dp.at(x);
}
template<typename T = modint>
T factorial_inv(int x, vector<T>& dp = dp_fac_inv) {
assert(x >= 0);
//既に計算済み
if ((int)dp.size() > x) {
return dp.at(x);
}
int n = dp.size();
//dp サイズを x + 1 に伸ばす。
for (int i = n; i < x + 1; i++) {
if (i == 0) dp.push_back((T)1);
else dp.push_back(dp.back() / i);
}
return dp.at(x);
}
// 二項係数 N_C_a
template<typename T = modint, typename U = int>
T my_comb(U N, U a, vector<T>& dp = dp_fac, vector<T>& dp_inv = dp_fac_inv) {
if (N < a) return (T)0;
T ans = factorial<T>(N, dp);
ans *= factorial_inv<T>(a, dp_inv);
ans *= factorial_inv<T>(N - a, dp_inv);
return ans;
}
ld now_clock() {
ld t = (ld)clock() / (ld)CLOCKS_PER_SEC;
return t;
}
//周期問題を解く。
//get_max追加。loopの先頭に何か追加する機能を削除。
template<typename U = long long>
class rho_loop {
private:
int s = -1; // loopの開始位置
int period = -1; //loopの周期
U sum_period = 0; //1周期での和
U sum_base = 0; //周期開始前の和
U sum_offset = 0; //開始位置をずらした際の sum の補正。
deque<U> index_vec; //同じ要素が二つ出現したら、ループする。
deque<U> body; //本体のベクトル
deque<U> sum; // body の和
unordered_map<U, int> cnt; //index の出現回数を記録。(2になったらループ発生)
unordered_map<U, int> index_mp;
public:
//コンストラクタ
rho_loop() : s(-1), period(-1) {};
//配列にappendして、ループ検出した場合 true を返す。
//基本的に最初にこれを true を返すまでやって、配列を構築する。
bool append(U _index, U y) {
assert(s == -1 && period == -1);
index_vec.push_back(_index);
body.push_back(y);
if (index_vec.size() == 1) sum.push_back(y);
else sum.push_back(y + sum.back());
++cnt[_index];
if (cnt[_index] == 2) {
s = index_mp[_index];
period = index_vec.size() - 1 - index_mp[_index];
assert(body.at(s) == body.back());
index_vec.pop_back();
body.pop_back();
sum.pop_back();
if (s != 0) sum_base = sum.at(s - 1);
else sum_base = 0;
sum_period = sum.back() - sum_base;
return true;
}
else {
index_mp[_index] = (int)index_vec.size() - 1;
return false;
}
}
U get_index(long long i) {
assert(s != -1 && period != -1);
if (i <= s) return index_vec.at(i);
i -= s;
i %= period;
return index_vec.at(s + i);
}
U get(long long i) {
assert(s != -1 && period != -1);
if (i <= s) return body.at(i);
i -= s;
i %= period;
return body.at(s + i);
}
//始めから i 番目 (0-indexed) の sum
U get_sum(long long i) {
assert(s != -1 && period != -1);
if (i <= s) return sum.at(i) + sum_offset;
i -= s;
U res = sum_base;
res += sum_period * (i / period);
i %= period;
res += sum.at(s + i) - sum_base;
res += sum_offset;
return res;
}
//始めから i 番目 (0-indexed) の sum の要素までで、最大のものを返す。
U get_sum_max(long long i) {
assert(s != -1 && period != -1);
if (sum_period <= 0) {
U MAX = -(std::numeric_limits<U>::max() / 2 - 100000);
for(int k = 0; k < min((long long)sum.size(), i + 1); ++k) MAX = max(MAX, sum.at(k));
return MAX + sum_offset;
}
else {
U MAX = -(std::numeric_limits<U>::max() / 2 - 100000);
for (int k = 0; k < min((long long)s, i + 1); ++k) MAX = max(MAX, sum.at(k));
i -= s;
if (i < 0) return MAX + sum_offset;
long long N = i / period;
long long Q = i % period;
for (int k = 0; k < min((long long)period, i + 1); ++k) {
if (k <= Q) MAX = max(MAX, sum.at(k + s) + sum_period * N);
else MAX = max(MAX, sum.at(k + s) + sum_period * (N - 1));
}
return MAX + sum_offset;
}
}
//1つ進めて先頭の要素を削除。
void proceed() {
assert(s != -1 && period != -1);
sum_offset -= body.front();
if (s > 0) {
s--;
index_vec.pop_front();
body.pop_front();
sum.pop_front();
}
else {
int i = index_vec.front();
int x = body.front();
index_vec.pop_front();
body.pop_front();
sum.pop_front();
index_vec.push_back(i);
body.push_back(x);
if (index_vec.size() == 1) sum.push_back(x);
else sum.push_back(x + sum.back());
}
}
//デバック用
void show_debug() {
assert(s != -1 && period != -1);
cout << "i: "; show_index();
cout << "body: "; show_body();
cout << "sum: "; show_sum();
}
private:
//デバック用
void show_index() {
rep(i, (int)index_vec.size()) {
cout << index_vec.at(i) << " ";
}
cout << endl;
}
//デバック用
void show_body() {
rep(i, (int)body.size()) {
cout << body.at(i) << " ";
}
cout << endl;
}
//デバック用
void show_sum() {
rep(i, (int)sum.size()) {
cout << sum.at(i) + sum_offset << " ";
}
cout << endl;
}
};
signed main() {
long long N, K;
cin >> N >> K;
vector<long long> P(N);
rep(i, N) cin >> P.at(i);
vector<int> res(N, -1);
rep(i, N) {
if (res.at(i) != -1) continue;
int now = i;
rho_loop<long long> rl;
while (true) {
bool flag = rl.append(now, P.at(now));
now += P.at(now);
now %= N;
if (flag) break;
}
//rl.show_debug();
int from = i;
while (true) {
res.at(from) = rl.get_sum(K - 1) + from + 1;
rl.proceed();
from = rl.get_index(0);
if(res.at(from) != -1) break;
}
}
rep(i, N) {
cout << res.at(i) << endl;
}
}