結果

問題 No.78 クジ付きアイスバー
ユーザー ngtkana
提出日時 2020-04-08 00:06:11
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 3 ms / 5,000 ms
コード長 875 bytes
コンパイル時間 2,037 ms
コンパイル使用メモリ 196,180 KB
最終ジャッジ日時 2025-01-09 14:57:56
ジャッジサーバーID
（参考情報） 		judge2 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
#define ALL(v) std::begin(v),std::end(v)
using lint=long long;
using ld=long double;
int main(){
    std::cin.tie(nullptr);std::ios_base::sync_with_stdio(false);
    std::cout.setf(std::ios_base::fixed);std::cout.precision(15);
    lint n,k;std::cin>>n>>k;
    std::vector<lint>a(2*n),cost(2*n+1),stock(2*n+1);
    for(lint i=0;i<n;i++){
        char c;std::cin>>c;
        a.at(i)=a.at(i+n)=c-'0';
    }
    for(lint i=0;i<2*n;i++){
        lint s=stock.at(i);
        lint c=cost.at(i);
        if(s)s--;
        else c++;
        s+=a.at(i);
        stock.at(i+1)=s;
        cost.at(i+1)=c;
    }
    if(auto found=std::find(stock.begin()+n,stock.end(),0);found==a.end()){
        std::cout<<cost.at(std::min(k,2*n))<<'\n';
    }else{
        lint q=std::max(0ll,k/n-1);
        std::cout<<(cost.at(2*n)-cost.at(n))*q+cost.at(k-n*q)<<'\n';
    }
}
0