#708390 (C++17) No.20 砂漠のオアシス

• 問題: No.20 砂漠のオアシス
• ユーザー: 沙耶花
• 提出日時: 2021-10-20 23:20:27
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 25 ms / 5,000 ms
• コード長: 1,331 bytes
• コンパイル時間: 4,443 ms
• コンパイル使用メモリ: 271,096 KB
• 実行使用メモリ: 5,696 KB
• 最終ジャッジ日時: 2023-10-20 11:13:28
• 合計ジャッジ時間: 5,332 ms
• ジャッジサーバーID （参考情報）: judge11 / judge12

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 4,348 KB
testcase_01	AC	2 ms; 4,348 KB
testcase_02	AC	2 ms; 4,348 KB
testcase_03	AC	3 ms; 4,348 KB
testcase_04	AC	4 ms; 4,348 KB
testcase_05	AC	19 ms; 5,616 KB
testcase_06	AC	13 ms; 5,696 KB
testcase_07	AC	25 ms; 5,696 KB
testcase_08	AC	15 ms; 5,696 KB
testcase_09	AC	22 ms; 5,696 KB
testcase_10	AC	2 ms; 4,348 KB
testcase_11	AC	2 ms; 4,348 KB
testcase_12	AC	3 ms; 4,348 KB
testcase_13	AC	3 ms; 4,348 KB
testcase_14	AC	4 ms; 4,348 KB
testcase_15	AC	4 ms; 4,348 KB
testcase_16	AC	6 ms; 4,348 KB
testcase_17	AC	5 ms; 4,348 KB
testcase_18	AC	5 ms; 4,348 KB
testcase_19	AC	6 ms; 4,348 KB
testcase_20	AC	3 ms; 4,348 KB

ソースコード

#include <stdio.h>
#include <bits/stdc++.h>
#include <atcoder/all>
using namespace atcoder;
using mint = modint;
using namespace std;
#define rep(i,n) for (int i = 0; i < (n); ++i)
#define Inf 1000000001



int main(){
	
	int N,V,Ox,Oy;
	cin>>N>>V>>Ox>>Oy;
	Ox--;Oy--;
	swap(Ox,Oy);
	
	vector<vector<int>> a(N,vector<int>(N));
	rep(i,N){
		rep(j,N)cin>>a[i][j];
	}
	
	vector dis(N,vector(N,vector<int>(2,-Inf)));
	dis[0][0][0] = V;
	priority_queue<pair<int,array<int,3>>> Q;
	array<int,3> temp;
	rep(i,3)temp[i] = 0;
	Q.emplace(V,temp);
	
	vector<int> dy = {1,-1,0,0},dx = {0,0,1,-1};
	
	while(Q.size()>0){
		int D = Q.top().first;
		int y = Q.top().second[0],x = Q.top().second[1],f = Q.top().second[2];
		Q.pop();
		if(dis[y][x][f]!=D)continue;
		rep(i,4){
			int yy = y + dy[i],xx = x + dx[i];
			int ff = f;
			if(yy<0||yy>=N||xx<0||xx>=N)continue;
			int DD = D;
			DD -= a[yy][xx];
			if(DD<=0)continue;
			if(dis[yy][xx][ff] < DD){
				dis[yy][xx][ff] = DD;
				temp = {yy,xx,ff};
				Q.emplace(DD,temp);
			}
			if(yy==Ox&&xx==Oy&&f==0){
				ff = 1;
				DD *= 2;
				if(dis[yy][xx][ff] < DD){
					dis[yy][xx][ff] = DD;
					temp = {yy,xx,ff};
					Q.emplace(DD,temp);
				}
			}
			
		}
	}
	
	int ans = max(dis.back().back()[0],dis.back().back()[1]);
	if(ans==-Inf)cout<<"NO"<<endl;
	else cout<<"YES"<<endl;
	
	return 0;
}