链接:https://ac.nowcoder.com/acm/contest/879/A
来源:牛客网
你在一个有 n 个城市的国家中行走,城市从 1 到 n 依次编号
任意两个城市之间都有一条双向道路可以通行,且你可以花一天的时间从当前所在的城市到达任意一个别的城市
由于你比较闲的无聊,所以你不会连续两天都呆在同一个城市,也就是说每天你所在的城市都不相同(这句话的意思是,对于相邻的两天,你所在的城市应该不同)
一开始你在 1 号城市,求经过 k 天后你回到 1 号城市的方案数
当然如果不存在任意一种方案就输出 0 就好了
输入描述:
第一行两个整数 n,k
输出描述:
一行一个整数表示答案对 998244353 取模后的结果
示例1
输入
复制
1 1
输出
复制
0
备注:
maker.cpp如下:
#include "bits/stdc++.h"
using namespace std;
int main() {
srand((unsigned long long) new char);
int n = int(8e8) + rand() % int(1e8);
int k = int(8e8) + rand() % int(1e8);
printf("%d %d\n", n, k);
}
(a/b)%mod=a%(b*mod)/b
代码:
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define ll long long
const ll mod=998244353;
ll n,k;
ll fun(ll a,ll b,ll c)
{
ll ans=0;
while(b)
{
ans=(ans+a*(b%10))%c;
a=a*10%c;
b=b/10;
}
return ans;
}
ll power(ll a,ll b,ll c)
{
ll ans=1;
while(b)
{
if(b&1) ans=fun(ans,a,c)%c;
a=fun(a,a,c);
b>>=1;
}
return ans%c;
}
int main()
{
while(~scanf("%lld%lld",&n,&k))
{
ll M=n*998244353;
if(k&1)
printf("%lld\n",((M-n+1)+power(n-1,k,M))%M/n);
else printf("%lld\n",(n-1+power(n-1,k,M))%M/n);
}
return 0;
}