結果

問題 No.1216 灯籠流し/Lanterns
ユーザー msm1993msm1993
提出日時 2020-10-08 10:24:09
言語 C++14
(gcc 13.2.0 + boost 1.83.0)
結果
AC  
実行時間 308 ms / 4,500 ms
コード長 5,625 bytes
コンパイル時間 2,037 ms
コンパイル使用メモリ 117,692 KB
実行使用メモリ 35,564 KB
最終ジャッジ日時 2023-09-27 11:46:03
合計ジャッジ時間 16,048 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,380 KB
testcase_01 AC 1 ms
4,376 KB
testcase_02 AC 2 ms
4,376 KB
testcase_03 AC 2 ms
4,376 KB
testcase_04 AC 53 ms
21,668 KB
testcase_05 AC 243 ms
31,604 KB
testcase_06 AC 21 ms
6,420 KB
testcase_07 AC 197 ms
22,596 KB
testcase_08 AC 153 ms
15,504 KB
testcase_09 AC 91 ms
15,716 KB
testcase_10 AC 83 ms
12,228 KB
testcase_11 AC 63 ms
15,396 KB
testcase_12 AC 43 ms
19,896 KB
testcase_13 AC 96 ms
24,300 KB
testcase_14 AC 44 ms
15,448 KB
testcase_15 AC 49 ms
15,704 KB
testcase_16 AC 83 ms
26,164 KB
testcase_17 AC 61 ms
8,000 KB
testcase_18 AC 48 ms
14,948 KB
testcase_19 AC 118 ms
20,228 KB
testcase_20 AC 78 ms
18,092 KB
testcase_21 AC 63 ms
17,324 KB
testcase_22 AC 78 ms
12,344 KB
testcase_23 AC 71 ms
25,620 KB
testcase_24 AC 15 ms
5,956 KB
testcase_25 AC 12 ms
5,892 KB
testcase_26 AC 48 ms
9,832 KB
testcase_27 AC 49 ms
14,756 KB
testcase_28 AC 63 ms
21,724 KB
testcase_29 AC 80 ms
23,564 KB
testcase_30 AC 73 ms
21,420 KB
testcase_31 AC 68 ms
11,436 KB
testcase_32 AC 92 ms
24,028 KB
testcase_33 AC 43 ms
14,408 KB
testcase_34 AC 256 ms
32,244 KB
testcase_35 AC 287 ms
35,564 KB
testcase_36 AC 191 ms
33,148 KB
testcase_37 AC 150 ms
26,944 KB
testcase_38 AC 137 ms
26,956 KB
testcase_39 AC 269 ms
32,116 KB
testcase_40 AC 308 ms
34,692 KB
testcase_41 AC 306 ms
34,860 KB
testcase_42 AC 308 ms
34,664 KB
testcase_43 AC 307 ms
34,672 KB
testcase_44 AC 298 ms
34,744 KB
testcase_45 AC 152 ms
29,056 KB
testcase_46 AC 153 ms
29,320 KB
testcase_47 AC 152 ms
29,068 KB
testcase_48 AC 147 ms
29,072 KB
testcase_49 AC 149 ms
29,112 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <algorithm>
#include <string>
#include <vector>
#include <cmath>
#include <map>
#include <queue>
#include <iomanip>
#include <set>
#include <tuple>
#define mkp make_pair
#define mkt make_tuple
#define rep(i,n) for(int i = 0; i < (n); ++i)
#define all(v) v.begin(),v.end()
using namespace std;
typedef long long ll;
const ll MOD=1e9+7;
template<class T> void chmin(T &a,const T &b){if(a>b) a=b;}
template<class T> void chmax(T &a,const T &b){if(a<b) a=b;}

class BIT{//1-indexed
public:

  vector<ll> bit;
  BIT(){}
  BIT(int size){
    bit.resize(size,0);
  }

  void initialize(int size){
      bit.resize(size,0);
  }

  ll sum(int i){
    ll s=0;
    while(i>0){
      s+=bit[i];
      i-=i&(-i);
    }
    return s;
  }

  void add(int i,ll x){//i!=0
    while(i<bit.size()){
      bit[i]+=x;
      i+=i&(-i);
    }
  }
};


struct Edge{
    int to,id;
    ll dist;
    Edge(int to,ll dist=1,int id=0):to(to),dist(dist),id(id){}
};

// this class's update method is empty.
struct HeavyLightDecomposition{
    vector<vector<Edge>> g;
    vector<int> in,out,head,par,dep,sz;
    int times;
    int root;
    vector<int> rev;
    vector<ll> dist;
    vector<ll> disrev;

    HeavyLightDecomposition(int V,vector<vector<Edge>> &G,int root=0):
        g(G),in(V),out(V),head(V),par(V),dep(V),sz(V),root(root),dist(V),rev(V),disrev(V){
            times=0;
            sz_dfs(root,-1);
            hld_dfs(root,-1);
            for(int i=0;i<V;i++) disrev[i]=dist[rev[i]];
    }
    
    void sz_dfs(int now,int p){
        par[now]=p;
        sz[now]=1;
        if(p==-1){
            dep[now]=0;
            dist[now]=0;
        }else dep[now]=dep[p]+1;

        for(auto &e:g[now]){
            if(e.to==p) continue;
            dist[e.to]=dist[now]+e.dist;
            sz_dfs(e.to,now);
            sz[now]+=sz[e.to];
            if(sz[e.to]>sz[g[now][0].to]) swap(e,g[now][0]);
        }
    }

    void hld_dfs(int now,int p){
        in[now]=times++;
        rev[in[now]]=now;
        for(auto e:g[now]){
            if(e.to==p) continue;
            head[e.to]=(e.to == g[now][0].to ? head[now] : e.to);
            hld_dfs(e.to,now);
        }
        out[now]=times;
    }


    int lca(int u,int v){
        for(;;v=par[head[v]]){
            if(in[u]>in[v]) swap(u,v);
            if(head[u]==head[v]) return u;
        }
    }
    int distance(int u,int v){
        return dep[u]+dep[v]-2*dep[lca(u,v)];
    }

    int climb(int v,ll lim){
        for(;;v=par[head[v]]){
            int u=head[v];
            if(u==0||dist[par[u]]+lim<dist[v]){
                int z=lower_bound(disrev.begin()+in[u],disrev.begin()+in[v]+1,dist[v]-lim)-disrev.begin();
                return rev[z];
            }
            lim-=(dist[v]-dist[par[u]]);
        }
    }

    /*ex)
        hld.update(u,x,[&](int a,int b,ll x){
            seg.add(a,b,x);
        });
    */
    template<class T,class G>
    void update(int v,const T &x,const G &g){//辺の時は0を使わない(INFのまま)
        //g(in[v]+1,out[v],x);//部分木(辺)
        //g(in[v],out[v],x);//部分木（頂点）
        //g(in[v],in[v]+1,x);//一点
    }

    /*ex)
        ll ans=0;
        hld.query(u,v,[&](int a,int b){
            ans+=seg.getSum(a,b);
        },true);
    */
    template<class F>
    void query(int u,int v,ll x,const F &f,bool isedge){
        for(;;v=par[head[v]]){
            if(in[u]>in[v]) swap(u,v);
            if(head[u]==head[v]) break;
            f(in[head[v]],in[v]+1,x);
        }
        if(isedge&&u==v) return;
        f(in[u]+isedge,in[v]+1,x);
    }
};

int main(){
  cin.tie(0);
  ios::sync_with_stdio(false);

  int N,Q;
  cin>>N>>Q;
  vector<vector<Edge>> g(N);
  rep(i,N-1){
      int a,b;
      ll c;
      cin>>a>>b>>c;
      a--;b--;
      g[a].push_back({b,c});
      g[b].push_back({a,c});
  }
  vector<int> type(Q),V(Q);
  vector<ll> T(Q),L(Q);
  rep(i,Q) cin>>type[i]>>V[i]>>T[i]>>L[i];
  rep(i,Q) V[i]--;

  HeavyLightDecomposition hld(N,g);
  int sz=1;
  while(sz<N) sz*=2;
  vector<vector<ll>> segment(2*sz);
  auto preadd = [&](int l,int r,ll x){
      l+=sz;r+=sz;
      while(l<r){
          if(l&1) segment[l++].push_back(x);
          if(r&1) segment[--r].push_back(x);
          l>>=1;r>>=1;
      }
  };
  vector<int> up(Q,-1);
  for(int q=0;q<Q;q++){
      if(type[q]==1) continue;
      up[q]=hld.climb(V[q],L[q]);
      hld.query(V[q],up[q],T[q]+hld.dist[V[q]],[&](int l,int r,ll x){
          preadd(l,r,x);
      },false);
  }

  vector<BIT> bit(2*sz);
  for(int i=0;i<2*sz;i++){
      sort(all(segment[i]));
      segment[i].erase(unique(all(segment[i])),segment[i].end());
      bit[i].initialize(segment[i].size()+1);
  }

  auto add = [&](int l,int r,ll x){
      l+=sz;r+=sz;
      while(l<r){
          if(l&1){
              int z=lower_bound(all(segment[l]),x)-segment[l].begin();
              bit[l].add(z+1,1);
              l++;
          }
          if(r&1){
              --r;
              int z=lower_bound(all(segment[r]),x)-segment[r].begin();
              bit[r].add(z+1,1);
          }
          l>>=1;r>>=1;
      }
  };
  for(int q=0;q<Q;q++){
      if(type[q]==0){
          hld.query(V[q],up[q],T[q]+hld.dist[V[q]],[&](int l,int r,ll x){
              add(l,r,x);
          },false);
      }else{
          ll target=T[q]+hld.dist[V[q]];
          int ans=0;
          int now=hld.in[V[q]]+sz;
          while(now>=1){
              int z=upper_bound(all(segment[now]),target)-segment[now].begin();
              ans+=bit[now].sum(z);
              now>>=1;
          }
          cout<<ans<<"\n";
      }
  }

  return 0;
}
0