結果

問題 No.78 クジ付きアイスバー
ユーザー ふっぴーふっぴー
提出日時 2018-01-04 11:07:36
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,285 bytes
コンパイル時間 1,859 ms
コンパイル使用メモリ 171,088 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-12-23 03:00:41
合計ジャッジ時間 5,476 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 WA * 2
other AC * 9 WA * 15 RE * 11
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"
using namespace std;
     
#define DEBUG(x) cout<<#x<<": "<<x<<endl;
#define DEBUG_VEC(v) cout<<#v<<":";for(int i=0;i<v.size();i++) cout<<" "<<v[i]; cout<<endl
     
typedef long long ll;
#define vi vector<int>
#define vl vector<ll>
#define vii vector< vector<int> >
#define vll vector< vector<ll> >
#define vs vector<string>
#define pii pair<int,int>
#define pis pair<int,string>
#define psi pair<string,int>
#define pll pair<ll, ll>
const int inf = 1000000001;
const ll INF = 1e16;
#define MOD 1000000007
#define mod 1000000009
#define pi 3.14159265358979323846
#define Sp(p) cout<<setprecision(15)<<fixed<<p<<endl;
int dx[4] = { 1,0,-1,0 }, dy[4] = { 0,1,0,-1 };
int dx2[8] = { 1,1,0,-1,-1,-1,0,1 }, dy2[8] = { 0,1,1,1,0,-1,-1,-1 };


ll getsum(vs s) {
  ll res = 0;
  for (int i = 0; i < s.size(); i++) {
    res += stoll(s[i]);
  }
  return res;
}

int main() {
  int n, k, i, j;
  cin >> n >> k;
  string s;
  cin >> s;
  string s2 = s + s;
  vector<pii> cnst(2 * n + 1);
  for (i = 0; i < 2 * n; i++) {
    cnst[i + 1] = cnst[i];
    if (cnst[i].second == 0) {
      cnst[i + 1].first++;
    }
    else {
      cnst[i + 1].second--;
    }
    cnst[i + 1].second += s2[i] - '0';
  }
  if (k <= 2 * n) {
    cout << cnst[k].first << endl;
    return 0;
  }

  int firstcnt = cnst[n].first;
  int firststock = cnst[n].second;
  int secondcnt = cnst[2 * n].first;
  int secondstock = cnst[2 * n].second;

  // DEBUG(firstcnt);
  // DEBUG(firststock);
  // DEBUG(secondcnt);
  // DEBUG(secondstock);

  
  if (firststock == secondstock) {
    int ans = firstcnt;
    int nokori = k - n;
    DEBUG(nokori);
    ans += nokori / n * (secondcnt - firstcnt);
    nokori %= n;
    ans += cnst[nokori].first;
    int stock = secondstock;
    if (stock > 0) {
      for (i = 0; i < nokori; i++) {
	if (cnst[i].second == 0) {
	  ans--;
	  stock--;
	  if (stock == 0) {
	    break;
	  }
	}
      }
    }
    cout << ans << endl;
    return 0;
    
  }
  else {
    string S;
    for (i = 0; i < 60; i++) {
      S += s;
    }
    int stock = 0, cnt = 0;
    i = 0;
    while (stock < 55 || i < k) {
      if (stock == 0) {
	cnt++;
      }
      else {
	stock--;
      }
      stock += S[i] - '0';
      i++;
    }
    cout << cnt << endl;
    return 0;
  }
  
  
}
0