結果
問題 | No.78 クジ付きアイスバー |
ユーザー |
|
提出日時 | 2020-12-26 17:30:18 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 2,527 bytes |
コンパイル時間 | 1,674 ms |
コンパイル使用メモリ | 131,216 KB |
最終ジャッジ日時 | 2025-01-17 07:31:25 |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 20 WA * 15 |
ソースコード
#include <iostream>#include <vector>#include <algorithm>#include <cmath>#include <queue>#include <string>#include <map>#include <set>#include <stack>#include <tuple>#include <deque>#include <array>#include <numeric>#include <bitset>#include <iomanip>#include <cassert>#include <chrono>#include <random>#include <limits>#include <iterator>#include <functional>#include <sstream>#include <fstream>#include <complex>#include <cstring>#include <unordered_map>using namespace std;using ll = long long;using P = pair<int, int>;constexpr int INF = 1001001001;constexpr int mod = 1000000007;// constexpr int mod = 998244353;template<class T>inline bool chmax(T& x, T y){if(x < y){x = y;return true;}return false;}template<class T>inline bool chmin(T& x, T y){if(x > y){x = y;return true;}return false;}int main(){ios::sync_with_stdio(false);cin.tie(nullptr);int n, K;string s;cin >> n >> K >> s;int free = 0, buy = 0;for(int i = 0; i < n; ++i){if(free == 0) buy += 1;else free -= 1;free += s[i] - '0';}if(free == 0){int ans = K / n * buy;K %= n;if(K){for(int i = 0; i < n; ++i){if(free == 0){ans += 1;if(--K == 0) break;}else free -= 1;free += s[i] - '0';}}cout << ans << endl;return 0;}vector<P> dat(1, {free, buy});for(int i = 0; i < 1000000; ++i){buy = 0;for(int i = 0; i < n; ++i){if(free == 0) buy += 1;else free -= 1;free += s[i] - '0';}if(buy == 0 || dat.back() == make_pair(free, buy)) break;dat.emplace_back(free, buy);}int sz = dat.size(), sum = 0;for(auto [f, b] : dat) sum += b;int ans = 0;if(K >= sz * n) ans = sum;else{int m = K / n;for(int i = 0; i < m; ++i){ans += dat[i].second;}K %= n;if(K){free = dat[m].first;for(int i = 0; i < n; ++i){if(free == 0){ans += 1;if(--K == 0) break;}else free -= 1;free += s[i] - '0';}}}cout << ans << endl;return 0;}