Atcoder ABC-265E题解

$ABC265E-Warp$

题目

题目传送门

Takahashi is at the origin of a two-dimensional plane.

Takahashi will repeat teleporting $N$ times. In each teleportation, he makes one of the following moves:

• Move from the current coordinates $(x,y)$ to $(x+A,y+B)$
• Move from the current coordinates $(x,y)$ to $(x+C,y+D)$
• Move from the current coordinates $(x,y)$ to $(x+E,y+F)$

There are obstacles on $M$ points $(X_1,Y_2),\dots ,(X_M,Y_M)$ on the plane; he cannot teleport to these coordinates.

How many paths are there resulting from the N teleportations?

Find the count modulo $998244353$.

题目大意

一共有$N$次移动次数

每次移动可以选择

从$(x,y)$移动到$(x+A,y+B)$或$(x+C,y+D)$或$(x+E,y+F)$

起始位置位于$(0,0)$

有$M$个地方无法走,分别是$(X_1,Y_2),\dots ,(X_M,Y_M)$

问有多少种移动$N$次的不同方案，答案对$998244353$取模

数据范围

• $1\le N\le 300$
• $0\le M\le 10$
• $-10^9\le A,B,C,D,E,F\le 10^9$
• 保证$(A,B),(C,D),(E,F)$互不相同
• $-10^9\le x_i,y_i\le 10^9$
• 保证所有输入都为整数

读入格式

---

$N$ $M$

$A$ $B$ $C$ $D$ $E$ $F$

$x_1$ $y_1$
…

$x_M$ $y_M$

---

内存和时间限制

• 时间限制$:3s$
• 空间限制$:1024MB$

分析

设三种移动方式分别为一,二,三,移动次数分别为$i$,$j$,$k$

我们可以发现，由于初始坐标为$(0,0)$,因此,当前步数所处的点的位置可以用$i$,$j$,$k$来表示

即$(i\times a+j\times c+k\times e,i\times b+j\times d+k\times f)$

初步想法为四维$DP$,即$f[temp][i][j][k]$表示前temp步，移法一走了$i$次，移法二走了$j$次，移法三走了$k$次

因为$N$的最大值为$300$，开四维数组$N^4$会超内存

我们考虑一下类似于$NOIP2008$传纸条的优化思路(没做过的点这里)

显然$temp=i+j+k$,那么我们便可以只设置一个三维数组，为$f[temp][i][j]$,其中$k$可以通过$temp-i-j$算出

由于本题只让求方案数

因此我们本次的方案数就等于上次走移法一的方案数$+$上次走移法二的方案数$+$上一次走移法三的方案数

由此，我们可以得到一个$DP$式子:

$\boxed{f[temp][i][j]=(f[temp-1][i][j]+f[temp-1][i-1][j]+f[temp-1][i][j-1])}$

显然，$DP$的数组为$f[0][0][0]=1$

我们分别枚举$temp,i,j$,三重循环即可计算出答案

得到$DP$式子后，我们再去思考如何判断这个位置能不能走，一个很显然的想法就是用$map$进行判断

即

map<pair<int,int>,bool> fl;

然后对于每一个$x_i,y_i$,我们进行如下操作

fl.insert(make_pair(make_pair(x,y),1));

枚举$temp,i,j$时，只需判断移法一走$i$步，移法二走$j$步，移法三走$k$步所到达的点是否在$map$数组中即可

所以代码总时间复杂度为$O(N^{3}logM)$

代码实现

本人蒟蒻，代码仅供参考

#include<bits/stdc++.h>
#define mod 998244353//注意要取模（分析里没有说）
#define PD(i,j,k) make_pair(i*a+j*c+k*e,i*b+j*d+k*f)//计算当前的位置
using namespace std;
int n,m,a,b,c,d,e,f,x,y,dp[306][305][305],ans;
map<pair<int,int>,bool> fl;
int main()
{
   scanf("%d %d %d %d %d %d %d %d",&n,&m,&a,&b,&c,&d,&e,&f);
   for(int i=0;i<m;++i)
      scanf("%d %d",&x,&y),fl.insert(make_pair(make_pair(x,y),1));//将(x,y)添加进map数组
   dp[0][0][0]=1;//初值设定
   for(int temp=1;temp<=n;++temp)
      for(int i=0;i<=temp;++i)
         for(int j=0,k=temp-i-j;j+i<=temp;++j,--k)
         {
            if(fl.find(PD(i,j,k))!=fl.end())//判断是否能走
               continue;
            dp[temp][i][j]=((dp[temp-1][i][j]+dp[temp-1][i-1][j])%mod+dp[temp-1][i][j-1])%mod;//计算当前DP值
            if(temp==n)
               ans=(ans+dp[temp][i][j])%mod;//统计答案
         }
   printf("%d",ans);
   return 0;
}