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");
}