#904252 (C++17) No.2431 Viral Hotel

• 問題: No.2431 Viral Hotel
• ユーザー: 👑 potato167
• 提出日時: 2023-08-16 01:44:02
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: WA
• 実行時間: -
• コード長: 1,270 bytes
• コンパイル時間: 2,000 ms
• コンパイル使用メモリ: 212,736 KB
• 実行使用メモリ: 12,456 KB
• 最終ジャッジ日時: 2024-11-24 11:00:59
• 合計ジャッジ時間: 8,351 ms
• ジャッジサーバーID （参考情報）: judge1 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	WA	-
testcase_01	WA	-
testcase_02	WA	-
testcase_03	WA	-
testcase_04	WA	-
testcase_05	WA	-
testcase_06	AC	42 ms; 5,248 KB
testcase_07	AC	40 ms; 5,248 KB
testcase_08	AC	40 ms; 5,248 KB
testcase_09	AC	2 ms; 5,248 KB
testcase_10	AC	2 ms; 5,248 KB
testcase_11	AC	2 ms; 5,248 KB
testcase_12	AC	189 ms; 12,392 KB
testcase_13	WA	-
testcase_14	WA	-
testcase_15	WA	-
testcase_16	AC	90 ms; 9,796 KB
testcase_17	WA	-
testcase_18	WA	-
testcase_19	WA	-
testcase_20	WA	-
testcase_21	AC	34 ms; 5,352 KB
testcase_22	WA	-
testcase_23	WA	-
testcase_24	WA	-
testcase_25	WA	-
testcase_26	WA	-
testcase_27	WA	-
testcase_28	WA	-
testcase_29	WA	-
testcase_30	WA	-
testcase_31	WA	-
testcase_32	WA	-
testcase_33	WA	-
testcase_34	AC	7 ms; 5,248 KB
testcase_35	WA	-
testcase_36	WA	-
testcase_37	WA	-
testcase_38	AC	40 ms; 6,400 KB
testcase_39	AC	1 ms; 5,248 KB
testcase_40	AC	1 ms; 5,248 KB
testcase_41	AC	2 ms; 5,248 KB
testcase_42	AC	2 ms; 5,248 KB
testcase_43	WA	-
testcase_44	WA	-
testcase_45	WA	-

ソースコード

#include<bits/stdc++.h>
using namespace std;
using ll = long long;
#define all(p) p.begin(),p.end()
#define rep(i,a,b) for(int i=(int)a;i<(int)b;i++)
const int mod=998244353;

int main(){
	int N,K,M,P;
	cin>>N>>K>>M>>P;
	vector<vector<int>> G(N);
	rep(i,0,M){
		int a,b;
		cin>>a>>b;
		a--,b--;
		G[a].push_back(b);
		G[b].push_back(a);
	}
	struct event
	{
		int type;
		int time;
		int ind;
	};
	auto comp_event=[&](event l,event r)->bool{
		if(l.time==r.time) return l.type>r.type;
		return l.time>r.time;
	};
	vector<int> p(N);
	priority_queue<event,vector<event>,std::function<bool(event,event)>> pq(comp_event);
	vector<int> S(N);
	rep(i,0,N) cin>>S[i];
	rep(i,0,K){
		int x;
		cin>>x;
		x--;
		pq.push({1,P,x});
		pq.push({2,S[x],x});
		p[x]++;
	}
	int ans=0;
	while(!pq.empty()){
		auto tmp=pq.top();
		pq.pop();
		//cout<<tmp.type<<" "<<tmp.time<<" "<<tmp.ind+1<<endl;
		if(tmp.type==1){
			p[tmp.ind]=-2;
		}
		if(tmp.type==2){
			if(p[tmp.ind]==-1) continue;
			for(auto x:G[tmp.ind]){
				if(p[x]==-1||p[x]==-2) continue;
				p[x]++;
				if(p[x]==1){
					pq.push({2,tmp.time+S[x],x});
					pq.push({1,tmp.time+P,x});
				}
				if(p[x]==2){
					pq.push({3,tmp.time,x});
				}
			}
		}
		if(tmp.type==3){
			ans++;
			p[tmp.ind]=-1;
		}
	}
	cout<<ans<<"\n";
}