mayoko’s diary

プロコンとかいろいろ。

yukicoder No.176 2種類の切手

星3つはありそう。

問題:No.176 2種類の切手 - yukicoder

解法:uwiさんのコードを参考にさせていただきました。
Bが1000より大きい時は,T/Bは10^6以下に抑えられるので,p*A+q*Bのqの値を全探索して最適な値を調べれば良い。
Bが1000より小さい時に同じやり方でやるとTLEで死ぬ可能性がある。そこでA*iをBで割ったあまりを考える。もしある値tをBで割ったあまりがA*iをBで割ったあまりで表現できるなら,t=A*i+B*q(qは適当な整数)とすることでtという値を作れることになる。これを利用して解を求める。
以下ソースコード

const ll INF = 1ll<<51;

ll dist[2200];

int main() {
    ll A, B, T;
    cin >> A >> B >> T;
    if (B >= 1000) {
        ll ans = INF;
        for (ll i = 0; i <= 1000000; i++) {
            ll rest = T-i*B;
            if (rest < 0) ans = min(ans, i*B);
            else {ll tmp = rest/A*A+i*B;
                if (tmp < T) tmp += A;
                ans = min(ans, tmp);
            }
        }
        cout << ans << endl;
    } else {
        for (int i = 0; i < 2200; i++) dist[i] = INF;
        for (ll i = 0; i < 2200; i++) {
            ll m = (A*i)%B;
            dist[m] = min(dist[m], i);
        }
        while (1) {
            ll m = T % B;
            if (A*dist[m] <= T) {
                cout << T << endl;
                break;
            }
            T++;
        }
    }
    return 0;
}