結果

問題 No.994 ばらばらコイン
コンテスト
ユーザー kjnh10
提出日時 2020-02-21 21:46:54
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 6,385 bytes
コンパイル時間 2,449 ms
コンパイル使用メモリ 181,752 KB
実行使用メモリ 818,152 KB
最終ジャッジ日時 2024-10-08 21:32:57
合計ジャッジ時間 4,980 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other WA * 1 MLE * 1 -- * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

// varibable settings
#define int long long
const int INF=1e18;

// define basic macro {{{
#define _overload3(_1,_2,_3,name,...) name
#define _rep(i,n) repi(i,0,n)
#define repi(i,a,b) for(int i=(int)(a);i<(int)(b);++i)
#define rep(...) _overload3(__VA_ARGS__,repi,_rep,)(__VA_ARGS__)
#define _rrep(i,n) rrepi(i,0,n)
#define rrepi(i,a,b) for(int i=(int)((b)-1);i>=(int)(a);--i)
#define rrep(...) _overload3(__VA_ARGS__,rrepi,_rrep,)(__VA_ARGS__)
#define each(i,a) for (auto&& i : a)
#define all(x) (x).begin(),(x).end()
#define sz(x) ((int)(x).size())
#define pb(a) push_back(a)
#define mp(a, b) make_pair(a, b)
#define mt(a, b, c) make_tuple(a, b, c)
#define ub upper_bound
#define lb lower_bound
#define posl(A, x) (lower_bound(all(A), x)-A.begin())
#define posu(A, x) (upper_bound(all(A),x)-A.begin())
template<class T> inline void chmax(T &a, const T &b) { if((a) < (b)) (a) = (b); }
template<class T> inline void chmin(T &a, const T &b) { if((a) > (b)) (a) = (b); }

#define divceil(a,b) ((a)+(b)-1)/(b)
#define is_in(x, a, b) ((a)<=(x) && (x)<(b))
#define uni(x) sort(all(x));x.erase(unique(all(x)),x.end())
#define slice(l, r) substr(l, r-l)

typedef long long ll;
typedef vector<int> vi;
typedef vector<vi> vvi;
typedef long double ld;
typedef pair<int,int> pii;
typedef tuple<int,int,int> iii;

template<typename T> using PQ = priority_queue<T, vector<T>, greater<T>>;
struct Fast { Fast(){ std::cin.tie(0); ios::sync_with_stdio(false); } } fast;

#if defined(PCM) || defined(LOCAL)
  #include "lib/dump.hpp"
#else
  #define dump(...) 42
  #define dump_1d(...) 42
  #define dump_2d(...) 42
  #define cerrendl 42
#endif
//}}}


template<typename T> struct SegmentTree { // {{{
  private:
    using F = function<T(T,T)>;
    int n;  // 元の配列のサイズ
    int N;  // n以上の最小の2冪
    vector<T> node;
    F merge;
    T identity;

  public:
    SegmentTree(){}
    SegmentTree(vector<T> a, F f, T id):merge(f), identity(id) {
      n = a.size();
      N = 1; while(N < n) N *= 2;
      node.resize(2*N-1, identity);
      for(int i=0; i<n; i++) node[i+N-1] = a[i];
      for(int i=N-2; i>=0; i--) node[i] = merge(node[2*i+1], node[2*i+2]);
    }
    SegmentTree(int n, F f, T id) : SegmentTree(vector<T>(n, id), f, id) {}

    T& operator[](int i) { return node[i+N-1]; }

    void update(int x, T val) {
      x += (N - 1);
      node[x] = val;
      while(x > 0) {
        x = (x - 1) / 2;
        node[x] = merge(node[2*x+1], node[2*x+2]);
      }
    }

    void add(int x, T val) {
      x += (N - 1);
      node[x] += val;
      while(x > 0) {
        x = (x - 1) / 2;
        node[x] = merge(node[2*x+1], node[2*x+2]);
      }
    }

    // query for [l, r)
    T query(int a, int b, int k=0, int l=0, int r=-1) {
      if(r < 0) r = N;
      if(r <= a || b <= l) return identity;
      if(a <= l && r <= b) return node[k];

      T vl = query(a, b, 2*k+1, l, (l+r)/2);
      T vr = query(a, b, 2*k+2, (l+r)/2, r);
      return merge(vl, vr);
    }

    friend ostream& operator<<(ostream &os, SegmentTree<T>& sg){ //
      os << "[";
      for(int i=0; i<sg.n; i++){
        os << sg[i] << (i==sg.n-1?"]\n":", ");
      }
      return os;
    }
};
// }}}
struct tree{/*{{{*/
  int n;
  vector<int> par;  // par[i]: dfs木における親
  vector<int> cost;  // par[i]: dfs木における親への辺のコスト
  vector<int> dfstrv; // dfstrv[i]: dfs木でi番目に訪れるノード。dpはこれを逆順に回す
  vector<int> ord;  // ord[u]: uのdfs木における訪問順
  vector<int> pos;  // pos[u]: uのdfs終了時のカウンター
  vector<int> psize; // psize[u]: uのpartial tree size
  // vの部分木は[ord[v], pos[v])
  // ordとdfstrvは逆変換

  vector<int> depth;  // depth[i]: dfs木でのiの深さ
  vector<int> ldepth;  //  ldepth[i]: dfs木でのrootからの距離
  vector<vector<pair<int, int>>> g; // 辺(隣接リスト)
  vector<vector<int>> children;
  vector<int> euler_tour;
  vector<int> et_fpos;  // euler_tour first occurence position
  SegmentTree<int> _seg;  // seg(map(ord, euler_tour), mymin, 1e18)

  int _counter = 0;

  tree(int n):
    n(n),par(n),cost(n,1),ord(n),pos(n),
    psize(n),depth(n),ldepth(n),g(n),children(n),et_fpos(n)
  {};

  void add_edge(int u, int v, int cost){
    g[u].emplace_back(v, cost);
    g[v].emplace_back(u, cost);
  }
  void add_edge(int u, int v){
    g[u].emplace_back(v, 1);
    g[v].emplace_back(u, 1);
  }

  void build(int root){
    _counter = 0;
    par[root] = -1;
    cost[root] = INF;
    _dfs_tree(root, -1);
    // _dfs_et(root);
    // vector<int> ini(2*n-1); rep(i, 2*n-1) ini[i] = ord[euler_tour[i]];
    // _seg = SegmentTree<int>(ini, [](auto a, auto b){return min(a,b);}, 1e18);
  }

  void _dfs_tree(int u, int pre){
    dfstrv.pb(u);
    ord[u] = _counter;
    if (pre!=-1){
      depth[u] = depth[pre]+1; 
      ldepth[u] = ldepth[pre]+cost[u]; 
    }

    _counter++;
    each(el, g[u]){
      int v = el.first;
      if (v==pre) continue;

      children[u].pb(v);
      par[v] = u;
      cost[v] = el.second;
      _dfs_tree(v, u);
    }
    pos[u] = _counter;
    psize[u] = pos[u] - ord[u];
  }

  void _dfs_et(int u){
    et_fpos[u] = euler_tour.size();
    euler_tour.pb(u);
    each(v, children[u]){
      _dfs_et(v);
      euler_tour.pb(u);
    }
  }

  int lca(int u, int v){
    if (u==v) return u;
    if (et_fpos[u]>et_fpos[v]) swap(u, v);
    return dfstrv[_seg.query(et_fpos[u], et_fpos[v])];
  }

  int dist(int u, int v){
    int p = lca(u, v);
    return depth[u] + depth[v] - 2*depth[p];
  }
  int ldist(int u, int v){  // length dist
    int p = lca(u, v);
    return ldepth[u] + ldepth[v] - 2*ldepth[p];
  }
  pair<int, int> diameter(){
    int u, v;
    int max_len = *max_element(all(ldepth));
    rep(i, n){
      if(ldepth[i]==max_len){
        u = i; break;
      }
    }
    int md = -1;
    rep(i, n){
      int d = ldist(u, i);
      if (d>md){ v = i; md = d; }
    }
    return mp(u, v);
  }

};/*}}}*/

signed main() {
  int n,k;cin>>n>>k;
  if (k>n){
    cout << -1 << endl;
    return 0;
  }
  tree tr(n);
  rep(i, n-1){
    int a,b;cin>>a>>b;
    a--;b--;
    tr.add_edge(a, b);
    tr.add_edge(b, a);
  }
  tr.build(0);

  int ans = 0;
  sort(all(tr.depth));
  rep(i, k){
    ans += tr.depth[i];
  }
  cout << ans << endl;

  return 0;
}
0