2023百度之星第三场 新的阶乘 题解（数论/分解质因数）

题目

 思路

用线性筛把每个合数的分解成 最小质数prims[j] 乘以  一个常数x（x可能是质数可能是合数）的形式，然后从后往前操作，把每个合数的x再按照最小质数prims[j] 乘以  一个常数x的方法分解，这样就能得到质数的次数。

代码

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int mod = 1e9 + 7, N = 1e7 + 10;
int n, st[N], is_primes[N], ans[N];
pair<int, int> pre[N];
vector<int>primes;
void get()
{
    for(int i = 2; i <= n; i ++)
    {
        if(!st[i]) primes.push_back(i), is_primes[i] = 1, pre[i] = {i, 1};
        for(int j = 0; primes[j] * i <= n; j ++)
        {
            int t = primes[j] * i;
            pre[t] = {primes[j], i};
            st[t] = 1;
            if(i % primes[j] == 0) break;
        }
    }
    for(int i = n; i >= 1; i --)
    {
        if(is_primes[i]) continue;
        auto [x, y] = pre[i];
        ans[x] += ans[i], ans[y] += ans[i];
    }
}
signed main()
{
	ios::sync_with_stdio(0);
	cin.tie(0); cout.tie(0);
    cin >> n;
    for(int i = 2; i <= n; i ++) ans[i] = n + 1 - i;
    get();
    cout << "f(" << n << ")=";
    int cnt = 0; 
    for(int i = 2; i <= n; i ++)
    {
        if(is_primes[i])
        {
            if(cnt) cout << "*";
            if(ans[i] == 1) cout << i;
            else cout << i << "^" << ans[i];
            cnt ++;
        }

    }
}