#if !defined(MYLOCAL)//提出時用テンプレート #pragma GCC optimize("Ofast") #if defined(NDEBUG) #undef NDEBUG #endif #include "bits/stdc++.h" #if !defined(_MSC_VER) && __has_include() #include using namespace atcoder; #endif using namespace std; using ll=long long; using dd=double; using vll=vector< ll>; using vdd=vector< dd>; using vvll=vector< vll>; using vvdd=vector; using vvvll=vector< vvll>; using vvvdd=vector; using vvvvll=vector; using pll=pair; using tll=tuple; using qll=tuple; using vpll=vector< pll>; using vtll=vector< tll>; using vqll=vector< qll>; using vvpll=vector; using vvtll=vector; using vvqll=vector; using namespace chrono; constexpr ll INF = 1001001001001001001; struct Fast{ Fast(){ cin.tie(0); ios::sync_with_stdio(false); cout<::max_digits10); } } fast; #define REPS(i, S, E) for (ll i = (S); i <= (E); i++) #define REP(i, N) REPS(i, 0, (N)-1) #define DEPS(i, S, E) for (ll i = (E); i >= (S); i--) #define DEP(i, N) DEPS(i, 0, (N)-1) #define EXPAND( x ) x//VS用おまじない #define overload3(_1,_2,_3,name,...) name #define overload4(_1,_2,_3,_4,name,...) name #define overload5(_1,_2,_3,_4,_5,name,...) name #define rep3(i, S, E) for (ll i = (S); i <= (E); i++) #define rep4(i, S, E, t) for (ll i = (S); i <= (E); i+=(t)) #define rep(...) EXPAND(overload4(__VA_ARGS__,rep4,rep3,_,_)(__VA_ARGS__)) #define dep3(i, E, S) for (ll i = (E); i >= (S); i--) #define dep4(i, E, S, t) for (ll i = (E); i >= (S); i-=(t)) #define dep(...) EXPAND(overload4(__VA_ARGS__, dep4, dep3,_,_)(__VA_ARGS__)) #define each2(e,v) for (auto && e:v) #define each3(a,b,v) for (auto &&[a,b]:v) #define each4(a,b,c,v) for (auto &&[a,b,c]:v) #define each5(a,b,c,d,v) for (auto &&[a,b,c,d]:v) #define each(...) EXPAND(overload5(__VA_ARGS__,each5,each4,each3,each2,_)(__VA_ARGS__)) #define ALL1(v) (v).begin(), (v).end() #define ALL2(v,E) (v).begin(), (v).begin()+((E)+1) #define ALL3(v,S,E) (v).begin()+(S), (v).begin()+((E)+1) #define ALL(...) EXPAND(overload3(__VA_ARGS__, ALL3, ALL2, ALL1)(__VA_ARGS__)) #define all ALL #define RALL1(v) (v).rbegin(), (v).rend() #define RALL2(v,E) (v).rbegin(), (v).rbegin()+((E)+1) #define RALL3(v,S,E) (v).rbegin()+(S), (v).rbegin()+((E)+1) #define RALL(...) EXPAND(overload3(__VA_ARGS__, RALL3, RALL2, RALL1)(__VA_ARGS__)) #define rall RALL template inline bool chmax(T &a, T b) { if (a < b) { a = b; return true; }return false; } template inline bool chmin(T &a, T b) { if (a > b) { a = b; return true; }return false; } template inline T MaxE(vector&v,ll S,ll E){ T m=v[S]; rep(i,S,E)chmax(m,v[i]); return m; } template inline T MinE(vector&v,ll S,ll E){ T m=v[S]; rep(i,S,E)chmin(m,v[i]); return m; } template inline T MaxE(vector &v) { return MaxE(v,0,(ll)v.size()-1); } template inline T MinE(vector &v) { return MinE(v,0,(ll)v.size()-1); } template inline auto maxe(T &&v,ll S,ll E){ return *max_element(ALL(v,S,E)); } template inline auto maxe(T &&v){ return *max_element(ALL(v)); } template inline auto mine(T &&v,ll S,ll E){ return *min_element(ALL(v,S,E)); } template inline auto mine(T &&v){ return *min_element(ALL(v)); } template inline T Sum(vector &v,ll S,ll E){ T s=T(); rep(i,S,E)s+=v[i]; return s; } template inline T Sum(vector &v) { return Sum(v,0,v.size()-1); } template::type::value_type> inline U sum(T &&v,ll S,ll E) {return accumulate(all(v,S,E),U());} template inline auto sum(T &&v) {return sum(v,0,v.end()-v.begin()-1);} template inline ll sz(T &&v){ return (ll)v.size(); } inline ll CEIL(ll a,ll b){ return (a<0) ? -(-a/b) : (a+b-1)/b; } //負もOK inline ll FLOOR(ll a,ll b){ return -CEIL(-a,b); } //負もOK //pair用テンプレート template inline pair& operator+=(pair &a,const pair &b){ a.first+=b.first; a.second+=b.second; return a; } template inline pair& operator-=(pair &a,const pair &b){ a.first-=b.first; a.second-=b.second; return a; } template inline pair& operator*=(pair &a,const pair &b){ a.first*=b.first; a.second*=b.second; return a; } template inline pair& operator/=(pair &a,const pair &b){ a.first/=b.first; a.second/=b.second; return a; } template inline pair& operator%=(pair &a,const pair &b){ a.first%=b.first; a.second%=b.second; return a; } template inline pair& operator+=(pair &a,R b){ a.first+=b; a.second+=b; return a; } template inline pair& operator-=(pair &a,R b){ a.first-=b; a.second-=b; return a; } template inline pair& operator*=(pair &a,R b){ a.first*=b; a.second*=b; return a; } template inline pair& operator/=(pair &a,R b){ a.first/=b; a.second/=b; return a; } template inline pair& operator%=(pair &a,R b){ a.first%=b; a.second%=b; return a; } template inline pair operator+(const pair &a,R b){ pair c=a; return c+=b; } template inline pair operator-(const pair &a,R b){ pair c=a; return c-=b; } template inline pair operator*(const pair &a,R b){ pair c=a; return c*=b; } template inline pair operator/(const pair &a,R b){ pair c=a; return c/=b; } template inline pair operator%(const pair &a,R b){ pair c=a; return c%=b; } template inline pair operator-(R b,const pair &a){ pair c=-a; return c+=b; } template inline pair operator-(const pair &a,const pair &b){ pair c=a; return c-=b; } template inline pair operator-(const pair &a){ pair c=a; return c*=(-1); } template inline ostream &operator<<(ostream &os,const pair &a){ return os << a.first << ' ' << a.second; } //tuple用テンプレート 出力用のみ template inline ostream &operator<<(ostream &os,const tuple &a){ return os << get<0>(a) << ' ' << get<1>(a) << ' ' << get<2>(a); } template inline ostream &operator<<(ostream &os,const tuple &a){ return os << get<0>(a) << ' ' << get<1>(a) << ' ' << get<2>(a) << ' ' << get<3>(a); } //vector用テンプレート template inline ostream &operator<<(ostream &os,const vector &a){ for (ll i=0; i<(ll)a.size(); i++) os<<(i>0?" ":"")< inline array& operator+=(array &a,const array &b){ for (ll i=0; i<(ll)S; i++) a[i]+=b[i]; return a; } template inline array& operator-=(array &a,const array &b){ for (ll i=0; i<(ll)S; i++) a[i]-=b[i]; return a; } template inline array& operator*=(array &a,const array &b){ for (ll i=0; i<(ll)S; i++) a[i]*=b[i]; return a; } template inline array& operator/=(array &a,const array &b){ for (ll i=0; i<(ll)S; i++) a[i]/=b[i]; return a; } template inline array& operator%=(array &a,const array &b){ for (ll i=0; i<(ll)S; i++) a[i]%=b[i]; return a; } template inline array& operator+=(array &a,R b){ for (T &e: a) e+=b; return a; } template inline array& operator-=(array &a,R b){ for (T &e: a) e-=b; return a; } template inline array& operator*=(array &a,R b){ for (T &e: a) e*=b; return a; } template inline array& operator/=(array &a,R b){ for (T &e: a) e/=b; return a; } template inline array& operator%=(array &a,R b){ for (T &e: a) e%=b; return a; } template inline array operator+(const array &a,R b){ array c=a; return c+=b; } template inline array operator-(const array &a,R b){ array c=a; return c-=b; } template inline array operator*(const array &a,R b){ array c=a; return c*=b; } template inline array operator/(const array &a,R b){ array c=a; return c/=b; } template inline array operator%(const array &a,R b){ array c=a; return c%=b; } template inline array operator-(R b,const array &a){ array c=-a; return c+=b; } template inline array operator-(const array &a,const array &b){ array c=a; return c-=b; } template inline array operator-(const array &a){ array c=a; return c*=(-1); } template inline ostream &operator<<(ostream &os,const array &a){ for (ll i=0; i<(ll)S; i++) os<<(i>0?" ":"")< struct view1d; template struct view1dIter{ view1d *vw=nullptr; ll idx=INF; view1dIter(){} view1dIter(view1d *vw_,ll idx_):vw(vw_),idx(idx_){} view1dIter(const view1dIter &it):vw(it.vw),idx(it.idx){} R &operator*() { return (*vw)[idx]; } R &operator*() const{ return (*vw)[idx]; } R &operator[](ll i) { return (*vw)[idx+i]; } R &operator[](ll i)const{ return (*vw)[idx+i]; } auto &operator++(){ idx++; return *this; } auto &operator--(){ idx--; return *this; } auto operator++(int){ auto it=*this; idx++; return it; } auto operator--(int){ auto it=*this; idx--; return it; } auto &operator+=(ll n){ idx+=n; return *this; } auto &operator-=(ll n){ idx-=n; return *this; } auto operator+(ll n){ auto it=*this; return it+=n; } auto operator-(ll n){ auto it=*this; return it-=n; } ll operator- (const view1dIter &it)const{ return idx- it.idx; } bool operator< (const view1dIter &it)const{ return idx< it.idx; } bool operator> (const view1dIter &it)const{ return idx> it.idx; } bool operator<=(const view1dIter &it)const{ return idx<=it.idx; } bool operator>=(const view1dIter &it)const{ return idx>=it.idx; } bool operator!=(const view1dIter &it)const{ return idx!=it.idx; } bool operator==(const view1dIter &it)const{ return idx==it.idx; } using iterator_category = random_access_iterator_tag; using value_type = R; using difference_type = ll; using pointer = R*; using reference = R&; }; template struct view1d{ using Sll=array; T &data; //参照先データ Sll dsize; //参照先データ各軸のサイズ Sll s=Sll(); //view先頭 Sll d=Sll(); //view方向 ll len; //view長さ R &dummy; //範囲外の値参照 R dummyj=R(); //範囲外の値実体 /*---- コンストラクタ ----*/ view1d(T &v): data(v),dummy(dummyj){ SetDsize(v,dsize); d[S-1]=1; len=dsize[S-1]; } //このコンストラクタは、マイナス座標指定を末尾にしない(暫定) //代入時Array{...}等と書くため、使い勝手悪い。将来廃止するかも。 view1d(T &v,Sll s,Sll e,Sll d): data(v),s(s),d(d),len(Len(s,e,d)),dummy(dummyj){ SetDsize(v,dsize); } view1d(T &v,Sll s,Sll d,ll len,R &dmy_): data(v),s(s),d(d),len(len),dummy(dmy_){ SetDsize(v,dsize); } /*---- 演算 ----*/ template auto &operator =(const Q &b){ rep(i,0,len-1) (*this)[i] =b; return *this; } template auto &operator+=(const Q &b){ rep(i,0,len-1) (*this)[i]+=b; return *this; } template auto &operator-=(const Q &b){ rep(i,0,len-1) (*this)[i]-=b; return *this; } template auto &operator*=(const Q &b){ rep(i,0,len-1) (*this)[i]*=b; return *this; } template auto &operator/=(const Q &b){ rep(i,0,len-1) (*this)[i]/=b; return *this; } template auto &operator%=(const Q &b){ rep(i,0,len-1) (*this)[i]%=b; return *this; } auto &operator =(const string &b){ return CpSeq(b); } template auto &operator =(const char (&b)[Q]){ return *this=string(b); } template auto &operator =(const vector &b){ return CpSeq(b); } template auto &operator+=(const vector &b){ return PlSeq(b); } template auto &operator-=(const vector &b){ return MnSeq(b); } template auto &operator*=(const vector &b){ return PrSeq(b); } template auto &operator/=(const vector &b){ return DvSeq(b); } template auto &operator%=(const vector &b){ return RmSeq(b); } template auto &operator =(const view1d &b){ return CpSeq(b); } template auto &operator+=(const view1d &b){ return PlSeq(b); } template auto &operator-=(const view1d &b){ return MnSeq(b); } template auto &operator*=(const view1d &b){ return PrSeq(b); } template auto &operator/=(const view1d &b){ return DvSeq(b); } template auto &operator%=(const view1d &b){ return RmSeq(b); } template auto &CpSeq(const Q &b){ rep(i,0,min(sz(b),len)-1) (*this)[i] =b[i]; return *this; } template auto &PlSeq(const Q &b){ rep(i,0,min(sz(b),len)-1) (*this)[i]+=b[i]; return *this; } template auto &MnSeq(const Q &b){ rep(i,0,min(sz(b),len)-1) (*this)[i]-=b[i]; return *this; } template auto &PrSeq(const Q &b){ rep(i,0,min(sz(b),len)-1) (*this)[i]*=b[i]; return *this; } template auto &DvSeq(const Q &b){ rep(i,0,min(sz(b),len)-1) (*this)[i]/=b[i]; return *this; } template auto &RmSeq(const Q &b){ rep(i,0,min(sz(b),len)-1) (*this)[i]%=b[i]; return *this; } //template static bool eq(const Q &a,const P &b){ // return equals(ALL(a),ALL(b)); //} template static bool eq(const Q &a,const P &b){ if ((ll)a.size()!=(ll)b.size()) return false; rep(i,0,(ll)a.size()-1){ if (a[i]!=b[i]) return false; } return true; } template static bool lt(const Q &a,const P &b){ ll n=min((ll)a.size(),(ll)b.size()); rep(i,0,n-1){ if (a[i]b[i])return false; } return (ll)a.size()<(ll)b.size(); } template bool operator==( const view1d &b){ return eq(*this,b); } template bool operator==( const Q &b){ return eq(*this,b); } template friend bool operator==(const Q &a,const view1d &b){ return eq( a,b); } template bool operator==( const char (&b)[Q]){ return eq(*this,string(b)); } template bool operator!=( const view1d &b){ return !(*this==b); } template bool operator!=( const Q &b){ return !(*this==b); } template friend bool operator!=(const Q &a,const view1d &b){ return !( a==b); } template bool operator< ( const view1d &b){ return lt(*this,b); } template bool operator< ( const Q &b){ return lt(*this,b); } template friend bool operator< (const Q &a,const view1d &b){ return lt( a,b); } template bool operator> ( const view1d &b){ return lt(b,*this); } template bool operator> ( const Q &b){ return lt(b,*this); } template friend bool operator> (const Q &a,const view1d &b){ return lt(b,a ); } template bool operator<=( const view1d &b){ return !(*this>b); } template bool operator<=( const Q &b){ return !(*this>b); } template friend bool operator<=(const Q &a,const view1d &b){ return !( a>b); } template bool operator>=( const view1d &b){ return !(*this bool operator>=( const Q &b){ return !(*this friend bool operator>=(const Q &a,const view1d &b){ return !( a(OrgAt(s+d*i)); } const R &operator[](ll i) const { return OrgAt(s+d*i); } R &at(ll i){ if(i<0)i+=len; return (*this)[i]; } //vector tov(){ vector vvv(len); rep(i,0,len-1) vvv[i]=(*this)[i]; return vvv; } operator vector(){ vector v(len); rep(i,0,len-1) v[i]=(*this)[i]; return v; } bool contains(R a){ rep(i,0,len-1) if((*this)[i]==a)return true; return false; } auto begin(){ return view1dIter(this,0); } auto end(){ return view1dIter(this,len); } /*---- view設定 ----*/ view1d &dmy(R dmy_) { dummy=dmy_; return *this; }//ダミー値セット template view1d &st(Q... s_){//始点set 負は末尾からの位置 this->s=RevIfNeg(SllD(s_...)); this->len=AutoLen(this->s,this->d,this->dsize); return *this; } template view1d &en(Q... e_){//終了条件再設定 負は末尾から this->len=Len(s, RevIfNeg(SllD(e_...)), d); return *this; } template view1d &dir(Q... d_){//方向set、長さはdata端まで this->d=SllD(d_...); this->len=AutoLen(this->s,this->d,this->dsize); return *this; } template view1d &mv(Q... s_){ this->s+=SllD(s_...); return *this; }//平行移動 view1d &size(ll len_) { len=len_; return *this; }//長さset template view1d &size(Q &v){ len=(ll)v.size(); return *this; }//長さset view1d &rev() { s+=d*(len-1); d*=-1; return *this; }//反転 /*---- utility ----*/ template inline static ll sz(Q &v){ return (ll)v.size(); } Sll RevIfNeg(Sll pos){//!< 負なら末尾からの位置に変換 rep(i,0,S-1) if (pos[i]<0) pos[i]+=dsize[i]; return pos; } static ll AutoLen(Sll s_,Sll d_,Sll dsz){//位置s_から方向d_ではみ出すまでの長さ Sll e=dsz-1; rep(i,0,S-1) e[i] *= (d_[i]>=0);//方向が負の軸を0にする return Len(s_,e,d_); } /*---- 先頭位置s、方向d、終了条件eから長さlen計算 ----*/ //template static ll Len(array s,array e,array d){ // ll ret=INF; // rep(i,0,Q-1) chmin(ret, Len1(s[i],e[i],d[i])); // return ret; //} static ll Len(Sll s,Sll e,Sll d){ ll ret=INF; rep(i,0,S-1) chmin(ret, Len1(s[i],e[i],d[i])); return ret; } static ll Len1(ll s,ll e,ll d){ if (d==0) return INF; if (d<0){ s=-s; e=-e; d=-d; } if (s>e) return 0; return (e-s)/d+1; } /*---- 可変長引数をSllに変換 ----*/ template static Sll SllD(Q... args){ return SllRec(0,args...); } template static Sll SllRec(ll i,ll first,Q... rest) { Sll sll = (i==S-1) ? Sll() : SllRec(i+1,rest...); sll[i]=first; return sll; } static Sll SllRec(ll i){ return Sll(); } /*---- dataの位置posの値取得 ----*/ const R &OrgAt(Sll pos) const { rep(i,0,S-1){ if (pos[i]<0 || dsize[i]<=pos[i]) return dummy; } return OrgAt_(data,pos); } template using V = vector; template using VV = V>; template using VVV = V>>; using Vs = V; using VVs = VV; using ll1 = array; using ll2 = array; using ll3 = array; auto &OrgAt_(V &dat,ll1 pos)const{ auto [i] =pos; return dat[i]; } auto &OrgAt_(string &dat,ll1 pos)const{ auto [i] =pos; return dat[i]; } auto &OrgAt_(VV &dat,ll2 pos)const{ auto [i,j] =pos; return dat[i][j]; } auto &OrgAt_(Vs &dat,ll2 pos)const{ auto [i,j] =pos; return dat[i][j]; } auto &OrgAt_(VVV &dat,ll3 pos)const{ auto [i,j,k]=pos; return dat[i][j][k]; } auto &OrgAt_(VVs &dat,ll3 pos)const{ auto [i,j,k]=pos; return dat[i][j][k]; } /*---- dataの各軸size取得 ----*/ static void SetDsize(V &dat,ll1 &dsz){ dsz={sz(dat)}; } static void SetDsize(string &dat,ll1 &dsz){ dsz={sz(dat)}; } static void SetDsize(VV &dat,ll2 &dsz){ dsz={sz(dat),sz(dat[0])}; } static void SetDsize(Vs &dat,ll2 &dsz){ dsz={sz(dat),sz(dat[0])}; } static void SetDsize(VVV &dat,ll3 &dsz){ dsz={sz(dat),sz(dat[0]),sz(dat[0][0])}; } static void SetDsize(VVs &dat,ll3 &dsz){ dsz={sz(dat),sz(dat[0]),sz(dat[0][0])}; } typedef view1dIter iterator; using value_type = R; }; template using V = vector; template using VV = V>; template using VVV = V>>; template view1d( VVV ) -> view1d< VVV,3,T>; template view1d( VV ) -> view1d< VV,2,T>; template view1d( V ) -> view1d< V,1,T>; ; view1d(VV ) -> view1d,3,char>; ; view1d( V ) -> view1d< V,2,char>; ; view1d( string ) -> view1d< string ,1,char>; template view1d( VVV,S,S,S) -> view1d< VVV,3,T>; template view1d( VV,S,S,S) -> view1d< VV,2,T>; template view1d( V,S,S,S) -> view1d< V,1,T>; template view1d(VV,S,S,S) -> view1d,3,char>; template view1d( V,S,S,S) -> view1d< V,2,char>; template view1d( string ,S,S,S) -> view1d< string ,1,char>; template struct view2dIter{ VIEW2D *vw=nullptr; ll idx=INF; view2dIter(){} view2dIter(VIEW2D *vw_,ll idx_):vw(vw_),idx(idx_){} view2dIter(const view2dIter &it):vw(it.vw),idx(it.idx){} auto &operator*() { return (*vw)[x(idx)][y(idx)]; } auto &operator*() const{ return (*vw)[x(idx)][y(idx)]; } auto &operator[](ll i) { return (*vw)[x(idx+i)][y(idx+i)]; } auto &operator[](ll i)const{ return (*vw)[x(idx+i)][y(idx+i)]; } auto &operator++(){ idx++; return *this; } auto &operator--(){ idx--; return *this; } auto operator++(int){ auto it=*this; idx++; return it; } auto operator--(int){ auto it=*this; idx--; return it; } auto &operator+=(ll n){ idx+=n; return *this; } auto &operator-=(ll n){ idx-=n; return *this; } auto operator+(ll n){ auto it=*this; return it+=n; } auto operator-(ll n){ auto it=*this; return it-=n; } ll operator- (const view2dIter &it)const{ return idx- it.idx; } bool operator< (const view2dIter &it)const{ return idx< it.idx; } bool operator> (const view2dIter &it)const{ return idx> it.idx; } bool operator<=(const view2dIter &it)const{ return idx<=it.idx; } bool operator>=(const view2dIter &it)const{ return idx>=it.idx; } bool operator!=(const view2dIter &it)const{ return idx!=it.idx; } bool operator==(const view2dIter &it)const{ return idx==it.idx; } ll x(ll i)const{ return i/vw->leny(); } ll y(ll i)const{ return i%vw->leny(); } using iterator_category = random_access_iterator_tag; using value_type = typename VIEW2D::value_type; using difference_type = ll; using pointer = value_type*; using reference = value_type&; }; template struct view2d{ using Sll=array; T &data; //参照先データ Sll dsize; //参照先データ各軸のサイズ Sll s=Sll(); //view始点 Sll dx=Sll(); //view x軸方向 Sll dy=Sll(); //view y軸方向 ll xl; //view x軸長さ ll yl; //view y軸長さ R dummy=R(); //範囲外の値 /*---- コンストラクタ ----*/ view2d(T &v): data(v){ view1d::SetDsize(v,dsize); dx[S-2]=1; dy[S-1]=1; xl=dsize[S-2]; yl=dsize[S-1]; } /*---- 演算 ----*/ //template auto &operator =(const Q &b){ rep(i,0,xl-1) (*this)[i] =b; return *this; } //template auto &operator+=(const view2d &b){ // rep(i,0,xl-1) (*this)[i] // return PlSeq(b); } /*---- getter ----*/ ll size() const { return xl; } array shape() const { return {xl,yl}; } ll lenx() const { return xl; } ll leny() const { return yl; } ///pll shape() const { return {xl,yl}; } view1d operator[](ll i){ return view1d(data,s+dx*i,dy,yl,dummy); } const view1d operator[](ll i) const { return view1d(data,s+dx*i,dy,yl,dummy); } R &at(ll i,ll j){ if(i<0)i+=xl; if(j<0)j+=yl; return (*this)[i][j]; } //vector> tovv(){ // vector> vvv(xl); // ///rep(i,0,xl-1) vvv[i]=(*this)[i].tov(); return vvv; // rep(i,0,xl-1) vvv[i]=(*this)[i]; return vvv; //} operator vector>(){ vector> v(xl); rep(i,0,xl-1) v[i]=(*this)[i]; return v; } auto begin(){ return view2dIter(this,0); } auto end(){ return view2dIter(this,xl*yl); } #if defined(_DEBUG) void dump(){ ::dump(tovv()); } #endif /*---- view設定 ----*/ view2d &dmy(R dmy_) { dummy=dmy_; return *this; }//ダミー値セット template view2d &st(Q... s_){//始点set 負は末尾からの位置 this->s=RevIfNeg(view1d::SllD(s_...)); this->xl=view1d::AutoLen(this->s,this->dx,this->dsize); this->yl=view1d::AutoLen(this->s,this->dy,this->dsize); return *this; } template view2d &dirx(Q... d_){//x軸set、長さはdata端まで this->dx=view1d::SllD(d_...); this->xl=view1d::AutoLen(this->s,this->dx,this->dsize); return *this; } template view2d &diry(Q... d_){//y軸set、長さはdata端まで this->dy=view1d::SllD(d_...); this->yl=view1d::AutoLen(this->s,this->dy,this->dsize); return *this; } template view2d &mv(Q... s_){//平行移動 this->s += view1d::SllD(s_...); return *this; } view2d &lenx(ll xl_) { xl=xl_; return *this; } view2d &leny(ll yl_) { yl=yl_; return *this; } view2d &shape(ll xl_,ll yl_) { xl=xl_; yl=yl_; return *this; } template view2d &shape(Q &v) { xl=v.lenx(); yl=v.leny(); return *this; } view2d &rot90 () { s+=dx*(xl-1); swap(xl,yl); swap(dx,dy); dy*=-1; return *this; } view2d &rot270() { s+=dy*(yl-1); swap(xl,yl); swap(dx,dy); dx*=-1; return *this; } view2d &rot180() { s+=dx*(xl-1)+dy*(yl-1); dx*=-1; dy*=-1; return *this; } view2d &revx() { s+=dx*(xl-1); dx*=-1; return *this; } view2d &revy() { s+=dy*(yl-1); dy*=-1; return *this; } view2d &swapxy() { swap(xl,yl); swap(dx,dy); return *this; }//xy軸入替 /*---- utility ----*/ Sll RevIfNeg(Sll pos){//!< 負なら末尾からの位置に変換 rep(i,0,S-1) if (pos[i]<0) pos[i]+=dsize[i]; return pos; } using value_type = R; }; template view2d( VVV) -> view2d< VVV,3,T>; template view2d( VV) -> view2d< VV,2,T>; ; view2d(VV) -> view2d,3,char>; ; view2d( V) -> view2d< V,2,char>; /* zipは参照のpairを返す。それをさらに参照するのはまずいので、コピーする。 */ template struct zipIter{ ZIP *z=nullptr; ll idx=INF; zipIter(){} zipIter(ZIP *z_,ll idx_):z(z_),idx(idx_){} zipIter(const zipIter &it):z(it.z),idx(it.idx){} auto operator*() { return (*z)[idx]; }//参照ではなく値戻し auto operator*() const{ return (*z)[idx]; }//同上 auto operator[](ll i) { return (*z)[idx+i]; }//同上 auto operator[](ll i)const{ return (*z)[idx+i]; }//同上 auto &operator++(){ idx++; return *this; } auto &operator--(){ idx--; return *this; } auto operator++(int){ auto it=*this; idx++; return it; } auto operator--(int){ auto it=*this; idx--; return it; } auto &operator+=(ll n){ idx+=n; return *this; } auto &operator-=(ll n){ idx-=n; return *this; } auto operator+(ll n){ auto it=*this; return it+=n; } auto operator-(ll n){ auto it=*this; return it-=n; } ll operator- (const zipIter &it)const{ return idx- it.idx; } bool operator< (const zipIter &it)const{ return idx< it.idx; } bool operator> (const zipIter &it)const{ return idx> it.idx; } bool operator<=(const zipIter &it)const{ return idx<=it.idx; } bool operator>=(const zipIter &it)const{ return idx>=it.idx; } bool operator!=(const zipIter &it)const{ return idx!=it.idx; } bool operator==(const zipIter &it)const{ return idx==it.idx; } using iterator_category = random_access_iterator_tag; using value_type = typename ZIP::value_type; using difference_type = ll; using pointer = value_type*; using reference = value_type&; }; template struct zip2{ T &t; S &s; zip2(T &t_,S &s_):t(t_),s(s_){} ll size() const { return t.end()-t.begin(); } auto operator[](ll i) { return make_pair(ref(t.begin()[i]),ref(s.begin()[i])); } auto operator[](ll i)const{ return make_pair(ref(t.begin()[i]),ref(s.begin()[i])); } auto begin(){ return zipIter(this,0); } auto end(){ return zipIter(this,size()); } using value_type = pair; }; template struct zip3{ T &t; S &s; R &r; zip3(T &t_,S &s_,R &r_):t(t_),s(s_),r(r_){} ll size() const { return t.end()-t.begin(); } auto operator[](ll i) { return make_tuple(ref(t.begin()[i]),ref(s.begin()[i]),ref(r.begin()[i])); } auto operator[](ll i)const{ return make_tuple(ref(t.begin()[i]),ref(s.begin()[i]),ref(r.begin()[i])); } auto begin(){ return zipIter(this,0); } auto end(){ return zipIter(this,size()); } using value_type = tuple; }; #endif//テンプレートend #if defined(_MSC_VER) && __has_include() #include using namespace atcoder; #endif struct{ system_clock::time_point st = system_clock::now(); ll operator()()const{return duration_cast(system_clock::now()-st).count()/1000;} } timeget; ////////////////////////////////////////// struct fraction{ ll bunsi=0,bunbo=1; /*---- utilty ----*/ fraction &norm(){//正規化(約分,分母正) ll g=gcd(bunsi,bunbo); bunsi/=g,bunbo/=g; if (bunbo<0) bunsi*=-1,bunbo*=-1; return *this; } fraction &invsg(){//正負反転 bunsi*=-1; return *this; } /*---- I/F ----*/ fraction(){} fraction(ll bunsi_,ll bunbo_):bunsi(bunsi_),bunbo(bunbo_){ norm(); } fraction(ll b):bunsi(b){}//整数用コンストラクタ fraction(ll bunsi_,ll bunbo_,bool b)//正規化無のコンストラクタ :bunsi(bunsi_),bunbo(bunbo_){} fraction &operator+=(fraction b){ ll bo=bunbo*b.bunbo; ll si=bunsi*b.bunbo + b.bunsi*bunbo; bunsi=si,bunbo=bo; return norm(); } fraction &operator-=(fraction b){ return (*this)+=b.invsg(); } fraction &operator*=(fraction b){ bunsi*=b.bunsi; bunbo*=b.bunbo; return norm(); } fraction &operator/=(fraction b){ return (*this)*=b.inv(); } fraction &operator+=(ll b){ bunsi+=bunbo*b; return *this; } fraction &operator-=(ll b){ return (*this)+=-b; } fraction &operator*=(ll b){ bunsi*=b; return norm(); } fraction &operator/=(ll b){ bunbo*=b; return norm(); } fraction operator-()const{ return {-bunsi,bunbo,0}; } fraction operator+(const fraction &b)const{ return fraction(*this)+=b; } fraction operator-(const fraction &b)const{ return fraction(*this)-=b; } fraction operator*(const fraction &b)const{ return fraction(*this)*=b; } fraction operator/(const fraction &b)const{ return fraction(*this)/=b; } fraction operator+(ll b)const{ return fraction(*this)+=b; } fraction operator-(ll b)const{ return fraction(*this)-=b; } fraction operator*(ll b)const{ return fraction(*this)*=b; } fraction operator/(ll b)const{ return fraction(*this)/=b; } friend fraction operator+(ll a,const fraction &b){ return b+a; } friend fraction operator-(ll a,const fraction &b){ return -b+a; } friend fraction operator*(ll a,const fraction &b){ return b*a; } friend fraction operator/(ll a,const fraction &b){ return b.inv()*a; } bool operator==(const fraction &b)const{ return bunsi==b.bunsi && bunbo==b.bunbo; } bool operator!=(const fraction &b)const{ return !(*this==b); } bool operator<(const fraction &b)const{ return (b-*this).bunsi>0; } bool operator>(const fraction &b)const{ return b<*this; } bool operator>=(const fraction &b)const{ return !(*this=*this; } friend ostream &operator << (ostream &os,const fraction &a){ return os<y,x<=y,x>=y cout << x; //"a/b"と表示 */ template struct mll_{ using Int = long long; using ll = long long; ll val_=0; /*---- utility ----*/ mll_ &norm(){ return normR().normS(); }//正規化 mll_ &normR(){ val_%=MOD; return *this; }//剰余正規化のみ mll_ &normS(){ if (val_<0) val_+=MOD; return *this; }//正負正規化のみ mll_ &normP(){ if (val_>=MOD) val_-=MOD; return *this; }//加算時正規化 mll_ &invsg(){ val_=-val_; return normS(); }//正負反転 ll modinv(int a){//a^-1 mod MOD int ypre=0,y=1,apre=MOD; while (a>1){ int t=apre/a; apre-=a*t,swap(a,apre); ypre-=y*t,swap(y,ypre); } return y<0 ? y+MOD: y; } /*---- I/F ----*/ mll_(){} mll_(ll v): val_(v){ norm(); } mll_(ll v,bool b): val_(v){} //正規化無のコンストラクタ Int val()const{ return (Int)val_; } bool isnone() const { return val_==-1; } //true:値なし mll_ &none() { val_=-1; return *this; } //値なしにする mll_ &inv(){ val_=modinv((int)val_); return *this; } mll_ &operator+=(mll_ b){ val_+=b.val_; return normP(); } mll_ &operator-=(mll_ b){ val_-=b.val_; return normS(); } mll_ &operator*=(mll_ b){ val_*=b.val_; return normR(); } mll_ &operator/=(mll_ b){ return *this*=b.inv(); } mll_ &operator+=(ll b){ return *this+=mll_(b); } mll_ &operator-=(ll b){ return *this-=mll_(b); } mll_ &operator*=(ll b){ return *this*=mll_(b); } mll_ &operator/=(ll b){ return *this/=mll_(b); } mll_ operator-()const{ return mll_(*this).invsg(); } mll_ operator+(mll_ b)const{ return mll_(*this)+=b; } mll_ operator-(mll_ b)const{ return mll_(*this)-=b; } mll_ operator*(mll_ b)const{ return mll_(*this)*=b; } mll_ operator/(mll_ b)const{ return mll_(*this)/=b; } mll_ operator+(ll b)const{ return mll_(*this)+=b; } mll_ operator-(ll b)const{ return mll_(*this)-=b; } mll_ operator*(ll b)const{ return mll_(*this)*=b; } mll_ operator/(ll b)const{ return mll_(*this)/=b; } friend mll_ operator+(ll a,mll_ b){ return b+a; } friend mll_ operator-(ll a,mll_ b){ return -b+a; } friend mll_ operator*(ll a,mll_ b){ return b*a; } friend mll_ operator/(ll a,mll_ b){ return mll_(a)/b; } bool operator==(mll_ b)const{ return val_==b.val_; } bool operator!=(mll_ b)const{ return val_!=b.val_; } bool operator==(ll b)const{ return *this==mll_(b); } bool operator!=(ll b)const{ return *this!=mll_(b); } friend bool operator==(ll a,mll_ b){ return mll_(a)==b; } friend bool operator!=(ll a,mll_ b){ return mll_(a)!=b; } friend ostream &operator<<(ostream &os,mll_ a){ return os << a.val_; } friend istream &operator>>(istream &is,mll_ &a){ return is >> a.val_; } mll_ pow(ll k)const{ mll_ ret(1,false),a(*this); for (; k>0; k>>=1,a*=a) if (k&1)ret*=a; return ret; } static constexpr int mod() { return MOD; } //enum{ modll=MOD }; }; #if 0 #define MODLL (1000000007LL) #else #define MODLL (998244353LL) #endif using mll = mll_; //using mll = fraction; using vmll = std::vector; using vvmll = std::vector; using vvvmll = std::vector; using vvvvmll = std::vector; template struct Vector: vector{ using Int = long long; using vT=vector; using cvT=const vector; using cT=const T; using vT::vT; //親クラスのコンストラクタの隠蔽を回避 using vT::begin,vT::end,vT::insert,vT::erase; auto it(Int i){ return begin()+i; } auto it(Int i)const{ return begin()+i; } Vector(cvT& b):vT(b){} Vector(vT&& b):vT(move(b)){} Vector(Int n,T s,T d){ iota(n,s,d); } template Vector &operator+=(S x){ for (T &e: *this) e+=x; return *this; } template Vector &operator-=(S x){ for (T &e: *this) e-=x; return *this; } template Vector &operator*=(S x){ for (T &e: *this) e*=x; return *this; } template Vector &operator/=(S x){ for (T &e: *this) e/=x; return *this; } template Vector &operator%=(S x){ for (T &e: *this) e%=x; return *this; } template Vector operator+(S x)const{ return Vector(*this)+=x; } template Vector operator-(S x)const{ return Vector(*this)-=x; } template Vector operator*(S x)const{ return Vector(*this)*=x; } template Vector operator/(S x)const{ return Vector(*this)/=x; } template Vector operator%(S x)const{ return Vector(*this)%=x; } Vector operator-()const{ return Vector(*this)*=-1; } template friend Vector operator-(S x,const Vector &a){ return -a+=x; } Int size()const{ return (Int)vT::size(); } Vector &push_back(cT& x){ vT::push_back(x); return *this; } Vector &pop_back(){ vT::pop_back(); return *this; } Vector &push_front(cT& x){ insert(begin(),x); return *this; } Vector &pop_front(){ erase(0); return *this; } Vector &insert(Int i,cT& x){ insert(it(i),x); return *this; } Vector &insert(Int i,cvT& b){ insert(it(i),b.begin(),b.end()); return *this; } Vector &erase(Int i){ erase(it(i)); return *this; } Vector &erase(Int l,Int r){ erase(it(l),it(r+1)); return *this; } Vector &concat(cvT &b){ insert(end(),b.begin(),b.end()); return *this; } Vector &concat(cvT &b,Int n){ for (int i=0;i Vector &sort(Pr pr){ std::sort(begin(),end(),pr); return *this; } Vector &uniq(){ erase(unique(begin(),end()),end()); return *this; } Vector &sortq(){ return sort().uniq(); } Vector &fill(Int l,Int r,cT& x){ std::fill(it(l),it(r+1),x); return *this; } template Vector &iota(Int n,T s=0,S d=1){ vT::resize(n); if (n==0) return *this; (*this)[0]=s; for (int i=1;i Int countif(Pr pr)const{ return Int(count_if(begin(),end(),pr)); } template Int countif(Int l,Int r,Pr pr)const{ return Int(count_if(it(l),it(r+1),pr)); } Int find(cT& x)const{ return Int(std::find(begin(),end(),x)-begin()); } Int find(Int l,Int r,cT& x)const{ return Int(std::find(it(l),it(r+1),x)-begin()); } template Int findif(Pr pr)const{ return Int(find_if(begin(),end(),pr)-begin()); } template Int findif(Int l,Int r,Pr pr)const{ return Int(find_if(it(l),it(r+1),pr)-begin()); } Vector findall(cT& x)const{ return findall(0,size()-1,x); } Vector findall(Int l,Int r,cT& x)const{ return findallif(l,r,[&](cT& y){return y==x;}); } template Vector findallif(Pr pr)const{ return findallif(0,size()-1,pr); } template Vector findallif(Int l,Int r,Pr pr)const{ Vector ret; for (Int i=l;i<=r;++i) if (pr((*this)[i])) ret.push_back(i); return ret; } Int flooridx(cT& x)const{ return Int(upper_bound(begin(),end(),x)-begin()-1); } Int ceilidx(cT& x)const{ return Int(lower_bound(begin(),end(),x)-begin()); } Int leftnmof(cT& x)const{ return flooridx(x)+1; } Int rightnmof(cT& x)const{ return size()-ceilidx(x); } bool contains(cT& x)const{ Int i=flooridx(x); return i>=0 && (*this)[i]==x; } template Int flooridx(cT& x,Pr pr)const{ return Int(upper_bound(begin(),end(),x,pr)-begin()-1); } template Int ceilidx(cT& x,Pr pr)const{ return Int(lower_bound(begin(),end(),x,pr)-begin()); } template Int leftnmof(cT& x,Pr pr)const{ return flooridx(x,pr)+1; } template Int rightnmof(cT& x,Pr pr)const{ return size()-ceilidx(x,pr); } template bool contains(cT& x,Pr pr)const{ Int i=flooridx(x,pr); return i>=0 && (*this)[i]==x; } }; /* vll a={9,8,7},b={1,2,3}; vpll p={{5,3},{7,8},{0,2},}; - -------- 操作系 -------- a+=x a-=x a*=x a/=x a%=x a+x a-x a*x a/x a%x -a x-a //∀i a[i]にxを演算 a.push_front(x); a.push_back(x); a.pop_front(); a.pop_back(); a.insert(i,x); //a[i]にx挿入 a.insert(i,b); //a[i]にvll b挿入 a.erase(i); //a[i]削除 a.erase(l,r); //区間[l,r]削除 a.concat(b); //aにbを結合 a.concat(b,n); //aにbをn回結合 a.repeat(n); //aをn回繰り返す a.reverse(); //反転 a.sort(); //ソート a.rsort(); //逆順ソート p.sort([&](pll x,pll y){return x.second0; }; //検索条件 ll m=a.count(x); //xの個数 ll m=a.count(l,r,x); //xの個数in[l,r] ll m=a.countif(pr); //条件満たす個数 ll m=a.countif(l,r,pr); //条件満たす個数in[l,r] ll i=a.find(x); //xの最左位置i ない時N(配列長) ll i=a.find(l,r,x); //xの最左位置i in[l,r] ない時r+1 ll i=a.findif(pr); //条件満たす最左位置i ない時N(配列長) ll i=a.findif(l,r,pr); //条件満たす最左位置i in[l,r] ない時r+1 vll is=a.findall(x); //xの位置i列挙 vll is=a.findall(l,r,x); //xの位置i列挙in[l,r] vll is=a.findallif(pr); //条件満たす位置i列挙 vll is=a.findallif(l,r,pr); //条件満たす位置i列挙in[l,r] - -------- 昇順sort済み配列用 -------- ll i=a.flooridx(x); //x以下の最近傍位置i ない時-1 ll i=a.ceilidx(x); //x以上の最近傍位置i ない時N(配列長) ll m=a.leftnmof(x); //x以下の個数 ll m=a.rightnmof(x); //x以上の個数 bool b=a.contains(x); //xを含む - -------- 比較関数prでsort済みの配列用 -------- auto pr=[&](auto &x,auto &y){ return x>y; }; //降順ソート時 ll i=a.flooridx(x,pr); //x以左の最近傍位置i ない時-1 ll i=a.ceilidx(x,pr); //x以右の最近傍位置i ない時N(配列長) ll m=a.leftnmof(x,pr); //x以左の個数 ll m=a.rightnmof(x,pr); //x以右の個数 bool b=a.contains(x,pr); //xを含む a.concat(b,n).pop_back().rsort().uniq(); //連続適用できる */ namespace SolvingSpace{ template using vector = Vector; using vll=vector< ll>; using vmll=vector< mll>; using vdd=vector< dd>; using vvll=vector< vll>; using vvmll=vector< vmll>; using vvdd=vector< vdd>; using vvvll=vector< vvll>; using vvvmll=vector< vvmll>; using vvvdd=vector; using vvvvll=vector; using vvvvmll=vector; using vpll=vector< pll>; using vtll=vector< tll>; using vqll=vector< qll>; using vvpll=vector< vpll>; using vvtll=vector< vtll>; using vvqll=vector; template struct HeavyLightDecomposition{ using Int = long long; vector> *to=nullptr; //グラフ vector head; //head[v] 頂点vの属する重パスの根側端頂点 vector id; //id[v] 頂点vの行きがけ順番号 0-origin vector rev; //rev[ID] 行きがけ順番号IDの頂点v vector lst; //lst[v] 頂点vのサブ木内頂点の行きがけ順番号最大値 vector par; //par[v] 頂点vの親頂点 par[根]は-1 vector siz; //siz[v] 頂点vのサブ木内頂点数(v含む) vector dpt; //dpt[v] 頂点vの深さ(根が0) HeavyLightDecomposition(){} HeavyLightDecomposition(vector> &to_,Int root=0): to(&to_),head(size(),root),id(size()),rev(size()),lst(size()), par(size()),siz(size(),1),dpt(size()) { assert(!to_.empty()); dfsSz(root); dfsHld(root); } Int size()const{ return to->size(); } //頂点数 Int vof(Int id_)const{ return rev[id_]; } Int idof(Int v)const{ return id[v]; } Int parentof(Int v)const{ return par[v]; } Int subszof(Int v)const{ return siz[v]; } Int depthof(Int v)const{ return dpt[v]; } bool isleaf(Int v)const{ return id[v]==size()-1 || head[v]!=head[rev[id[v]+1]]; } vector> preorder()const{ //行きがけ順の(頂点,親頂点)列 vector> ret; for (int i=0;iid[v]) swap(u,v); v=par[head[v]]; } return id[u]id[v])swap(u,v); while (head[u]!=head[v] && id[u]> heavypaths()const{ //[stID0,enID0],[stID1,enID1],… vector> ret; Int enID=to->size()-1; while (enID!=-1){ Int stID=id[head[rev[enID]]]; ret.emplace_back(stID,enID); enID=stID-1; } reverse(ret.begin(),ret.end()); return ret; } vector> uv2idranges(Int u,Int v,bool isedge=false)const { //パスu→vを重パス分解 Int a=lca(u,v); //aを含む重パス上まで移動 auto func=[&](Int &t){ vector> t2a; while (head[t]!=head[a]){ t2a.emplace_back(id[t],id[head[t]]); t=par[head[t]]; } return t2a; }; vector> u2a=func(u),v2a=func(v); //aを含む重パス上の追加 if (!isedge){//頂点 if (u!=a) u2a.emplace_back(id[u],id[a]); else if (v!=a) v2a.emplace_back(id[v],id[a]); else u2a.emplace_back(id[a],id[a]); } else{//辺 if (u!=a) u2a.emplace_back(id[u],id[a]-1); else if (v!=a) v2a.emplace_back(id[v],id[a]-1); } //v2aを反転して結合 while (!v2a.empty()){ auto [st,en]=v2a.back(); v2a.pop_back(); u2a.emplace_back(en,st); } return u2a; } vector path(Int u,Int v)const{ //パスu→v 頂点列 Int a=lca(u,v); auto func=[&](Int t){ vector t2a; for (;t!=a;t=par[t]) t2a.push_back(t); return t2a; }; vector u2a=func(u),v2a=func(v); u2a.push_back(a); while (!v2a.empty()){ u2a.push_back(v2a.back()); v2a.pop_back(); } return u2a; } pair subtree2idrange(Int v)const{ //サブ木ID範囲 return {id[v],lst[v]}; } private: void dfsSz(Int v,Int p=-1){ par[v]=p; for (auto&& u: (*to)[v]){ if (u==p)continue; dpt[u]=dpt[v]+1; dfsSz(u,v); siz[v]+=siz[u]; } } Int idcnt=0; void dfsHld(Int v,Int p=-1){ vector &tov=(*to)[v]; id[v]=idcnt++; rev[id[v]]=v; //重パスの子を先頭へ Int mxsz=-1; for (auto&& u: tov){ if (u==p)continue; if (mxszenIDもあり vpll hs=hld.uv2idranges(u,v); vpll hs=hld.uv2idranges(u,v,true); //辺の値を乗せる時(LCA含まない) - パスu→vの頂点列 vll path=hld.path(u,v); - ID列上のサブ木範囲 auto[l,r]=hld.subtree2idrange(v); */ #if defined(_DEBUG) ll dumpW = 5; template void dump(vector &v){ rep(i,0,sz(v)-1) { cerr << ((i==0)?"[":" ") << setw(dumpW) << v[i]; } cerr << "]"; } void dump(vll &v){ rep(i,0,sz(v)-1) { cerr << ((i==0)?"[":" ") << setw(dumpW); if (v[i] == INF) cerr << "INF"; else if (v[i] == -INF) cerr << "-INF"; else cerr << v[i]; } cerr << "]"; } template void dump(vector> &v){ ll N=(ll)v.size(); string d[2][2]={{" ", "["}, {"\n", "]\n"}}; rep(i,0,N-1) { cerr << d[0][i==0]; dump(v[i]); cerr << d[1][i==N-1]; } } template void dump(vector>> &v){ ll N=(ll)v.size(); string d[2]={" <", "[ <"}; rep(i,0,N-1) { cerr << d[i==0] << i << ">\n"; dump(v[i]); } cerr << "]\n"; } template void dump(vector>>> &v){ ll N=(ll)v.size(); string d[2]={" ---<", "[ ---<"}; rep(i,0,N-1) { cerr << d[i==0] << i << ">---\n"; dump(v[i]); } cerr << "]\n"; } template void dump(vector>>>> &v){ ll N=(ll)v.size(); string d[2]={" ======<", "[ ======<"}; rep(i,0,N-1) { cerr << d[i==0] << i << ">======\n"; dump(v[i]); } cerr << "]\n"; } template void dump(map &m){ for (auto e: m) cout << e.first << " : " << e.second << '\n'; } #else template void dump(T &v){} #endif template struct dptable2d{ using Int = long long; ll sx,ex,sy,ey; T dmy,dmyval; //table外値 T none; //無効値(未計算の印) vector> tbl; vector> cm; //累積和配列 /*---- I/F ----*/ dptable2d(Int sx_,Int ex_,Int sy_,Int ey_,T inival,T dmyval_,T none_): sx(sx_),ex(ex_),sy(sy_),ey(ey_),dmy(dmyval_),dmyval(dmyval_),none(none_), tbl(ex-sx+1,vector(ey-sy+1,inival)),cm(ex-sx+2,vector(ey-sy+2,none)) { for (ll i=0;i dp(sx,ex,sy,ey, 0, 0, mll().none()); dptable2d dp(sx,ex,sy,ey, 0, 0, -INF*7); - -------- アクセス -------- dp(i,s) //dptable2d[i-sx][s-sy] mll sm=dp.sum(lx,rx,ly,ry); //矩形区間x=lx~rx,y=ly~ryのdptable2dの和 */ void cin2solve() { ll n; cin >> n; vector> uv; rep(i,0,n-1-1){ ll u; ll v; cin>>u>>v; uv.emplace_back(u-1,v-1); } auto makeGraph=[](ll n,vector> &ab,bool isDirected){ vvll to(n); for (auto [a,b]:ab){ to[a].push_back(b); if (!isDirected) to[b].push_back(a); } return to; }; vvll to=makeGraph(n,uv,false); HeavyLightDecomposition hld(to); ll D=3; dptable2d dp(0,n-1,0,D, 0, 0, -INF*7); ll ans=0; vpll vp=hld.preorder(); vp.reverse(); for (auto&&[v,p]:vp){ vll ep(D+1); for (ll u:to[v]){ if (u==p)continue; // ans+=(ep[0]+ep[1])*dp(u,0); ans+=(ep[0])*dp(u,1); //加算 rep(j,0,D) ep[j]+=dp(u,j); } ans+=ep[0]+ep[1]+ep[2]; dp(v,0)=1; rep(j,1,D) dp(v,j)=ep[j-1]; } cout << ans << '\n'; //dump(dp.tbl); return; } };//SolvingSpace ////////////////////////////////////////// int main(){ #if 1 //SolvingSpace::labo(); SolvingSpace::cin2solve(); //SolvingSpace::generand(); #else ll t; cin >> t; rep(i,0,t-1){ SolvingSpace::cin2solve(); //SolvingSpace::generand(); } #endif cerr << timeget() <<"ms"<< '\n'; return 0; }