其他分享
首页 > 其他分享> > DP: 走地图

DP: 走地图

作者:互联网

传送ABC265 https://atcoder.jp/contests/abc265/tasks/abc265_e

从原点移动n次 每次可以选择
(x,y) -> (x+a,y+b)
(x,y) -> (x+c,y+d)
(x,y) -> (x+e,y+f)
其中一种进行移动 求移动n次的方案数
容易想到 f(n,x,y)表示n次传送走到了(x,y) 比较困难
可以让是 f(i,j,k)表示选择了第一种传送i次 第j种传送j次 第3种传送k次
于是走到的点 x=ia+jc+ke y=ib+jd+kf

#include<bits/stdc++.h>
using namespace std;
typedef __float128 f128;
typedef __int128 i128;
typedef long long LL;
typedef pair<int,int>PII;
typedef pair<LL,LL>PLL;
const int mod=998244353;
LL dp[310][310][310];
set<PII>st;
int main()
{
    int n,m;
    int A,B,C,D,E,F;
    cin>>n>>m>>A>>B>>C>>D>>E>>F;
    while(m--){
        int x,y;
        scanf("%d%d",&x,&y);
        st.insert({x,y});
    }
    dp[0][0][0]=1;//选择0 0次数为1

    for(int i=0;i<=n;i++){
        for(int j=0;j<=n-i;j++){
            for(int k=0;k<=n-i-j;k++){//注意边界 i,j,k加起来不会超过n
                if(!i&&!j&&!k)continue; 
                LL x=1LL*A*i+1LL*C*j+1LL*E*k;
                LL y=1LL*B*i+1LL*D*j+1LL*F*k;
                if(st.count({x,y}))continue;
                if(i)dp[i][j][k]=(dp[i][j][k]+dp[i-1][j][k])%mod;//表示这次是进行了第一次选择
                if(j)dp[i][j][k]=(dp[i][j][k]+dp[i][j-1][k])%mod;
                if(k)dp[i][j][k]=(dp[i][j][k]+dp[i][j][k-1])%mod;
            }
        }
    }

    LL ans=0;
    for(int i=0;i<=n;i++){
        for(int j=0;j<=n-i;j++){
            ans=(ans+dp[i][j][n-i-j])%mod;
        }
    }

    cout<<ans<<'\n';
    

    return 0;
}

标签:__,typedef,传送,int,LL,310,地图,DP
来源: https://www.cnblogs.com/liang302/p/16612670.html