結果

問題 No.915 Plus Or Multiple Operation
コンテスト
ユーザー ningenMe
提出日時 2019-10-12 17:13:56
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 1,850 bytes
コンパイル時間 1,610 ms
コンパイル使用メモリ 173,716 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-11-07 03:19:31
合計ジャッジ時間 2,218 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 10
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

template <class T> void chmin(T& a, const T b){a=min(a,b);}

int main() {
    int Q;
    cin >> Q;
    assert(1<=Q&&Q<=20);
    while(Q--){
        long long A,B,C,D; cin >> A >> B >> C;
        assert(1<=A&&A<=1000000000LL);
        assert(1<=B&&B<=1000000000LL);
        assert(1<=C&&C<=1000000000LL);

        //C=1の場合はコーナーケース
        if(C==1){
            cout << -1 << endl;
            continue;
        }
        
        //AをC進数に直す O(logA)
        vector<long long> bit;
        D = A;
        while(D>0){
            bit.push_back(D%C);
            D /= C;
        }
        reverse(bit.begin(),bit.end());
        
        int M = bit.size();
        vector<long long> sum(M+1,0);    //上からi桁目まで構築したときの和: 1-indexed
        vector<long long> powC(M,1);     //Cの冪数
        vector<long long> dp(M+1,A);     //dp_i:上からi桁目まで構築したときの最小コスト
        
        //sum,powを前計算
        for(int i = 1; i < M; ++i) powC[i] = powC[i-1]*C;
        for(int i = 1; i <= M; ++i) {
            for(int j = 0; j < i; ++j) {
                sum[i] += bit[j]*powC[i-j-1];
            }
        }

        dp[0] = 0;
        for(int i = 0; i < M; ++i) {
            //積で次に遷移
            if(sum[i]*C==sum[i+1]) chmin(dp[i+1],dp[i]+1);
            //和で次に遷移
            if(sum[i]+C-1>=sum[i+1]) chmin(dp[i+1],dp[i]+1);
            //積＋和で次に遷移
            if(sum[i]*C+C-1>=sum[i+1]) chmin(dp[i+1],dp[i]+2);
            //和＋和で次の次に遷移
            if(i+2<=M&&sum[i]+2*C-2>=sum[i+2]) chmin(dp[i+2],dp[i]+2);
        }

        long long ans = dp[M]*B;
        assert(1<=ans&&ans<=1000000000000000000LL);
        cout << ans << endl;
    }
    return 0;
}
0