結果

問題 No.1019 最小格子三角形
ユーザー maspymaspy
提出日時 2020-03-26 13:57:09
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 96 ms / 2,000 ms
コード長 1,772 bytes
コンパイル時間 1,312 ms
コンパイル使用メモリ 162,256 KB
実行使用メモリ 12,804 KB
最終ジャッジ日時 2024-06-10 12:35:16
合計ジャッジ時間 3,653 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 17 ms
12,804 KB
testcase_01 AC 20 ms
12,684 KB
testcase_02 AC 18 ms
12,804 KB
testcase_03 AC 19 ms
12,672 KB
testcase_04 AC 19 ms
12,804 KB
testcase_05 AC 18 ms
12,680 KB
testcase_06 AC 18 ms
12,688 KB
testcase_07 AC 19 ms
12,640 KB
testcase_08 AC 18 ms
12,680 KB
testcase_09 AC 19 ms
12,688 KB
testcase_10 AC 20 ms
12,668 KB
testcase_11 AC 18 ms
12,596 KB
testcase_12 AC 18 ms
12,788 KB
testcase_13 AC 77 ms
12,680 KB
testcase_14 AC 94 ms
12,708 KB
testcase_15 AC 90 ms
12,720 KB
testcase_16 AC 76 ms
12,640 KB
testcase_17 AC 84 ms
12,672 KB
testcase_18 AC 78 ms
12,704 KB
testcase_19 AC 87 ms
12,768 KB
testcase_20 AC 78 ms
12,628 KB
testcase_21 AC 82 ms
12,804 KB
testcase_22 AC 94 ms
12,732 KB
testcase_23 AC 96 ms
12,696 KB
testcase_24 AC 95 ms
12,672 KB
testcase_25 AC 96 ms
12,692 KB
testcase_26 AC 95 ms
12,680 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