結果

問題 No.1019 最小格子三角形
ユーザー maspymaspy
提出日時 2020-03-26 13:57:09
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 136 ms / 2,000 ms
コード長 1,772 bytes
コンパイル時間 1,246 ms
コンパイル使用メモリ 148,676 KB
実行使用メモリ 12,996 KB
最終ジャッジ日時 2023-08-30 12:33:48
合計ジャッジ時間 4,425 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge13
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 21 ms
12,660 KB
testcase_01 AC 20 ms
12,684 KB
testcase_02 AC 21 ms
12,828 KB
testcase_03 AC 22 ms
12,684 KB
testcase_04 AC 21 ms
12,628 KB
testcase_05 AC 20 ms
12,816 KB
testcase_06 AC 21 ms
12,996 KB
testcase_07 AC 21 ms
12,752 KB
testcase_08 AC 21 ms
12,624 KB
testcase_09 AC 21 ms
12,636 KB
testcase_10 AC 21 ms
12,796 KB
testcase_11 AC 21 ms
12,616 KB
testcase_12 AC 21 ms
12,824 KB
testcase_13 AC 108 ms
12,628 KB
testcase_14 AC 133 ms
12,604 KB
testcase_15 AC 130 ms
12,816 KB
testcase_16 AC 107 ms
12,628 KB
testcase_17 AC 121 ms
12,660 KB
testcase_18 AC 111 ms
12,684 KB
testcase_19 AC 122 ms
12,624 KB
testcase_20 AC 109 ms
12,608 KB
testcase_21 AC 115 ms
12,800 KB
testcase_22 AC 132 ms
12,660 KB
testcase_23 AC 136 ms
12,684 KB
testcase_24 AC 136 ms
12,820 KB
testcase_25 AC 136 ms
12,636 KB
testcase_26 AC 136 ms
12,808 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

typedef long long ll;

#define MAX 1000010
const int MOD = 998244353;

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 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 S;
}

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

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