[ARC116C]Multiple Sequences

最新推荐文章于 2022-03-18 21:55:48 发布

OneInDark

最新推荐文章于 2022-03-18 21:55:48 发布

阅读量197

点赞数

分类专栏： C++ 数学

本文链接：https://blog.csdn.net/qq_42101694/article/details/117233174

版权

C++ 同时被 2 个专栏收录

547 篇文章 3 订阅

订阅专栏

数学

280 篇文章 1 订阅

订阅专栏

本文介绍了一种在AtCoder限时竞赛中快速求解组合问题的方法，利用质因数分解和递推思想，将时间复杂度从O(mlog⁡nlog⁡m)降低到O(mlog⁡log⁡m)。通过枚举质因数和计数方案，展示了高效的算法实现和代码片段。

摘要由CSDN通过智能技术生成

题目

传送门 to AtCoder

思路

我直接联想到狄利克雷卷积，因为能过，就直接打了（毕竟这种限时赛嘛，快点做很重要）。当然那是 $\mathcal O(m\log n\log m)$ 的。

说具体点。考虑 $d p$ 嘛。那么每次转移相当于 $f_{ab}\leftarrow f_a\cdot I_b$ 其中 $I$ 为常函数 $1$ ，即 $I_x=1$ 。这就是狄利克雷卷积嘛。

没想到，有一个更快的做法。枚举 $a_n$ 之后，单独考虑每个质因数，然后考虑次数每次增加 $1$ 的位置。一共 $n$ 个位置，每个位置可以放若干个 “加一”，这个质因数的次数是 $k$ ，相当于 $n$ 个自然数相加为 $k$ ，所以方案数是
${k+n-1\choose n-1}$

我们需要枚举每个数的质因数，这是 $\mathcal O(m\log\log m)$ 的。由于 $n - 1$ 不变，我们计算 $\frac{(k+n-1)!}{k!\cdot (n-1)!}$ 显然可以 $\mathcal O(k=\log m)$ 的递推，于是总复杂度是
$\mathcal O(m\log\log m)$

哪怕 $n$ 开到 $10^9$ 也没问题（只要有模数），因为真的是完全跟 $n$ 无关。

代码

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
# define rep(i,a,b) for(int i=(a); i<=(b); ++i)
# define drep(i,a,b) for(int i=(a); i>=(b); --i)
typedef long long int_;
inline int readint(){
	int a = 0; char c = getchar(), f = 1;
	for(; c<'0'||c>'9'; c=getchar())
		if(c == '-') f = -f;
	for(; '0'<=c&&c<='9'; c=getchar())
		a = (a<<3)+(a<<1)+(c^48);
	return a*f;
}
inline void writeint(int_ x){
	if(x > 9) writeint(x/10);
	putchar((x-x/10*10)^48);
}

const int MaxM = 200005;
const int Mod = 998244353;

int nxt[MaxM], num[MaxM];
vector<int> primes;
bool isPrime[MaxM];
void sieve(int n){
	memset(isPrime+2,1,n-1);
	for(int i=2,len=0; i<=n; ++i){
		if(isPrime[i]){
			primes.push_back(i), ++ len;
			num[i] = nxt[i] = 1;
		}
		for(int j=0; j<len&&primes[j]<=n/i; ++j){
			isPrime[i*primes[j]] = false;
			if(i%primes[j] == 0){
				num[i*primes[j]] = num[i]+1;
				nxt[i*primes[j]] = nxt[i];
				break;
			}
			num[i*primes[j]] = 1;
			nxt[i*primes[j]] = i;
		}
	}
}

int c[MaxM], inv[MaxM];
int main(){
	int n = readint(), m = readint();
	sieve(m); inv[1] = 1;
	rep(i,2,m) inv[i] = (0ll+Mod-Mod/i)*inv[Mod%i]%Mod;
	for(int i=c[0]=1; i<=m; ++i)
		c[i] = 1ll*c[i-1]*(i+n-1)%Mod*inv[i]%Mod;
	int ans = 1; // when a_n = 1
	for(int i=2; i<=m; ++i){
		int now = 1, x = i;
		for(; x!=1; x=nxt[x])
			now = 1ll*now*c[num[x]]%Mod;
		ans = (ans+now)%Mod;
	}
	printf("%d\n",ans);
	return 0;
}