結果

問題 No.1382 Travel in Mitaru city
ユーザー kotatsugame
提出日時 2021-02-07 21:22:35
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,381 bytes
コンパイル時間 905 ms
コンパイル使用メモリ 80,152 KB
実行使用メモリ 11,596 KB
最終ジャッジ日時 2024-07-04 14:36:07
合計ジャッジ時間 8,905 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 38 WA * 30
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp:40:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   40 | main()
      | ^~~~

ソースコード

diff # 

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
#include<vector>
struct UF{
	int n;
	vector<int>parent,rank,mx;
	UF(int n_=0):n(n_),parent(n_),rank(n_,1),mx(n_,(int)-1e9)
	{
		for(int i=0;i<n_;i++)parent[i]=i;
	}
	int find(int a){return parent[a]!=a?parent[a]=find(parent[a]):a;}
	bool same(int a,int b){return find(a)==find(b);}
	bool unite(int a,int b)
	{
		a=find(a),b=find(b);
		if(a==b)return false;
		if(rank[a]<rank[b])
		{
			mx[b]=max(mx[b],mx[a]);
			parent[a]=b;
			rank[b]+=rank[a];
		}
		else
		{
			mx[a]=max(mx[a],mx[b]);
			parent[b]=a;
			rank[a]+=rank[b];
		}
		return true;
	}
	int size(int a){return rank[find(a)];}
};
int N,M,S,T;
int P[1<<17];
vector<int>G[1<<17];
int dp[1<<17];
int id[1<<17];
main()
{
	cin>>N>>M>>S>>T;
	S--,T--;
	for(int i=0;i<N;i++)cin>>P[i];
	for(int i=0;i<M;i++)
	{
		int a,b;cin>>a>>b;
		a--,b--;
		G[a].push_back(b);
		G[b].push_back(a);
	}
	for(int i=0;i<N;i++)id[i]=i;
	sort(id,id+N,[](int i,int j){return P[i]>P[j];});
	UF uf(N);
	for(int idx=0;idx<N;idx++)
	{
		int u=id[idx];
		for(int v:G[u])if(P[v]>P[u])uf.unite(u,v);
		int mx=uf.mx[uf.find(u)];
		mx++;
		if(u==S)mx=0;
		uf.mx[uf.find(u)]=mx;
		if(idx+1<N&&P[id[idx]]>P[id[idx+1]])
		{
			for(int i=idx;i>=0&&P[id[idx]]==P[id[i]];i--)
			{
				for(int v:G[id[i]])if(P[v]==P[id[i]])uf.unite(v,id[i]);
			}
		}
	}
	cout<<uf.mx[uf.find(T)]<<endl;
}
0