过河

有一条河，河中有n块石头，现在从河的一边只能通过走石头到达对岸，每一步可以跨越至多3个石头。

但是不幸的是，有一块石头被上一个过河的人踩松后被踩松了，所以为了安全后来的人就不能再踩这一块石头了。

若现在有一个人想要到河的对岸去，他有多少种方法？

注：若我们将石头从 1 到 n进行编号的话，那么被踩松的石头编号为 k。

输入格式:

多组输入

对于每组输入在一行中给出 2 个整数 n 和 k。

数据保证有 1≤k<n≤10​6​​。

输出格式:

对于每组输入，输出一行，代表对应输入的过河方案数。

结果对 998244353 取余后输出。

输入样例:

在这里给出一组输入。例如：

4 3

输出样例:

在这里给出相应的输出。例如：

3

思路

一共分为四种情况，k=1，k=2，k=3和k>3的四种情况，每次都先把n=1，2，3的情况列出来，再循环n>=4的情况，k>3时还要再判断n是否等于k，等于的话要把 x[i] 的值改为0 

代码 

#include <iostream>
#include <cstdio>
using namespace std;
int main()
{
    long long x[1000010];
    int n,k;
    while(~scanf("%d %d",&n,&k))
    {
        if(k==3)
        {
            x[1]=1;
            x[2]=2;
            x[3]=0;
            for(int i=4;i<=n;i++)
            {
                x[i]=(x[i-1]+x[i-2]+x[i-3])%998244353;
            }
            printf("%d\n",x[n]);
        }
        else if(k==1)
        {
            x[1]=0;
            x[2]=1;
            x[3]=2;
            for(int i=4;i<=n;i++)
            {
                x[i]=(x[i-1]+x[i-2]+x[i-3])%998244353;
            }
            printf("%d\n",x[n]);
        }
        else if(k==2)
        {
            x[1]=1;
            x[2]=0;
            x[3]=2;
            for(int i=4;i<=n;i++)
            {
                x[i]=(x[i-1]+x[i-2]+x[i-3])%998244353;
            }
            printf("%d\n",x[n]);
        }
        else
        {
            x[1]=1;
            x[2]=2;
            x[3]=4;
            for(int i=4;i<=n;i++)
            {
                if(i!=k)
                    x[i]=(x[i-1]+x[i-2]+x[i-3])%998244353;
                else
                    x[i]=0;
            }
            printf("%d\n",x[n]);
        }
    }
    return 0;
}