NEUOJ 1343 Eat walnuts (容斥原理 + 逆元 + 唯一分解定理)

最新推荐文章于 2022-03-08 01:00:18 发布

Chen丶HC

最新推荐文章于 2022-03-08 01:00:18 发布

阅读量477

点赞数

分类专栏：容斥原理

本文链接：https://blog.csdn.net/chcnsn/article/details/80085370

版权

该博客介绍了如何利用容斥原理、逆元和唯一分解定理来解决NEUOJ 1343题，即求[1, n]中与m互素的数的和及其平方的和。通过初始化质数数组、获取输入数m的因数以及计算平方和与和的逆元，最终实现求解目标并输出结果。" 108904251,9271216,Kotlin方法与Lambda表达式详解,"['Android开发', 'Kotlin', '方法', 'Lambda']

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

经典问题，求[1, n]中与m互素的数的和与平方的和

#include <iostream>
#include <vector>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <string>
#include <map>
#include <queue>

using namespace std;
typedef long long ll;
vector<int> primes, factors;
const int maxn = 10000;
const int mod = 1000000007;
ll n, m;
bool vis[maxn + 5];

void init() {
    for(int i = 2; i <= maxn; ++i) {
        for(int j = i * i; j <= maxn; j += i) {
            vis[j] = true;
        }
    }
    for(int i = 2; i <= maxn; ++i)
        if(!vis[i])  {
            primes.push_back(i);
            //printf("prime = %d\n", i);
        }
}

void get_factors(ll m) {
    factors.clear();
    for(int i = 0; i < primes.size(); ++i) {
        int p = primes[i];
        if(p > m) break;
        if(m % p == 0) {