牛客小白月赛14A-简单计数【组合数+规律】

你在一个有 n 个城市的国家中行走，城市从 1 到 n 依次编号
任意两个城市之间都有一条双向道路可以通行，且你可以花一天的时间从当前所在的城市到达任意一个别的城市
由于你比较闲的无聊，所以你不会连续两天都呆在同一个城市，也就是说每天你所在的城市都不相同（这句话的意思是，对于相邻的两天，你所在的城市应该不同）
一开始你在 1 号城市，求经过 k 天后你回到 1 号城市的方案数

当然如果不存在任意一种方案就输出 0 就好了

 

输入描述:

第一行两个整数 n,k

输出描述:

一行一个整数表示答案对 998244353 取模后的结果

思路：打表找规律，然后可以推出来公式，发现是一个等比数列，然后用公式求和就行了，注意要用逆元（费马小定理）。

#include<set>
#include<map>
#include<cmath>
#include<cstdio>
#include<queue>
#include<vector>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn = 2e6 + 10;
const ll mod = 998244353;
ll n, k;
ll powermod(ll a, ll b, ll c)
{
    ll ans = 1;
    a = a % c;
    while(b)
    {
        if(b & 1)
            ans = ans * a % c;
        b >>= 1;
        a = a * a % c;
    }
    return ans;
}
ll calc(ll a, ll q, ll n)
{
    return a * (mod + 1 - powermod(q, n, mod)) % mod * powermod(mod + 1 - q, mod - 2, mod) % mod;
}

int main()
{

    cin >> n >> k;
    n--;
    ll A = calc(n, n * n % mod, k/2);
    ll B = calc(n * n % mod, n * n % mod, (k - 1)/2);
    ll ans;
    if(k & 1)
        ans = (B - A + mod) % mod;
    else
        ans = (A - B + mod) % mod;
    cout << ans << endl;
    return 0;
}