結果

問題 No.896 友達以上恋人未満
ユーザー kimiyukikimiyuki
提出日時 2019-09-27 22:47:16
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,987 ms / 3,500 ms
コード長 2,557 bytes
コンパイル時間 1,997 ms
コンパイル使用メモリ 207,320 KB
実行使用メモリ 135,224 KB
最終ジャッジ日時 2024-09-25 00:06:11
合計ジャッジ時間 7,478 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 2 ms
6,816 KB
testcase_02 AC 2 ms
6,812 KB
testcase_03 AC 2 ms
6,940 KB
testcase_04 AC 2 ms
6,940 KB
testcase_05 AC 389 ms
36,768 KB
testcase_06 AC 790 ms
36,760 KB
testcase_07 AC 422 ms
36,732 KB
testcase_08 AC 401 ms
36,736 KB
testcase_09 AC 965 ms
69,640 KB
testcase_10 AC 1,987 ms
135,224 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#define REP(i, n) for (int i = 0; (i) < (int)(n); ++ (i))
#define REP3(i, m, n) for (int i = (m); (i) < (int)(n); ++ (i))
#define REP_R(i, n) for (int i = (int)(n) - 1; (i) >= 0; -- (i))
#define REP3R(i, m, n) for (int i = (int)(n) - 1; (i) >= (int)(m); -- (i))
#define ALL(x) std::begin(x), std::end(x)
#define dump(x) cerr << #x " = " << x << endl
using namespace std;
template <class T> using reversed_priority_queue = priority_queue<T, vector<T>, greater<T> >;
template <class T, class U> inline void chmax(T & a, U const & b) { a = max<T>(a, b); }
template <class T, class U> inline void chmin(T & a, U const & b) { a = min<T>(a, b); }
template <typename X, typename T> auto make_table(X x, T a) { return vector<T>(x, a); }
template <typename X, typename Y, typename Z, typename... Zs> auto make_table(X x, Y y, Z z, Zs... zs) { auto cont = make_table(y, z, zs...); return vector<decltype(cont)>(x, cont); }
template <typename T> ostream & operator << (ostream & out, vector<T> const & xs) { REP (i, (int)xs.size() - 1) out << xs[i] << ' '; if (not xs.empty()) out << xs.back(); return out; }

int main() {
    // input
    int m, n, mulX, addX, mulY, addY, MOD; cin >> m >> n >> mulX >> addX >> mulY >> addY >> MOD;
    vector<int> X(m);
    vector<int> Y(m);
    vector<int> A(m);
    vector<int> B(m);
    REP (i, m) cin >> X[i];
    REP (i, m) cin >> Y[i];
    REP (i, m) cin >> A[i];
    REP (i, m) cin >> B[i];

    // solve
    int mask = MOD - 1;
    vector<int64_t> z(MOD);
    REP (i, m) {
        z[X[i]] += Y[i];
    }
    int64_t x_i = X.back();
    int64_t y_i = Y.back();
    REP3 (i, m, n) {
        x_i = (x_i * mulX + addX) & mask;
        y_i = (y_i * mulY + addY) & mask;
        z[x_i] += y_i;
    }
    vector<int64_t> memo(100000, -1);
    auto go = [&](int64_t step) {
        if (step < memo.size() and memo[step] != -1) return memo[step];
        int64_t cnt = 0;
        for (int64_t x = 0; x < MOD; x += step) {
            cnt += z[x];
        }
        if (step < memo.size()) memo[step] = cnt;
        return cnt;
    };

    // output
    int64_t acc = 0;
    REP (j, m) {
        int64_t cnt = go(A[j]) - go((int64_t)A[j] * B[j]);
        acc ^= cnt;
        cout << cnt << endl;
    }
    int64_t a_j = A.back();
    int64_t b_j = B.back();
    REP3 (j, m, n) {
        a_j = ((a_j * mulX + addX + MOD - 1) & mask) + 1;
        b_j = ((b_j * mulY + addY + MOD - 1) & mask) + 1;
        int64_t cnt = go(a_j) - go(a_j * b_j);
        acc ^= cnt;
    }
    cout << acc << endl;

    return 0;
}
0