結果
| 問題 |
No.399 動的な領主
|
| ユーザー |
 smkyb
|
| 提出日時 | 2025-03-25 11:15:30 |
| 言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
| 結果 |
AC
|
| 実行時間 | 579 ms / 2,000 ms |
| コード長 | 11,937 bytes |
| コンパイル時間 | 4,122 ms |
| コンパイル使用メモリ | 289,344 KB |
| 実行使用メモリ | 16,148 KB |
| 最終ジャッジ日時 | 2025-03-25 11:15:41 |
| 合計ジャッジ時間 | 10,985 ms |
|
ジャッジサーバーID （参考情報） |
judge3 / judge5 |
（要ログイン）
| ファイルパターン | 結果 |
|---|---|
| other | AC * 19 |
ソースコード
#include <bits/stdc++.h>
using namespace std;
#include <atcoder/lazysegtree>
//非可換なら
#define HLD_NON_COMMUTATIVE false
//addが使えるなら
#define HLD_HAS_OPERATION_ADD false
//区間作用があるなら
#define HLD_HAS_LAZY_APPLY true
template <int MAXN, class S, S (*op)(S, S), S (*e)(), class F, S (*mapping)(F, S), F (*composition)(F, F), F (*id)()>
struct hld_tree{
using node_type = atcoder::lazy_segtree<S, op, e, F, mapping, composition, id>;
int n;
int dep[MAXN], in[MAXN], out[MAXN], head[MAXN], par[MAXN];
#if HLD_NON_COMMUTATIVE
node_type node[2];
#else
node_type node;
#endif
hld_tree(int _n, vector<int>& g, const vector<int>& start, const vector<S>& v, int r = 0):
n(_n){
assert(0 <= r && r < n);
assert(n == (int)v.size());
if((int)g.size() == (n-1)*2){
int sub_siz[MAXN], idx[MAXN+1];
fill(sub_siz, sub_siz+n, 1);
memcpy(idx, start.data(), sizeof(int)*start.size());
int cur = r;
par[r] = -1;
while(true){
if(idx[cur] < start[cur+1]-(cur!=r)){
int nex = g[idx[cur]++];
if(nex == par[cur]){
g[--idx[cur]] = g[start[cur+1]-1];
continue;
}
par[nex] = cur;
cur = nex;
} else {
if(cur == r) break;
sub_siz[par[cur]] += sub_siz[cur];
cur = par[cur];
}
}
memcpy(idx, start.data(), sizeof(int)*start.size());
int state[MAXN]{}, heavy[MAXN];
int node_siz = 0;
cur = r;
head[cur] = r;
dep[cur] = 0;
while(1){
if(state[cur] == 0){
in[cur] = node_siz++;
heavy[cur] = -1;
for(int i = start[cur]; i < start[cur+1]-(cur!=r); i++){
int nex = g[i];
if(heavy[cur] == -1 || sub_siz[heavy[cur]] < sub_siz[nex]){
heavy[cur] = nex;
}
}
if(heavy[cur] != -1){
int nex = heavy[cur];
dep[nex] = dep[cur];
head[nex] = head[cur];
state[cur] = 2;
cur = nex;
} else state[cur] = 2;
} else if(state[cur] == 2){
if(idx[cur] < start[cur+1]-(cur!=r)){
int nex = g[idx[cur]++];
if(nex == heavy[cur]) continue;
dep[nex] = dep[cur]+1;
head[nex] = nex;
cur = nex;
} else state[cur] = 3;
} else {
out[cur] = node_siz;
if(cur == r) break;
cur = par[cur];
}
}
} else if((int)g.size() == (n-1)){
int sub_siz[MAXN], idx[MAXN];
fill(sub_siz, sub_siz+n, 1);
memcpy(idx, start.data(), sizeof(int)*start.size());
int cur = r;
par[r] = -1;
while(true){
if(idx[cur] < start[cur+1]){
int nex = g[idx[cur]++];
par[nex] = cur;
cur = nex;
} else {
if(cur == r) break;
sub_siz[par[cur]] += sub_siz[cur];
cur = par[cur];
}
}
memcpy(idx, start.data(), sizeof(int)*start.size());
int state[MAXN]{}, heavy[MAXN];
int node_siz = 0;
cur = r;
head[cur] = r;
dep[cur] = 0;
while(1){
if(state[cur] == 0){
in[cur] = node_siz++;
heavy[cur] = -1;
for(int i = start[cur]; i < start[cur+1]; i++){
int nex = g[i];
if(heavy[cur] == -1 || sub_siz[heavy[cur]] < sub_siz[nex]){
heavy[cur] = nex;
}
}
if(heavy[cur] != -1){
int nex = heavy[cur];
dep[nex] = dep[cur];
head[nex] = head[cur];
state[cur] = 2;
cur = nex;
} else state[cur] = 2;
} else if(state[cur] == 2){
if(idx[cur] < start[cur+1]){
int nex = g[idx[cur]++];
if(nex == heavy[cur]) continue;
dep[nex] = dep[cur]+1;
head[nex] = nex;
cur = nex;
} else state[cur] = 3;
} else {
out[cur] = node_siz;
if(cur == r) break;
cur = par[cur];
}
}
} else assert(0);
vector<S> node_v(n);
for(int i = 0; i < n; i++) node_v[in[i]] = v[i];
#if HLD_NON_COMMUTATIVE
node[0] = node_type(node_v);
reverse(node_v.begin(), node_v.end());
node[1] = node_type(node_v);
#else
node = node_type(node_v);
#endif
}
#if HLD_HAS_OPERATION_ADD
#if HLD_NON_COMMUTATIVE
void add(int p, const S& x){
assert(0 <= p && p < n);
node[0].add(in[p], x);
node[1].add(n-in[p]-1, x);
}
#else
void add(int p, const S& x){
assert(0 <= p && p < n);
node.add(in[p], x);
}
#endif
#endif
#if HLD_NON_COMMUTATIVE
S prod(int l, int r){
assert(0 <= l && 0 <= r && l < n && r < n);
S l_ans = e();
S r_ans = e();
bool rev = false;
while(true){
if(dep[l] < dep[r]) swap(l, r), rev ^= 1;
if(dep[l] != dep[r] || head[l] != head[r]){
if(!rev) l_ans = op(l_ans, node[1].prod(n-in[l]-1, n-in[head[l]]));
else r_ans = op(node[0].prod(in[head[l]], in[l]+1), r_ans);
l = par[head[l]];
} else {
if(in[l] > in[r]) swap(l, r), rev ^= 1;
if(!rev) l_ans = op(l_ans, node[0].prod(in[l], in[r]+1));
else r_ans = op(node[1].prod(n-in[r]-1, n-in[l]), r_ans);
return op(l_ans, r_ans);
}
}
}
S prod(int r){
assert(0 <= r && r < n);
return node[0].prod(in[r], out[r]);
}
S get(int p){
assert(0 <= p && p < n);
return node[0].get(in[p]);
}
void set(int p, const S& x){
assert(0 <= p && p < n);
node[0].set(in[p], x);
node[1].set(n-in[p]-1, x);
}
#else
S prod(int l, int r){
assert(0 <= l && 0 <= r && l < n && r < n);
S ans = e();
while(true){
if(dep[l] < dep[r]) swap(l, r);
if(dep[l] != dep[r] || head[l] != head[r]){
ans = op(ans, node.prod(in[head[l]], in[l]+1));
l = par[head[l]];
} else {
if(in[l] > in[r]) swap(l, r);
ans = op(ans, node.prod(in[l], in[r]+1));
return ans;
}
}
}
S prod(int r){
assert(0 <= r && r < n);
return node.prod(in[r], out[r]);
}
S get(int p){
assert(0 <= p && p < n);
return node.get(in[p]);
}
void set(int p, const S& x){
assert(0 <= p && p < n);
node.set(in[p], x);
}
#endif
int lca(int l, int r){
assert(0 <= l && 0 <= r && l < n && r < n);
while(true){
if(dep[l] < dep[r]) swap(l, r);
if(dep[l] != dep[r] || head[l] != head[r]){
l = par[head[l]];
} else {
if(in[l] > in[r]) swap(l, r);
return l;
}
}
}
int dist(int l, int r){
assert(0 <= l && 0 <= r && l < n && r < n);
int d = 0;
while(true){
if(dep[l] < dep[r]) swap(l, r);
if(dep[l] != dep[r] || head[l] != head[r]){
d += in[l] - in[head[l]] + 1;
l = par[head[l]];
} else {
if(in[l] > in[r]) swap(l, r);
d += in[r] - in[l];
return d;
}
}
}
#if HLD_HAS_LAZY_APPLY
#if HLD_NON_COMMUTATIVE
void apply(int l, int r, const F& x){
while(true){
if(dep[l] < dep[r]) swap(l, r);
if(dep[l] != dep[r] || head[l] != head[r]){
node[0].apply(in[head[l]], in[l]+1, x);
node[1].apply(n-in[l]-1, n-in[head[l]], x);
l = par[head[l]];
} else {
if(in[l] > in[r]) swap(l, r);
node[0].apply(in[l], in[r]+1, x);
node[1].apply(n-in[r]-1, n-in[l], x);
break;
}
}
}
void apply(int r, const F& x){
node[0].apply(in[r], out[r], x);
node[1].apply(n-out[r], n-in[r], x);
}
#else
void apply(int l, int r, const F& x){
while(true){
if(dep[l] < dep[r]) swap(l, r);
if(dep[l] != dep[r] || head[l] != head[r]){
node.apply(in[head[l]], in[l]+1, x);
l = par[head[l]];
} else {
if(in[l] > in[r]) swap(l, r);
node.apply(in[l], in[r]+1, x);
break;
}
}
}
void apply(int r, const F& x){
node.apply(in[r], out[r], x);
}
#endif
#endif
};
void read_csr_ud_graph(vector<int>& g, vector<int>& start, int m, int num = -1){
assert((int)g.size() == 2*m);
int n = (int)start.size()-1;
vector<array<int, 2>> E(m);
for(int i = 0; i < m; i++){
int u, v;
cin >> u >> v;
u += num; v += num;
start[u+1]++;
start[v+1]++;
E[i][0] = u;
E[i][1] = v;
}
for(int i = 1; i <= n; i++) start[i] += start[i-1];
auto C = start;
for(int i = 0; i < m; i++){
auto [u, v] = E[i];
g[C[u]++] = v;
g[C[v]++] = u;
}
}
struct S{
long long value;
int size;
};
using F = long long;
S op(S a, S b){ return {a.value+b.value, a.size+b.size}; }
S e(){ return {0, 0}; }
S mapping(F f, S x){ return {x.value + f*x.size, x.size}; }
F composition(F f, F g){ return f+g; }
F id(){ return 0; }
int main(){
cin.tie(0)->sync_with_stdio(0);
int n, q;
cin >> n;
vector<int> G((n-1)*2), start(n+1);
read_csr_ud_graph(G, start, n-1);
hld_tree<100000, S, op, e, F, mapping, composition, id> tree(n, G, start, vector<S>(n, {1, 1}));
long long ans = 0;
cin >> q;
while(q--){
int u, v;
cin >> u >> v;
u--; v--;
ans += tree.prod(u, v).value;
tree.apply(u, v, 1);
}
cout << ans << '\n';
}