P8152 「PMOI-5」破译の论 题解

题目传送门

分析

第一次划分会变成 $n^2$ 块，之后每次划分都会增加 $n^2-1$ 块（减一是因为自己那一块不能重复算），所以答案就是 $k\cdot (n^2-1)+1$。

需要注意一下取模。

代码实现

#include <bits/stdc++.h>
using namespace std;
 
long long n,k;
long long ans;
int main(){
    cin>>n>>k;
    cout<<((k*(n%998244353*n%998244353-1))%998244353+1+998244353)%998244353;
    //system("pause");
}