結果

問題 No.896 友達以上恋人未満
ユーザー maspymaspy
提出日時 2020-05-07 18:44:54
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,308 ms / 3,500 ms
コード長 2,073 bytes
コンパイル時間 682 ms
コンパイル使用メモリ 76,464 KB
実行使用メモリ 103,904 KB
最終ジャッジ日時 2023-09-16 08:03:50
合計ジャッジ時間 12,758 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge11
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 892 ms
92,100 KB
testcase_01 AC 855 ms
92,928 KB
testcase_02 AC 816 ms
92,472 KB
testcase_03 AC 894 ms
91,428 KB
testcase_04 AC 880 ms
91,740 KB
testcase_05 AC 959 ms
90,872 KB
testcase_06 AC 1,308 ms
90,248 KB
testcase_07 AC 1,060 ms
90,368 KB
testcase_08 AC 927 ms
89,868 KB
testcase_09 AC 905 ms
89,760 KB
testcase_10 AC 929 ms
103,904 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#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;
}
0