#913555 (C++23) No.2325 Skill Tree

提出ソース

結果

問題	No.2325 Skill Tree
ユーザー	eQe
提出日時	2023-09-16 20:12:32
言語	C++23 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	211 ms / 3,000 ms
コード長	7,230 bytes
コンパイル時間	6,042 ms
コンパイル使用メモリ	331,444 KB
実行使用メモリ	29,628 KB
最終ジャッジ日時	2024-07-03 19:30:32
合計ジャッジ時間	17,467 ms
ジャッジサーバーID （参考情報）	judge4 / judge2

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 5,248 KB
testcase_01	AC	2 ms 5,376 KB
testcase_02	AC	2 ms 5,376 KB
testcase_03	AC	2 ms 5,376 KB
testcase_04	AC	1 ms 5,376 KB
testcase_05	AC	2 ms 5,376 KB
testcase_06	AC	2 ms 5,376 KB
testcase_07	AC	56 ms 9,748 KB
testcase_08	AC	61 ms 12,276 KB
testcase_09	AC	72 ms 12,452 KB
testcase_10	AC	105 ms 17,568 KB
testcase_11	AC	88 ms 15,824 KB
testcase_12	AC	179 ms 29,628 KB
testcase_13	AC	189 ms 27,164 KB
testcase_14	AC	183 ms 26,964 KB
testcase_15	AC	182 ms 26,964 KB
testcase_16	AC	182 ms 27,088 KB
testcase_17	AC	163 ms 25,428 KB
testcase_18	AC	161 ms 25,428 KB
testcase_19	AC	164 ms 25,428 KB
testcase_20	AC	156 ms 25,428 KB
testcase_21	AC	159 ms 25,556 KB
testcase_22	AC	197 ms 28,888 KB
testcase_23	AC	196 ms 28,504 KB
testcase_24	AC	206 ms 28,628 KB
testcase_25	AC	196 ms 28,632 KB
testcase_26	AC	193 ms 28,756 KB
testcase_27	AC	192 ms 26,832 KB
testcase_28	AC	197 ms 26,816 KB
testcase_29	AC	192 ms 26,708 KB
testcase_30	AC	195 ms 26,708 KB
testcase_31	AC	193 ms 26,708 KB
testcase_32	AC	171 ms 25,044 KB
testcase_33	AC	166 ms 25,048 KB
testcase_34	AC	171 ms 25,172 KB
testcase_35	AC	208 ms 28,368 KB
testcase_36	AC	211 ms 28,204 KB
testcase_37	AC	205 ms 28,372 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include<bits/stdc++.h>
#include<atcoder/all>
namespace my{
using namespace atcoder;
using ml=modint998244353;
auto&operator<<(std::ostream&o,const ml&x){return o<<x.val();}
auto&operator>>(std::istream&i,ml&x){long long t;i>>t;x=t;return i;}
}
namespace my{void main();}
int main(){my::main();}
namespace my{
#define eb emplace_back
#define all(a) (a).begin(),(a).end()
#define RD(T,...) T __VA_ARGS__;li(__VA_ARGS__)
#define LL(...) RD(ll,__VA_ARGS__)
#define VV(n,...) vl __VA_ARGS__;setsize({n},__VA_ARGS__);vi(__VA_ARGS__)
#define JO(a,b) a##b
#define jo(a,b) JO(a,b)
#define usymbol(a) jo(a,__LINE__)
#define FE(v,e,...) for(au&&__VA_OPT__([)e __VA_OPT__(,__VA_ARGS__]):v)
#define FO(n) for(ll usymbol(i)=n;usymbol(i)-->0;)
#define FOR(i,...) for(au[i,i##O,i##E]=rng(0,__VA_ARGS__);i<i##O;i+=i##E)
#define fe(v,...) FE(v,__VA_ARGS__)
#define fo(i,...) FO##__VA_OPT__(R)(i __VA_OPT__(,__VA_ARGS__))
#define of(i,...) for(au[i,i##O,i##E]=rng(1,__VA_ARGS__);i>=i##O;i-=i##E)
#define I template
#define J class
#define au auto
#define rr return
using namespace std;
using vo=void;using bo=bool;
vo solve();
I<J A,J B>constexpr bo same=is_same_v<A,B>;
using is=istream;using os=ostream;
using dd=long double;using ll=long long;using ull=unsigned long long;
using lll=__int128_t;using ulll=__uint128_t;using str=string;
os&operator<<(os&o,const ulll&x){rr(x<10?o:o<<x/10)<<ll(x%10);}
os&operator<<(os&o,const lll&x){rr o<<str(x<0,'-')<<ulll(x>0?x:-x);}
constexpr dd ee=1e-12;
constexpr ll oo=3e18;
constexpr ll dx[]{-1,0,1,0,-1,1,1,-1},dy[]{0,-1,0,1,-1,-1,1,1};
constexpr char sp=32;
constexpr char nl=10;
au rng(bo s,ll a,ll b=oo,ll c=1){if(b==oo)b=a,(s?b:a)=0;rr tuple{a-s,b,c};}
I<J T>ll len(const T&a){rr a.size();}
I<J T>bo amax(T&a,const T&b){rr a<b?a=b,1:0;}
I<J...A>au max(const A&...a){rr max(initializer_list<common_type_t<A...>>{a...});}

I<J A,J B=A>struct cp{
  A a={};B b={};
  cp(){}
  cp(A a,B b):a(a),b(b){}
  cp(pair<A,B>p):a(p.first),b(p.second){}
  bo operator==(const cp&c)const{rr a==c.a&&b==c.b;}
  bo operator<(const cp&c)const{rr a!=c.a?a<c.a:b<c.b;}
  bo operator>(const cp&c)const{rr a!=c.a?a>c.a:b>c.b;}

  friend is&operator>>(is&i,cp&c){rr i>>c.a>>c.b;}
  friend os&operator<<(os&o,const cp&c){rr o<<c.a<<sp<<c.b;}
};using cl=cp<ll>;
I<J A,J B=A,J C=A>struct tr{
  A a={};B b={};C c={};
  tr(){}
  tr(A a,B b,C c):a(a),b(b),c(c){}
  bo operator==(const tr&t)const{rr a==t.a&&b==t.b&&c==t.c;}

  au operator<(const tr&t)const{rr a!=t.a?a<t.a:b!=t.b?b<t.b:c<t.c;}
  friend is&operator>>(is&i,tr&t){rr i>>t.a>>t.b>>t.c;}
  friend os&operator<<(os&o,const tr&t){rr o<<t.a<<sp<<t.b<<sp<<t.c;}
};using tl=tr<ll>;

I<J T>T&srt(T&a){sort(all(a));rr a;}
I<J T>decltype(au)bg(T&a){rr*begin(a);}
I<J T>au pof(T&a){au r=bg(a);a.pop_front();rr r;}

I<J T,size_t n>using ar=array<T,n>;
I<J T,size_t n>os&operator<<(os&o,const ar<T,n>&a){fo(i,n)o<<a[i]<<str(i!=n-1,sp);rr o;}
I<J T,J U=T>using map=std::map<T,U>;
I<J T,J U>os&operator<<(os&o,const map<T,U>&m){fe(m,e)o<<e.first<<sp<<e.second<<nl;rr o;}
I<J T,J U=T>using umap=unordered_map<T,U>;
I<J T,J U>os&operator<<(os&o,const umap<T,U>&m){fe(m,e)o<<e.first<<sp<<e.second<<nl;rr o;}
I<size_t n>using bset=bitset<n>;
I<size_t n>os&operator<<(os&o,const bset<n>&b){fo(i,n)o<<b[i];rr o;}

I<J...A>os&operator<<(os&o,const tuple<A...>&t){apply([&](const au&...a){ll i=sizeof...(a);(((o<<a<<str(--i>0,sp))),...);},t);rr o;}

I<J T>struct deq:deque<T>{
  deq(const initializer_list<T>&a={}){fe(a,e)this->eb(e);}
  deq(const vector<T>&a){fe(a,e)this->eb(e);}
  friend os&operator<<(os&o,const deq&q){fe(q,e)o<<e<<str(&e!=&q.back(),sp);rr o;}
};

I<J T,J F>struct prique:priority_queue<T,vector<T>,F>{
  prique(const initializer_list<T>&a={}){fe(a,e)this->emplace(e);}
  prique(const vector<T>&a){fe(a,e)this->emplace(e);}
  friend T bg(prique&q){rr q.top();}
  friend T pot(prique&q){T r=q.top();q.pop();rr r;}
  friend os&operator<<(os&o,prique q){while(len(q))o<<pot(q)<<str(len(q)>0,sp);rr o;}
};
I<J T>using qmax=prique<T,less<>>;
I<J T>using qmin=prique<T,greater<>>;

I<J V>struct ve;
I<J V>constexpr bo isv=0;
I<J V>constexpr bo isv<ve<V>> =1;
I<J V>constexpr bo isv<vector<V>> =1;
I<J V>au rawv(V){if constexpr(isv<V>)rr rawv(V(1)[0]);else rr V();}

I<J V>struct ve:vector<V>{
  using vector<V>::vector;
  using T=decltype(rawv(V()));
  I<J U>ve(const vector<U>&v={}){static_assert(isv<V> ==isv<U>);fe(v,e)this->eb(e);}

  ve&operator+=(const ve&u){au&v=*this;fo(i,len(v))v[i]+=u[i];rr v;}ve operator+(const ve&u)const{rr ve(*this)+=u;}
  ve&operator-=(const ve&u){au&v=*this;fo(i,len(v))v[i]-=u[i];rr v;}ve operator-(const ve&u)const{rr ve(*this)-=u;}
  ve&operator^=(const ve&u){fe(u,e)this->eb(e);rr*this;}ve operator^(const ve&u)const{rr ve(*this)^=u;}
  ve&operator+=(const T&x){au&v=*this;fe(v,e)e+=x;rr v;}ve operator+(const T&x)const{rr ve(*this)+=x;}
  ve&operator-=(const T&x){au&v=*this;fe(v,e)e-=x;rr v;}ve operator-(const T&x)const{rr ve(*this)-=x;}
  ve&operator*=(const T&x){au&v=*this;fe(v,e)e*=x;rr v;}ve operator*(const T&x)const{rr ve(*this)*=x;}
  ve&operator++(){rr*this+=1;}
  ve&operator--(){rr*this-=1;}
  ve operator-()const{rr ve(*this)*=-1;}
  I<size_t n>au&operator+=(const bset<n>&a){fo(i,n)(*this)[i]+=a[i];rr*this;}
  I<size_t n>au&operator-=(const bset<n>&a){fo(i,n)(*this)[i]-=a[i];rr*this;}

  au lower_bound(const V&x)const{rr std::lower_bound(all(*this),x);}
  au upper_bound(const V&x)const{rr std::upper_bound(all(*this),x);}

  I<J F>au scan(F f)const{cp<T,bo>r;fe(*this,e)if constexpr(!isv<V>)r.b?f(r.a,e),r:r={e,1};else if(au s=e.scan(f);s.b)r.b?f(r.a,s.a),r:r=s;rr r;}
  T max()const{rr scan([](T&a,const T&b){a<b?a=b:0;}).a;}

  vo af(const V&x={}){this->emplace(this->begin(),x);}

  friend is&operator>>(is&i,ve&v){fe(v,e)i>>e;rr i;}
  friend os&operator<<(os&o,const ve&v){fe(v,e)o<<e<<str(&e!=&v.back(),isv<V>?nl:sp);rr o;}
};
I<J T=ll,size_t n,size_t i=0>au vec(const ll(&s)[n],T x={}){if constexpr(n==i+1)rr ve<T>(s[i],x);else{au X=vec<T,n,i+1>(s,x);rr ve<decltype(X)>(s[i],X);}}
I<ll n,J...A>vo setsize(const ll(&l)[n],A&...a){((a=vec(l,rawv(a))),...);}
I<J T>using vve=ve<ve<T>>;using vl=ve<ll>;using vvl=vve<ll>;
using gl=vve<ll>;using ge=vve<cl>;

vo io(){cin.tie(0)->sync_with_stdio(0);cout<<fixed<<setprecision(15);cerr<<nl;}
I<J...A>vo li(A&...a){(cin>>...>>a);}
I<J...A>vo vi(A&...a){fo(i,(len(a)&...))(cin>>...>>a[i]);}
I<char c=sp,J...A>vo pp(const A&...a){ll i=sizeof...(a);((cout<<a<<str(--i>0,c)),...);cout<<nl;}

I<J F>struct rec:F{rec(F&&f):F(std::forward<F>(f)){}I<J...A>decltype(au)operator()(A&&...a)const{rr F::operator()(*this,std::forward<A>(a)...);}};
vo main(){io();ll T=1;fo(T)solve();}
vo solve(){
  LL(N);
  VV(N-1,l,a);--a;
  l.af(),a.af();

  LL(Q);
  VV(Q,f,x);
  fo(i,Q)if(f[i]==2)x[i]--;

  ve<cl>v;
  fo(i,1,N)v.eb(l[i],~i);
  fo(i,Q)if(f[i]==1)v.eb(x[i],i);
  srt(v);

  vl res(Q,-1);
  dsu uf(N);
  gl g(N);

  fo(i,len(v)){
    if(v[i].b<0){
      ll k=~v[i].b;
      uf.merge(k,a[k]);
      g[a[k]].eb(k);
    }else{
      ll k=v[i].b;
      res[k]=uf.size(0);
    }
  }

  deq<ll>q{0};
  vl dp(N,-1);
  while(len(q)){
    ll u=pof(q);
    fe(g[u],v){
      amax(dp[v],max(dp[u],l[v]));
      q.eb(v);
    }
  }

  fo(i,Q)if(f[i]==2)res[i]=dp[x[i]];

  fo(i,Q)pp(res[i]);
}}

yukicoder

結果

テストケース

ソースコード