[COGS 2877]老m凯的疑惑

Description

Margatroid退役之后沉迷文化课

这天，写完数学作业之后的他脑洞大开，决定出一道比NOIP2017 D1T1《小凯的疑惑math》还要好的题

题面是这样的

$$ f(n)=n^2\\ g(n)=\sum_{i=1}^{n^3}[f(i)<n]\\\\ k(n)=\sum_{i=1}^{n^3}[g(i)<n] $$

试求$k(n)\ \text{mod}\ 998244353$

Input

一行一个整数$n$

Output

一行一个整数$k(n)$

Sample Input

Sample Output

Hint

出题人沉迷文化课，无心造数据，满足数据是以10为首项，10为公比的等比数列

题解

由题： $$g(n) = \sum_{i=1} [i^2 < n]$$

显然：

$$g(n) =
\begin{cases}
\sqrt n-1& \text{ n 是完全平方数}\\
\lfloor \sqrt n \rfloor& \text{otherwise}
\end{cases}$$

构造等价函数： $$g(n) = \lfloor \sqrt {n-1} \rfloor$$

同理，由题： $$k(n) = \sum_{i=1} [\sqrt {i-1} < n]$$

因为 $n$ 是正整数，所以 $k(n)$ 等价于：

\begin{aligned}
   k(n) &= \sum_{i=1} [i-1 < n^2]\\
   & = \sum_{i=1} [i <= n^2]\\
   & = n^2
\end{aligned}

 //It is made by Awson on 2018.1.2

 #include <set>

 #include <map>

 #include <cmath>

 #include <ctime>

 #include <queue>

 #include <stack>

 #include <cstdio>

 #include <string>

 #include <vector>

 #include <cstdlib>

 #include <cstring>

 #include <iostream>

 #include <algorithm>

 #define LL long long

 #define Max(a, b) ((a) > (b) ? (a) : (b))

 #define Min(a, b) ((a) < (b) ? (a) : (b))

 using namespace std;

 const LL MOD = ;

 LL n;

 void work() {

     scanf("%lld", &n);

     n = n%MOD*n%MOD;

     printf("%lld\n", n);

 }

 int main() {

     work();

     return ;

 }

秒客网