[AGM 2022 资格赛] 地牢 解题记录
题意简述
有 L L L 个 N × M N \times M N×M 大小的地牢,对于每个单元格:
-
0
代表空单元格。 -
-9
代表障碍。 -
-1
代表每关的出口,走到后传送到下一关的相同位置。 -
x
代表一个敌人,能力为 x x x,你需要大于等于 x x x 的能力值才能战胜他,并获取他的能力值。
初始时,你位于 ( 1 , 1 ) (1,1) (1,1),拥有 1 1 1 能力值。
题目分析
首先可以发现一个显而易见的规律,那就是打敌人先从能力值小的开始(并且也只能从能力值小的开始),不然可能会出现在某一关遇到能力值大的敌人,但是由于上一关遗漏了某些能力值小的敌人而无法击败。
有了这个发现后,我们再来考虑哪些敌人可以被打到。
对于这种迷宫走格子的问题,我们一般考虑使用搜索。此题的数据范围为 1 ≤ L × N × M ≤ 1 0 6 1 \leq L \times N \times M \leq 10^6 1≤L×N×M≤106,每个点只遍历一遍,所以时间复杂度最多 O ( L × N × M ) O(L \times N \times M) O(L×N×M) 可以支持搜索。
在搜索的过程中标记每一关可以打到的敌人,按升序排序后累加答案就行。
AC Code
#include<bits/stdc++.h>
#define arrout(a,n) rep(i,1,n)std::cout<<a[i]<<" "
#define arrin(a,n) rep(i,1,n)std::cin>>a[i]
#define rep(i,x,n) for(int i=x;i<=n;i++)
#define dep(i,x,n) for(int i=x;i>=n;i--)
#define erg(i,x) for(int i=head[x];i;i=e[i].nex)
#define dbg(x) std::cout<<#x<<":"<<x<<" "
#define mem(a,x) memset(a,x,sizeof a)
#define all(x) x.begin(),x.end()
#define arrall(a,n) a+1,a+1+n
#define PII std::pair<int,int>
#define m_p std::make_pair
#define u_b upper_bound
#define l_b lower_bound
#define p_b push_back
#define CD const double
#define CI const int
#define int long long
#define il inline
#define ss second
#define ff first
#define itn int
CI N=1e6+5;
int T,n,m,xx=1,yy=1,ans=1,e[N];
int32_t mp[5005][5005];
bool vis[5005][5005];
int dx[]={1,-1,0,0},dy[]{0,0,1,-1};
void dfs(int x,int y) {
rep(i,0,3) {
x+=dx[i],y+=dy[i];
if(!vis[x][y]&&x>=1&&x<=n&&y>=1&&y<=m&&mp[x][y]!=-9) {
vis[x][y]=1;
if(mp[x][y]==-1) {
xx=x,yy=y;
}
if(mp[x][y]>0) {
e[++e[0]]=mp[x][y];
}
dfs(x,y);
}
x-=dx[i],y-=dy[i];
}
}
signed main() {
std::cin>>T>>n>>m;
while(T--) {
mem(e,0);
rep(i,1,n) {
rep(j,1,m) {
std::cin>>mp[i][j];
vis[i][j]=0;
}
}
dfs(xx,yy);
std::sort(arrall(e,e[0]));
for(int i=1;i<=e[0]&&ans>=e[i];i++){
ans+=e[i];
}
}
std::cout<<ans;
return 0;
}