結果

問題 No.1019 最小格子三角形
ユーザー maspy
提出日時 2020-03-07 22:27:11
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 1,709 bytes
コンパイル時間 1,716 ms
コンパイル使用メモリ 160,176 KB
実行使用メモリ 25,520 KB
最終ジャッジ日時 2025-01-02 01:38:25
合計ジャッジ時間 23,098 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 3
other WA * 10 RE * 8 TLE * 6
権限があれば一括ダウンロードができます

ソースコード

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