#387023 (C++14) No.900 aδδitivee

提出ソース

結果

問題	No.900 aδδitivee
ユーザー	👑 tute7627
提出日時	2019-10-05 01:15:41
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	WA
実行時間	-
コード長	6,869 bytes
コンパイル時間	2,320 ms
コンパイル使用メモリ	195,392 KB
実行使用メモリ	57,048 KB
最終ジャッジ日時	2024-10-04 03:26:48
合計ジャッジ時間	9,302 ms
ジャッジサーバーID （参考情報）	judge1 / judge3

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 5,248 KB
testcase_01	WA	-
testcase_02	WA	-
testcase_03	WA	-
testcase_04	WA	-
testcase_05	WA	-
testcase_06	WA	-
testcase_07	WA	-
testcase_08	WA	-
testcase_09	WA	-
testcase_10	WA	-
testcase_11	WA	-
testcase_12	WA	-
testcase_13	WA	-
testcase_14	WA	-
testcase_15	WA	-
testcase_16	WA	-
testcase_17	WA	-
testcase_18	WA	-
testcase_19	WA	-
testcase_20	WA	-
testcase_21	WA	-
testcase_22	WA	-
testcase_23	WA	-
testcase_24	WA	-
testcase_25	WA	-
testcase_26	WA	-
testcase_27	WA	-
testcase_28	WA	-

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

ソースコード

raw source code

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

#define endl '\n'
#define ALL(a)  (a).begin(),(a).end()
#define ALLR(a)  (a).rbegin(),(a).rend()
#define spa << " " <<
#define lfs <<fixed<<setprecision(10)<<
#define test cout<<"test"<<endl;
#define fi first
#define se second
#define MP make_pair
#define PB push_back
#define EB emplace_back
#define rep(i,n,m) for(ll i = n; i < (ll)(m); i++)
#define rrep(i,n,m) for(ll i = n - 1; i >= (ll)(m); i--)
using ll = long long;
using ld = long double;
const ll MOD = 1e9+7;
//const ll MOD = 998244353;
const ll INF = 1e18;
using P = pair<ll, ll>;
template<typename T>
void chmin(T &a,T b){if(a>b)a=b;}
template<typename T>
void chmax(T &a,T b){if(a<b)a=b;}
void pmod(ll &a,ll b){a=(a+b)%MOD;}
void pmod(ll &a,ll b,ll c){a=(b+c)%MOD;}
void qmod(ll &a,ll b){a=(a*b)%MOD;}
void qmod(ll &a,ll b,ll c){a=(b*c)%MOD;}
ll median(ll a,ll b, ll c){return a+b+c-max({a,b,c})-min({a,b,c});}
void ans1(bool x){if(x) cout<<"Yes"<<endl;else cout<<"No"<<endl;}
void ans2(bool x){if(x) cout<<"YES"<<endl;else cout<<"NO"<<endl;}
void ans3(bool x){if(x) cout<<"Yay!"<<endl;else cout<<":("<<endl;}
template<typename T1,typename T2>
void ans(bool x,T1 y,T2 z){if(x)cout<<y<<endl;else cout<<z<<endl;}  
template<typename T>
void debug(vector<vector<T>>v,ll h,ll w){for(ll i=0;i<h;i++)
{cout<<v[i][0];for(ll j=1;j<w;j++)cout spa v[i][j];cout<<endl;}};
void debug(vector<string>v,ll h,ll w){for(ll i=0;i<h;i++)
{for(ll j=0;j<w;j++)cout<<v[i][j];cout<<endl;}};
template<typename T>
void debug(vector<T>v,ll n){if(n!=0)cout<<v[0];
for(ll i=1;i<n;i++)cout spa v[i];cout<<endl;};
template<typename T>
vector<vector<T>>vec(ll x, ll y, T w){
  vector<vector<T>>v(x,vector<T>(y,w));return v;}
ll gcd(ll x,ll y){ll r;while(y!=0&&(r=x%y)!=0){x=y;y=r;}return y==0?x:y;}
template<typename T>
vector<ll>dx={1,0,-1,0,1,1,-1,-1};
vector<ll>dy={0,1,0,-1,1,-1,1,-1};
template<typename T>
vector<T> make_v(size_t a,T b){return vector<T>(a,b);}
template<typename... Ts>
auto make_v(size_t a,Ts... ts){
  return vector<decltype(make_v(ts...))>(a,make_v(ts...));
}

struct edge {
  ll to,cost;
  edge(ll x,ll y):to(x),cost(y){};
  edge():to(0LL),cost(0LL){};
};
//辺のオイラーツアー
struct Eulertour{
  ll n,m;
  ll root=0;
  vector<ll>vertex;//頂点列
  vector<pair<edge,ll>>data;//辺情報
  vector<ll>depth;//0から
  vector<vector<edge>>G;
  vector<pair<ll,ll>>index;//<最左index,最右index>
  Eulertour(vector<vector<edge>>g):G(g),n(g.size()){
    depth.assign(n,-1);
    index.assign(n,make_pair(10*n,10*n));
    dfs(root,0);
    m=vertex.size();
  }
  void dfs(ll k,ll d){
    depth[k]=d;
    for(ll i=0;i<G[k].size();i++){
      if(depth[G[k][i].to]==-1){
        chmin(index[k].fi,(ll)vertex.size());
        index[k].se = (ll)vertex.size();
        vertex.push_back(k);
        data.push_back(make_pair(G[k][i],1));

        dfs(G[k][i].to,d+1);

        chmin(index[G[k][i].to].fi,(ll)vertex.size());
        index[G[k][i].to].se = (ll)vertex.size();
        vertex.push_back(G[k][i].to);
        data.push_back(make_pair(G[k][i],-1));
      }
    }
    if(k==root){
      chmin(index[root].fi,(ll)vertex.size());
      index[root].se = (ll)vertex.size();
      vertex.push_back(root);
      data.push_back(MP(edge(),0));
    }
  }
  ll deep(ll x){return depth[x];}
  vector<ll>output(){//辺のコスト列を返す
    vector<ll>ret(m);
    for(ll i=0;i<m;i++){
      ret[i]=data[i].fi.cost*data[i].se;
    } 
    return ret;
  }
  vector<P>output2(){//辺のコスト列と正の辺の個数のペア
    vector<P>ret(m);
    for(ll i=0;i<m;i++){
      ret[i].fi=data[i].fi.cost*data[i].se;
      ret[i].se=max(0LL,data[i].se);
    } 
    return ret;
  }
};  

template<typename T>
struct LazySegmentTree{
  using F = function<T(T,T)>;
  vector<T> data, lazy; 
  ll n,lastlen = 1;
  F func = [](T a, T b){return MP(a.fi+b.fi,a.se+b.se);};
  T iden = MP(0,0); //identity element
  function<T(T,T,ll)> f_lazy = [](T a, T b,ll range){return MP(a.fi+b.fi*(2*a.se-range),a.se);};
  T iden_l = MP(0,0);
  function<T(T,ll)> f_trans = [](T a, ll range){return a;};
  LazySegmentTree(vector<T> v){
    n = (ll)v.size();
    while(lastlen < n)lastlen *= 2;
    data.assign(lastlen*2-1,iden);
    lazy.assign(lastlen*2-1,iden_l);
    for(ll i=0;i<n;i++)data[i+lastlen-1] = v[i];
    for(ll i=lastlen-2;i>=0;i--){
      data[i] = func(data[2*i+1], data[2*i+2]);
    }
  }
  void eval(ll k, ll left, ll right){
    if(lazy[k] != iden_l){
      data[k] = f_lazy(data[k],
      f_trans(lazy[k],right-left),right-left);
      if(k <= lastlen - 2){
        lazy[2*k+1] = f_lazy(lazy[2*k+1],lazy[k],right-left);
        lazy[2*k+2] = f_lazy(lazy[2*k+2],lazy[k],right-left);
      }
      lazy[k] = iden_l;
    }
  }
  void update(ll a,ll b,T x,ll point=0LL,
  ll left=0LL,ll right=-1LL){
    if(right<0)right=lastlen;
    eval(point,left,right);
    if(b <= left || right <= a);
    else if(a <= left && right <= b ){
      lazy[point] = x;
      eval(point,left,right);
    }
    else{
      update(a,b,x,point*2+1, left, (left+right)/2);
      update(a,b,x,point*2+2, (left+right)/2, right);
      data[point] = func(data[2*point+1], data[2*point+2]);
    }      
  }
  T query(ll a,ll b,ll point=0LL,ll left=0LL,ll right=-1LL){
    if(right<0)right=lastlen;
    T ret = iden;
    eval(point,left,right);
    if(b <= left || right <= a);
    else if(a <= left && right <= b ){
      ret = func(ret,data[point]);
    }
    else{
      T p=query(a,b,point*2+1, left, (left+right)/2);
      T q=query(a,b,point*2+2, (left+right)/2, right);
      data[point] = func(data[2*point+1], data[2*point+2]);
      ret = func(ret,p);
      ret = func(ret,q);
    }      
    return ret;
  }
  void print(){
    ll num=0;
    for(ll i=lastlen;i>0;i/=2)num++;
    ll tmp=1;
    for(ll i=0;i<num;i++,tmp*=2){
      for(ll j=0;j<tmp;j++){
        if(data[tmp-1+j]==iden)cout<<"x"<<" ";
        else cout<<data[tmp-1+j]<<" ";
      };
      cout<<"|";
    }
    cout<<endl;
    tmp=1;
    for(ll i=0;i<num;i++,tmp*=2){
      for(ll j=0;j<tmp;j++){
        if(lazy[tmp-1+j]==iden_l)cout<<"x"<<" ";
        else cout<<lazy[tmp-1+j]<<" ";
      }
      cout<<"|";
    }
    cout<<endl;
  }
};

int main(){
  cin.tie(NULL);
  ios_base::sync_with_stdio(false);
  ll res=0,res1=INF,res2=-INF,buf=0;
  bool judge = true;
  ll n;cin>>n;
  vector<vector<struct edge>>g(n);
  for(ll i=0;i<n-1;i++){
    ll u,v,c;
    cin>>u>>v>>c;
    g[u].emplace_back(v,c);
    g[v].emplace_back(u,c);
  }
  Eulertour et(g);
  auto v=et.output2();
  LazySegmentTree<P>segtree(v);
  ll q;cin>>q;
  while(q--){
    ll sw;cin>>sw;
    if(sw==1){
      ll a,x;cin>>a>>x;
      ll l=et.index[a].fi,r=et.index[a].se;
      segtree.update(l,r,MP(x,0));
    }
    else {
      ll a;cin>>a;
      ll l=et.index[0].fi;
      ll r=et.index[a].fi;
      cout<<segtree.query(l,r).fi<<endl;
    }
  }
  return 0;
}

yukicoder

結果

テストケース

ソースコード