P9836 种树

容易想到分解因数。

对于一个数 p p p 的因数个数,假设它可以被分解质因数成 a 1 i 1 a 2 i 2 a 3 i 3 ⋯ a k c k a_1^{i_1} a_2^{i_2} a_3^{i_3}\cdots a_k^{c_k} a1i1a2i2a3i3⋯akck 的形式,则其因数个数为 ( i 1 + 1 ) ( i 2 + 1 ) ( i 3 + 1 ) ⋯ ( i k + 1 ) (i_1+1)(i_2+1)(i_3+1)\cdots(i_k+1) (i1+1)(i2+1)(i3+1)⋯(ik+1)。

我们对序列 p p p 和 w w w 质因数分解之后再考虑这个问题。对于每次从 w w w 拆分出的一个质因子 A A A,我们假设一棵树 p i p_i pi,原来它的贡献为 x x x,对于该棵树高的质因数拆分中质因子 A A A 的出现次数为 t t t,则乘上该质因子之后它的贡献会变为 x ⋅ t + 2 t + 1 x\cdot \dfrac{t+2}{t+1} x⋅t+1t+2,容易证明分子越小即 t t t 越小对答案的贡献越大。

由于数据范围是 1 0 4 10^4 104,质因数的上界是很有限的。设 a ( i , j ) a(i,j) a(i,j) 表示 i j i^j ij 在所有树高的质因数拆分中出现了多少次,每一次取到的一个 w w w 的质因子 A A A,把它计算到当前存在且 k k k 最小的 A k A^k Ak 上即可。

注意筛 n n n 的质因数要筛到 n n n。

cpp 复制代码
#include <bits/stdc++.h>
using namespace std;
#define int long long

const int maxn=1e4+5;
int a[maxn][30],ans=1,p[maxn],mod=998244353,n,w;

void calc(int x)
{
	for(int i=2;i<=max(x,w);i++)
	{
		int cnt=0;
		while(x%i==0) x/=i,cnt++;
		a[i][cnt]++;
		ans=ans*(cnt+1)%mod;
		// cout<<cnt<<endl;
	}
}

signed main()
{
	cin>>n>>w;
	for(int i=1;i<=n;i++) cin>>p[i],calc(p[i]);
	// for(int i=1;i<=n;i++) cout<<a[i][1]<<endl;
	for(int A=2;A<=w;A++)
		while(w%A==0)
		{
			int k=0;
			while(!a[A][k]) k++;	
			ans/=(k+1),ans%=mod,ans*=(k+2),ans%=mod;
			w/=A,a[A][k]--,a[A][k+1]++;
			// cout<<w<<endl;
		}	
	cout<<ans;
	return 0;
}
相关推荐
AI成长日志17 分钟前
【笔面试算法学习专栏】双指针专题·简单难度两题精讲:167.两数之和II、283.移动零
学习·算法·面试
旖-旎31 分钟前
分治(库存管理|||)(4)
c++·算法·leetcode·排序算法·快速选择算法
青稞社区.34 分钟前
ICLR‘26 Oral | 当 LLM Agent 在多轮推理中迷失时:T3 如何让强化学习重新学会主动推理
人工智能·算法·agi
春花秋月夏海冬雪38 分钟前
代码随想录刷题 - 贪心Part1
java·算法·贪心·代码随想录
环黄金线HHJX.42 分钟前
Tuan符号系统重塑智能开发
开发语言·人工智能·算法·编辑器
汀、人工智能1 小时前
[特殊字符] 第2课:字母异位词分组
数据结构·算法·链表·数据库架构··字母异位词分组
小O的算法实验室2 小时前
2026年SEVC,面向主动成像卫星任务规划问题的群体智能与动态规划混合框架,深度解析+性能实测
算法·论文复现·智能算法·智能算法改进
网安INF2 小时前
数据结构第一章复习:基本概念与算法复杂度分析
数据结构·算法
幻风_huanfeng2 小时前
人工智能之数学基础:什么是凸优化问题?
人工智能·算法·机器学习·凸优化
三雷科技3 小时前
使用 `dlopen` 动态加载 `.so` 文件
开发语言·c++·算法