結果
問題 | No.896 友達以上恋人未満 |
ユーザー | maspy |
提出日時 | 2020-05-07 18:44:54 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 830 ms / 3,500 ms |
コード長 | 2,073 bytes |
コンパイル時間 | 660 ms |
コンパイル使用メモリ | 76,148 KB |
実行使用メモリ | 103,516 KB |
最終ジャッジ日時 | 2024-07-03 09:33:33 |
合計ジャッジ時間 | 9,267 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 616 ms
89,708 KB |
testcase_01 | AC | 602 ms
90,848 KB |
testcase_02 | AC | 621 ms
89,924 KB |
testcase_03 | AC | 594 ms
90,712 KB |
testcase_04 | AC | 607 ms
90,240 KB |
testcase_05 | AC | 661 ms
91,404 KB |
testcase_06 | AC | 830 ms
89,700 KB |
testcase_07 | AC | 681 ms
90,172 KB |
testcase_08 | AC | 660 ms
90,268 KB |
testcase_09 | AC | 661 ms
89,836 KB |
testcase_10 | AC | 705 ms
103,516 KB |
ソースコード
#include <iostream> #include <vector> #include<cassert> #define U 16777217 typedef long long ll; using namespace std; bool is_prime[U]; vector<int> primes; ll dp[U]; ll X[1010]; ll Y[1010]; ll A[1010]; ll B[1010]; int M, N, mX, aX, mY, aY, MOD; void make_prime_table() { for (int i = 2; i < U; i++) { is_prime[i] = true; } for (int p = 2; p < U; p++) { if (p * p >= U) { break; } for (int n = p * p; n < U; n += p) { is_prime[n] = false; } } primes.push_back(2); for (int p = 3; p < U; p += 2) { if (is_prime[p]) { primes.push_back(p); } } } void get_input() { cin >> M >> N >> mX >> aX >> mY >> aY >> MOD; for (int i = 0; i < M; i++) { cin >> X[i]; } for (int i = 0; i < M; i++) { cin >> Y[i]; } for (int i = 0; i < M; i++) { cin >> A[i]; } for (int i = 0; i < M; i++) { cin >> B[i]; } } void XY() { for (int i = 0; i < M; i++) { dp[X[i]] += Y[i]; } ll x = X[M - 1]; ll y = Y[M - 1]; ll mask = MOD - 1; for (int i = M; i < N; i++) { x = (x * mX + aX & mask); y = (y * mY + aY & mask); dp[x] += y; } } void zeta() { for (int p : primes) { for (int i = (U - 1) / p; i > 0; i--) { dp[i] += dp[p * i]; } } } ll f(ll a, ll b) { ll x; x = dp[a]; if (a * b < U) x -= dp[a * b]; return x; } void AB() { ll ans = 0; ll a, b, x; for (int i = 0; i < M; i++) { x = f(A[i], B[i]); cout << x << endl; ans ^= x; } a = A[M - 1]; b = B[M - 1]; ll mask = MOD - 1; for (int i = M; i < N; i++) { a = (a * mX + aX - 1 & mask) + 1; b = (b * mY + aY - 1 & mask) + 1; ans ^= f(a, b); } cout << ans << endl; } int main() { make_prime_table(); get_input(); XY(); zeta(); AB(); return 0; }