Lady Layton with Math(杜教筛)

Lady Layton with Math

https://ac.nowcoder.com/acm/problem/200008

Lady Layton with Math

图片说明

/*
  Author : lifehappy
*/
#include <bits/stdc++.h>

using namespace std;

typedef long long ll;

const int N = 2e6 + 10, mod = 1e9 + 7;

int prime[N], cnt, n;

ll phi[N];

bool st[N];

void init() {
    phi[1] = 1;
    for(int i = 2; i < N; i++) {
        if(!st[i]) {
            prime[++cnt] = i;
            phi[i] = i - 1;
        }
        for(int j = 1; j <= cnt && 1ll * i * prime[j] < N; j++) {
            st[i * prime[j]] = 1;
            if(i % prime[j] == 0) {
                phi[i * prime[j]] = phi[i] * prime[j];
                break;
            }
            phi[i * prime[j]] = phi[i] * (prime[j] - 1);
        }
    }
    for(int i = 1; i < N; i++) {
        phi[i] = (phi[i] + phi[i - 1]) % mod;
    }
}

unordered_map<int, ll> ans_s;

ll Djs(int n) {
    if(n < N) return phi[n];
    if(ans_s.count(n)) return ans_s[n];
    ll ans = 1ll * n * (n + 1) / 2 % mod;
    for(int l = 2, r; l <= n; l = r + 1) {
        r = n / (n / l);
        ans = (ans - (r - l + 1) * Djs(n / l) % mod + mod) % mod;
    }
    return ans_s[n] = ans;
}

int main() {
    // freopen("in.txt", "r", stdin);
    // freopen("out.txt", "w", stdout);
    // ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
    int T;
    init();
    scanf("%d", &T);
    while(T--) {
        ll ans = 0;
        scanf("%d", &n);
        for(int l = 1, r; l <= n; l = r + 1) {
            r = n / (n / l);
            ans = (ans + 1ll * (Djs(r) - Djs(l - 1)) * (2ll * Djs(n / l) - 1) % mod + mod) % mod;
        }
        printf("%lld\n", ans);
    }
    return 0;
}
全部评论
强!
点赞 回复 分享
发布于 2020-11-08 00:39

相关推荐

宇智波爱学习:我还没收到笔试
投递荣耀等公司10个岗位
点赞 评论 收藏
分享
尊尼获获:闺蜜在哪?
点赞 评论 收藏
分享
如题如果提出了一个薪资,A不成功,会有可能被取消offer吗
爱打瞌睡的柯基:最想去你们公司 但是别家开的高一些,希望能申请高一点 不管结果如何都谢谢你
点赞 评论 收藏
分享
评论
3
收藏
分享
牛客网
牛客企业服务