#P17255. [Aboi 2077] 全部夢だった!

    ID: 19729 远端评测题 3000ms 512MiB 尝试: 0 已通过: 0 显示难度NOI/NOI+/CTS 上传者: 标签>数论O2优化群论线性代数2077杜教筛亚线性快速求和算法

[Aboi 2077] 全部夢だった!

背景

月夜に浮かぶ になって

:::info[名词释义] 定义 SL2(Z)\text{SL}_2(\mathbb Z)NN 的三类同余子群 Γ0,Γ1,Γ\Gamma_0,\Gamma_1,\Gamma

$$\begin{aligned} \Gamma_0(N)&=\left\{\begin{bmatrix}a&b\\c&d\end{bmatrix}\in\text{SL}_2(\mathbb Z)\mathrel{\Bigg|}\begin{bmatrix}a&b\\c&d\end{bmatrix}\equiv\begin{bmatrix}*&*\\0&*\end{bmatrix}\pmod N\right\}\\ \Gamma_1(N)&=\left\{\begin{bmatrix}a&b\\c&d\end{bmatrix}\in\text{SL}_2(\mathbb Z)\mathrel{\Bigg|}\begin{bmatrix}a&b\\c&d\end{bmatrix}\equiv\begin{bmatrix}1&*\\0&1\end{bmatrix}\pmod N\right\}\\ \Gamma(N)&=\left\{\begin{bmatrix}a&b\\c&d\end{bmatrix}\in\text{SL}_2(\mathbb Z)\mathrel{\Bigg|}\begin{bmatrix}a&b\\c&d\end{bmatrix}\equiv\begin{bmatrix}1&0\\0&1\end{bmatrix}\pmod N\right\} \end{aligned}$$

* 表示无限制。

定义同余子群 Γ\Gamma尖点Q{}\mathbb Q\cup\{\infty\}Γ\Gamma 作用下的等价类。这里,对于 $\gamma=\begin{bmatrix}a&b\\c&d\end{bmatrix}\in\Gamma$,其作用定义为 γ(s)=as+bcs+d\gamma(s)=\dfrac{as+b}{cs+d},特别地有 γ()=ac\gamma(\infty)=\dfrac ac;若分母为 00 则值为 \infty

定义 ε(Γ)\varepsilon_\infty(\Gamma) 为同余子群 Γ\Gamma 的尖点数。 :::

题目描述

给定正整数 NN,对于给定 G{Γ0,Γ1,Γ}G\in\{\Gamma_0,\Gamma_1,\Gamma\},求 n=1Nε(G(n))\displaystyle\sum_{n=1}^N\varepsilon_\infty(G(n)),对 998244353998244353 取模。

输入格式

一行两个整数 N,oN,oo=0,1,2o=0,1,2 分别对应 G=Γ0,Γ1,ΓG=\Gamma_0,\Gamma_1,\Gamma

输出格式

一行一个整数表示答案对 998244353998244353 取模的值。

10 0
28
10 1
44
10 2
158

提示

Subtask 1:o=0o=0N1015N\le10^{15}
Subtask 2:o=1o=1N1011N\le10^{11}
Subtask 3:o=2o=2N1011N\le10^{11}