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

    ID: 17329 Type: RemoteJudge 3000ms 512MiB Tried: 0 Accepted: 0 Difficulty: (None) Uploaded By: Tags>数论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,o,o=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=0,N≤1015N\le10^{15};
Subtask 2:o=1o=1,N≤1011N\le10^{11};
Subtask 3:o=2o=2,N≤1011N\le10^{11}。