結果

問題 No.78 クジ付きアイスバー
ユーザー dnish
提出日時 2018-06-22 21:05:16
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 5,000 ms
コード長 1,008 bytes
コンパイル時間 1,545 ms
コンパイル使用メモリ 167,688 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-06-30 18:05:54
合計ジャッジ時間 2,858 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"
#define REP(i,n,N) for(ll i=(n); i<(N); i++)
#define RREP(i,n,N) for(ll i=(N-1); i>=n; i--)
#define CK(n,a,b) ((a)<=(n)&&(n)<(b))
#define ALL(v) (v).begin(),(v).end()
#define p(s) cout<<(s)<<endl
#define p2(a,b) cout<<(a)<<" "<<(b)<<endl
#define v2(T) vector<vector<T>>
typedef long long ll;
using namespace std;
const ll mod= 1e9+7;

int dp[110];
int main(){
    int N,K;
    string s;
    cin>>N>>K;
    cin>>s;
    int cnt=s[0]-'0';

    dp[0]=1;
    REP(i,1,N){
        if(cnt>0){
            cnt--;
            dp[i] = dp[i-1];
        }else{
            dp[i] = dp[i-1]+1;
        }
        cnt+=s[i]-'0';
    }
    int ans = dp[min(K,N)-1];
    if(K<=N){
        p(ans);
        return 0;
    }
    K-=N;
    dp[N-1]=0;
    REP(i,0,N){
        if(cnt>0){
            cnt--;
            dp[i]=dp[(i-1+N)%N];
        }else{
            dp[i]=dp[(i-1+N)%N]+1;
        }
        cnt+=s[i]-'0';
    }
    ans+=dp[N-1]*(K/N);
    if(K%N>0)ans+=dp[K%N-1];
    p(ans);
    return 0;
}
0