結果

問題 No.78 クジ付きアイスバー
ユーザー fantasiabaetica
提出日時 2018-06-29 15:40:19
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,313 bytes
コンパイル時間 535 ms
コンパイル使用メモリ 66,252 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-06-30 23:47:53
合計ジャッジ時間 1,607 ms
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1 WA * 3
other AC * 4 WA * 31
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <iostream>
typedef long long int ll;
using namespace std;

ll n, k;
int cost = 0;
int stock = 0;
char box[50];

void buy_box(){
    for (int i = 0; i < n; i++){
        int ice = box[i];
        cout << ice << " " << cost << " " << stock << endl;
        if (stock > 0){
            stock -= 1;
        } else {
            cost += 1;
        }
        if (ice == 2){
            stock += 2;
        } else if (ice == 1){
            stock += 1;
        } 
    }
}

int buy(int rest){
    int last_cost = 0;
    for (int i = 0; i < rest; i++){
        int ice = box[i];
        if (stock > 0){
            stock -= 1;
        } else {
            last_cost += 1;
        }
        if (ice == 2){
            stock += 2;
        } else if (ice == 1){
            stock += 1;
        } 
    }
    return last_cost;
}

int main(){
    cin >> n >> k;
    string ices;
    cin >> ices;
    for (int i = 0; i < n; i++){
        int ice = ices[i] - '0';
        box[i] = ice;
    }
    if (n > k){
        cout << buy(k) << endl;    
        return 0;
    }
    buy_box();
    if (cost <= stock){
        cout << cost << endl;
    } else {
        int cost_sum = cost;
        cost_sum += (k / n - 1) * (cost - stock);
        cost_sum += buy(n % k);
        cout << cost_sum << endl;
    }
    
    return 0;
}
0