P17255 [Aboi 2077] 全部夢だった!
题目背景
> 月夜に浮かぶ **星**になって
:::info[名词释义]
定义 $\text{SL}_2(\mathbb Z)$ 模 $N$ 的三类**同余子群** $\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$ 的**尖点**为 $\mathbb Q\cup\{\infty\}$ 在 $\Gamma$ 作用下的等价类。这里,对于 $\gamma=\begin{bmatrix}a&b\\c&d\end{bmatrix}\in\Gamma$,其作用定义为 $\gamma(s)=\dfrac{as+b}{cs+d}$,特别地有 $\gamma(\infty)=\dfrac ac$;若分母为 $0$ 则值为 $\infty$。
定义 $\varepsilon_\infty(\Gamma)$ 为同余子群 $\Gamma$ 的尖点数。
:::
题目描述
给定正整数 $N$,对于给定 $G\in\{\Gamma_0,\Gamma_1,\Gamma\}$,求 $\displaystyle\sum_{n=1}^N\varepsilon_\infty(G(n))$,对 $998244353$ 取模。
输入格式
一行两个整数 $N,o$,$o=0,1,2$ 分别对应 $G=\Gamma_0,\Gamma_1,\Gamma$。
输出格式
一行一个整数表示答案对 $998244353$ 取模的值。
说明/提示
Subtask 1:$o=0$,$N\le10^{15}$;
Subtask 2:$o=1$,$N\le10^{11}$;
Subtask 3:$o=2$,$N\le10^{11}$。