結果

問題 No.1019 最小格子三角形
ユーザー maspymaspy
提出日時 2020-03-07 22:27:11
言語 C++11
(gcc 11.4.0)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 1,709 bytes
コンパイル時間 3,204 ms
コンパイル使用メモリ 148,476 KB
実行使用メモリ 17,340 KB
最終ジャッジ日時 2023-08-30 12:32:47
合計ジャッジ時間 6,677 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 WA -
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 TLE -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;

#define MAX 1000010
bool is_prime[MAX];
ll mu[MAX];

vector<int> prime_table()
{
    is_prime[2] = true;
    for (int n = 3; n < MAX; n += 2)
    {
        is_prime[n] = true;
    }
    for (int p = 3; p * p < MAX; p += 2)
    {
        if (is_prime[p])
        {
            int q = p * p;
            for (int i = q; i < MAX; i += p)
            {
                is_prime[i] = false;
            }
        }
    }
    vector<int> primes;
    for (int p = 0; p < MAX; p++)
    {
        if (is_prime[p])
        {
            primes.emplace_back(p);
        }
    }
    return primes;
}

void mobius_table(vector<int> primes)
{
    for (int n = 1; n < MAX; n++)
    {
        mu[n] = 1;
    }
    for (int i = 0; i < primes.size(); i++)
    {
        ll p = primes[i];
        for (int j = p; j < MAX; j += p)
        {
            mu[j] *= (-1);
        }
        ll q = p * p;
        if (q >= MAX)
            continue;
        for (int j = q; j < MAX; j += q)
        {
            mu[j] = 0;
        }
    }
}


ll F(ll N, ll MOD){
    ll x_max = sqrt(N);
    ll S = 0;
    for (ll x = 1; x <= x_max; x++)
    {
        ll y_max = sqrt(N - x * x);
        S += x * (1 + 2 * y_max);
        S %= MOD;
    }
    return 4 * S % MOD;
}

ll f(ll N, ll MOD)
{
    ll ret = 0;
    for (ll d = 1; d <= N; d++)
    {
        ll n = (N * N) / (d * d);
        ret += F(n, MOD) * mu[d] * d % MOD;
    }
    ret %= MOD;
    if(ret < 0)
        ret += MOD;
    return ret;
}

int main()
{
    ll N;
    ll MOD;
    cin >> N >> MOD;
    vector<int> primes = prime_table();
    mobius_table(primes);
    cout << f(N, MOD) << endl;
    return 0;
}
0