結果

問題 No.1424 Ultrapalindrome
ユーザー umezo
提出日時 2021-03-12 23:18:17
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,845 bytes
コンパイル時間 2,232 ms
コンパイル使用メモリ 207,168 KB
最終ジャッジ日時 2025-01-19 15:45:35
ジャッジサーバーID
（参考情報） 		judge2 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 26 WA * 3
権限があれば一括ダウンロードができます
コンパイルメッセージ
In file included from /usr/include/c++/13/bits/stl_algobase.h:64,
                 from /usr/include/c++/13/algorithm:60,
                 from /usr/include/x86_64-linux-gnu/c++/13/bits/stdc++.h:51,
                 from main.cpp:5:
In constructor ‘constexpr std::pair<_T1, _T2>::pair(_U1&&, _U2&&) [with _U1 = int&; _U2 = int&; typename std::enable_if<(std::_PCC<true, _T1, _T2>::_MoveConstructiblePair<_U1, _U2>() && std::_PCC<true, _T1, _T2>::_ImplicitlyMoveConvertiblePair<_U1, _U2>()), bool>::type <anonymous> = true; _T1 = long long int; _T2 = int]’,
    inlined from ‘int main()’ at main.cpp:97:11:
/usr/include/c++/13/bits/stl_pair.h:688:42: warning: ‘mid’ may be used uninitialized [-Wmaybe-uninitialized]
  688 |         : first(std::forward<_U1>(__x)), second(std::forward<_U2>(__y))
      |                                          ^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
main.cpp: In function ‘int main()’:
main.cpp:85:7: note: ‘mid’ was declared here
   85 |   int mid;
      |       ^~~

ソースコード

diff # 

#define rep(i, n) for (int i = 0; i < (int)(n); i++)
#define ALL(v) v.begin(), v.end()
typedef long long ll;

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

const int INF=1e9;

struct Edge{
  int to;
  int w;
  Edge(int to,ll w) : to(to),w(w) {}
};

using Graph=vector<vector<Edge>>;
using pli=pair<ll,int>;

int dist[100010];

template<class T> bool chmin(T& a,T b){
  if(a>b){
    a=b;
    return true;
  }
  return false;
}

int main(){
  int n;
  cin>>n;
  
  Graph G(n);
  rep(i,n-1){
    int a,b;
    cin>>a>>b;
    a--,b--;
    G[a].push_back(Edge(b,1));
    G[b].push_back(Edge(a,1));
  }
  
  int x=-1,y=-1;
  set<int> se;
  rep(i,n){
    if(G[i].size()==1){
      if(x==-1) x=i;
      else if(y==-1) y=i;
      se.insert(i);
    }
  }
  
  if(se.size()==2){
    cout<<"Yes"<<endl;
    return 0;
  }
  
  int s=x;
  rep(i,n) dist[i]=INF;
  dist[s]=0;
  
  priority_queue<pli,vector<pli>,greater<pli>> que;
  que.push({dist[s],s});
  
  while(!que.empty()){
    int v=que.top().second;
    ll d=que.top().first;
    que.pop();
    
    if(d>dist[v]) continue;
    
    for(auto e:G[v]){
      if(chmin(dist[e.to],dist[v]+e.w)){
        que.push({dist[e.to],e.to});
      }
    }
  }
  
  int di=dist[y];
  
  if(di%2==1){
    cout<<"No"<<endl;
    return 0;
  }
  
  int mid;
  rep(i,n){
    if(dist[i]==di/2){
      mid=i;
      break;
    }
  }
  
  s=mid;
  rep(i,n) dist[i]=INF;
  dist[s]=0;
  
  que.push({dist[s],s});
  
  while(!que.empty()){
    int v=que.top().second;
    ll d=que.top().first;
    que.pop();
    
    if(d>dist[v]) continue;
    
    for(auto e:G[v]){
      if(chmin(dist[e.to],dist[v]+e.w)){
        que.push({dist[e.to],e.to});
      }
    }
  }
  
  bool b=true;
  for(auto p:se){
    if(p==mid) continue;
    if(dist[p]!=di/2) b=false;
  }
  if(b) cout<<"Yes"<<endl;
  else cout<<"No"<<endl;

  return 0;
}
0