結果

問題 No.1382 Travel in Mitaru city
ユーザー 蜜蜂蜜蜂
提出日時 2021-02-04 21:53:01
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 3,323 bytes
コンパイル時間 2,050 ms
コンパイル使用メモリ 185,712 KB
実行使用メモリ 20,096 KB
最終ジャッジ日時 2024-07-04 03:35:00
合計ジャッジ時間 12,860 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 3
other WA * 68
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;


#include <algorithm>
#include <cassert>
#include <vector>

namespace atcoder {

struct dsu {
  public:
    dsu() : _n(0) {}
    explicit dsu(int n) : _n(n), parent_or_size(n, -1) {}

    int merge(int a, int b) {
        assert(0 <= a && a < _n);
        assert(0 <= b && b < _n);
        int x = leader(a), y = leader(b);
        if (x == y) return x;
        if (-parent_or_size[x] < -parent_or_size[y]) std::swap(x, y);
        parent_or_size[x] += parent_or_size[y];
        parent_or_size[y] = x;
        return x;
    }

    bool same(int a, int b) {
        assert(0 <= a && a < _n);
        assert(0 <= b && b < _n);
        return leader(a) == leader(b);
    }

    int leader(int a) {
        assert(0 <= a && a < _n);
        if (parent_or_size[a] < 0) return a;
        return parent_or_size[a] = leader(parent_or_size[a]);
    }

    int size(int a) {
        assert(0 <= a && a < _n);
        return -parent_or_size[leader(a)];
    }

    std::vector<std::vector<int>> groups() {
        std::vector<int> leader_buf(_n), group_size(_n);
        for (int i = 0; i < _n; i++) {
            leader_buf[i] = leader(i);
            group_size[leader_buf[i]]++;
        }
        std::vector<std::vector<int>> result(_n);
        for (int i = 0; i < _n; i++) {
            result[i].reserve(group_size[i]);
        }
        for (int i = 0; i < _n; i++) {
            result[leader_buf[i]].push_back(i);
        }
        result.erase(
            std::remove_if(result.begin(), result.end(),
                           [&](const std::vector<int>& v) { return v.empty(); }),
            result.end());
        return result;
    }

  private:
    int _n;
    std::vector<int> parent_or_size;
};

}  // namespace atcoder

using namespace atcoder;

using ll = long long;
using ld = long double;
#define fi first
#define se second
#define pb push_back

int main(){
  int n,m,s,t;
  cin>>n>>m>>s>>t;
  s--;
  t--;
  
  int p[n];
  int copy[n];
  int newnumber[n];
  
  for(int i=0;i<n;i++){
    cin>>p[i];
    copy[i]=p[i];
  }
  
  sort(copy,copy+n);
  reverse(copy,copy+n);
  
  map<int,int> rank;
  //rank[順位]=人数
  map<int,int> seen;
  
  for(int i=0;i<n;i++){
    rank[copy[i]]=i;
  }
  
  for(int i=0;i<n;i++){
    newnumber[i]=rank[p[i]]-seen[p[i]];
    seen[p[i]]++;
    cout<<newnumber[i]<<" ";
  }
  cout<<endl;
  
  s=newnumber[s],t=newnumber[t];
  
  cout<<s<<" "<<t<<endl;
  
  
  vector<vector<int>> graph(n);
  for(int i=0;i<m;i++){
    int a,b;
    cin>>a>>b;
    a--;
    b--;
    a=newnumber[a],b=newnumber[b];
    graph[a].pb(b);
    graph[b].pb(a);
  }
  
  int dp[n];
  dsu d(n);
  
  for(int i=0;i<n;i++){
    dp[i]=-1*1e9;
  }
  
  int now=-1*1e9;
  
  for(int i=0;i<n;i++){
    if(i==s){
      dp[i]=0;
    }
    if(graph[i].size()!=0){
      for(int neighbor:graph[i]){
        if(neighbor<i){
          d.merge(neighbor,i);
        }
      }
      for(int neighbor:graph[i]){
        if(neighbor<i){
          if(d.same(neighbor,s)==true){
            dp[i]=max(dp[i],now+1);
          }
        }
        else{
          if(d.same(neighbor,s)==true){
            dp[i]=max(dp[i],now);
          }
        }
      }
    }
    now=max(now,dp[i]);
  }
  
  int ans=0;
  for(int i=0;i<n;i++){
    ans=max(ans,dp[i]);
  }
  cout<<ans<<endl;
}
0