(AtCoder Beginner Contest 232) E - Rook Path dp+组合数

题目大意：

给定一个 h*w（0<w,h<1e9） 的棋盘，现在有一个象棋棋子“车”在（x1，y1）现在你能执行k步走到（ x2 , y2 )，请问有几种走法。

题解

用dp来处理行和列，
一种是在当前行，另一种是不在当前行，即分别为dp[1],dp[0];
那么如果我在i步内能到达即是dp[1][i]，不能到就是dp[0][i]
这样就可以写出转移方程：

dp[i][1]=dp[i-1][0];//注意dp[i][1]不也可以由dp[i-1][1]转移得到，因为移动0步不算一次操作。
dp[i][0]=(H-1)*dp[i-1][1]+(H-2)*dp[i-1][0];

同理，对行也进行同样操作。
这样我们如果在第i步到达指定行，那么就必须在第k-i步到达指定列了。
即 dx[i][1]*dy[k-i][1];
然后操作i步可以在k步内任意组合，即C(i,k)//组合数注意处理逆元
这样枚举i就可以了
以下是ac代码：

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+5;
typedef long long LL;
const LL mod=998244353;
LL dx[N][2],dy[N][2];
LL f[N],inv[N];
int H,W,K;
LL C(int a,int b)
{
    return f[a]*inv[b]%mod*inv[a-b]%mod;
}
LL ksm(LL a,LL n)
{
    LL res=1;
    while(n)
    {
        if(n&1)res=a%mod*res%mod;
        a=a%mod*a%mod;
        n>>=1;
    }
    return res;
}
void init()
{
    f[0]=inv[0]=1;
    for(int i=1;i<=K;i++) f[i]=f[i-1]*i%mod;
    inv[K]=ksm(f[K],mod-2);
    for(int i=K-1;i;i--) inv[i]=inv[i+1]*(i+1)%mod;
}
void solve()
{
    scanf("%d %d %d",&H,&W,&K);
    init();
    int x1,x2,y1,y2;
    dx[0][1]=dy[0][1];
    scanf("%d %d %d %d",&x1,&y1,&x2,&y2);
    if(x1==x2) dx[0][1]=1;
    else dx[0][0]=1;
    if(y1==y2) dy[0][1]=1;
    else dy[0][0]=1;
    for(int i=1;i<=K;i++)
    {
        dx[i][1]=(dx[i-1][0])%mod;
        dx[i][0]=((H-1)%mod*dx[i-1][1]%mod+(H-2)%mod*dx[i-1][0]%mod)%mod;
        dy[i][1]=(dy[i-1][0])%mod;
        dy[i][0]=((W-1)%mod*dy[i-1][1]%mod+(W-2)%mod*dy[i-1][0]%mod)%mod;
    }
    LL ans=0;
    for(int i=0;i<=K;i++)
    {
        LL res=dx[i][1]*dy[K-i][1]%mod;
        res=res*C(K,i)%mod;
        ans=(ans+res)%mod;
    }
    printf("%lld",ans);

}
int main()
{
    //int t;
    //scanf("%d",&t);
    //while(t--)
    //{
        solve();
    //}
}