牛客小白月赛14 A简单计数

链接：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;
}