yukicoder No.176 2種類の切手
星3つはありそう。
解法: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; }