結果
問題 | No.78 クジ付きアイスバー |
ユーザー |
![]() |
提出日時 | 2020-07-25 11:18:26 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,580 bytes |
コンパイル時間 | 1,028 ms |
コンパイル使用メモリ | 90,348 KB |
最終ジャッジ日時 | 2025-01-12 05:24:16 |
ジャッジサーバーID (参考情報) |
judge3 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 25 WA * 10 |
ソースコード
#include <iostream>#include <vector>#include <queue>#include <set>using namespace std;using ll = long long;using vi = vector<int>;#define rep(i,n) for(int i=0,_i=(n);i<_i;++i)int main() {int N, K; cin >> N >> K;string S; cin >> S;vi G(N, 0);rep(i, N) {if (S[i] == '0') {G[i] = (i+1) % N;continue;}int atari = S[i] - '0';int j = 0;while (atari > 0) {--atari;++j;atari += S[(i+j)%N] - '0';if (j > N)break;}if (j > N)G[i] = N;elseG[i] = (i+j+1) % N;}if (G[0] == N) {cout << 1 << endl;return 0;}set<int> st;queue<int> q;q.push(0);vector<pair<int, int>> v;int ans = 0;while (true) {int p = q.front(); q.pop();if (p == N) {cout << ans << endl;return 0;}q.push(G[p]);++ans;int distance = (G[p] > p ? G[p] - p : N + (G[p] - p));K -= distance;if (K <= 0) {cout << ans << endl;return 0;}if (st.find(p) == st.end()) {v.emplace_back(p, distance);st.insert(p);continue;}vi circle;ll sum = 0;rep(i, v.size()) {if (v[i].first == p) {while(i < (int)v.size()) {circle.push_back(v[i].second);sum += v[i].second;++i;}break;}}ans += (K / sum) * circle.size();K %= sum;rep(i, circle.size()) {if (K <= 0)break;K -= circle[i];++ans;}cout << ans << endl;break;}return 0;}