new file: P3258.cpp

P3258.cpp

@@ -0,0 +1,99 @@
+#include<bits/stdc++.h>
+#define int long long
+using namespace std;
+int n;
+int a[300005];
+int fa[300005];
+int dfn[300005];
+int son[300005];
+int dep[300005];
+int top[300005];
+int cnt[300005];
+vector<int> fri[300005];
+int dfnnow;
+struct SEG{
+    int num[1200005], lt[1200005];
+    void pu(int now) {
+        num[now] = num[now * 2] + num[now * 2 + 1];
+    }
+    void pd(int now, int nl, int nr) {
+        int nmid = (nl + nr) >> 1, l = lt[now];
+        num[now * 2] += l * (nmid - nl + 1);
+        num[now * 2 + 1] += l * (nr - nmid);
+        lt[now * 2] += l;
+        lt[now * 2 + 1] += l;
+        lt[now] = 0;
+    }
+    void add(int now, int nl, int nr, int l, int r, int nn) {
+        if(nr < l || nl > r) return;
+        if(l <= nl && nr <= r) {
+            num[now] += (nr - nl + 1) * nn;
+            lt[now] += nn;
+            return;
+        }
+        int nmid = (nl + nr) >> 1;
+        pd(now, nl, nr);
+        add(now * 2, nl, nmid, l, r, nn);
+        add(now * 2 + 1, nmid + 1, nr, l, r, nn);
+        pu(now);
+    }
+    int query(int now, int nl, int nr, int l, int r) {
+        if(nr < l || nl > r) return 0;
+        if(l <= nl && nr <= r) {
+            return num[now];
+        }
+        int nmid = (nl + nr) >> 1;
+        pd(now, nl, nr);
+        return query(now * 2, nl, nmid, l, r) + query(now * 2 + 1, nmid + 1, nr, l, r);
+    }
+}seg;
+void dfs1(int now, int from) {
+    cnt[now] = 1;
+    for(int i : fri[now]) {
+        if(i == from) continue;
+        fa[i] = now;
+        dep[i] = dep[now] + 1; 
+        dfs1(i, now);
+        cnt[now] += cnt[i];
+        if(cnt[son[now]] < cnt[i]) son[now] = i;
+    }
+}
+void dfs2(int now, int from) {
+    top[now] = from;
+    dfn[now] = ++dfnnow;
+    if(son[now]) dfs2(son[now], from);
+    for(int i : fri[now]) {
+        if(i == fa[now] || i == son[now]) continue;
+        dfs2(i, i);
+    }
+}
+void edit(int a, int b, int nn = 1) {
+    while(top[a] != top[b]) {
+        if(dep[top[a]] < dep[top[b]]) swap(a, b);
+        seg.add(1, 1, n, dfn[top[a]], dfn[a], nn);
+        a = fa[top[a]];
+    }
+    if(dep[a] < dep[b]) swap(a, b);
+    seg.add(1, 1, n, dfn[b], dfn[a], nn);
+}
+signed main() {
+    scanf("%lld", &n);
+    for(int i = 1; i <= n; ++i) {
+        scanf("%lld", &a[i]);
+    }
+    for(int i = 1, u, v; i < n; ++i) {
+        scanf("%lld%lld", &u, &v);
+        fri[u].push_back(v);
+        fri[v].push_back(u);
+    }
+    dfs1(1, -1);
+    dfs2(1, 1);
+    for(int i = 1; i < n; ++i) {
+        edit(a[i], a[i + 1]);
+        edit(a[i + 1], a[i + 1], -1);
+    }
+    for(int i = 1; i <= n; ++i) {
+        printf("%lld\n", seg.query(1, 1, n, dfn[i], dfn[i]));
+    }
+    return 0;
+}

P3258.md

@@ -0,0 +1,48 @@
+# P3258 [JLOI2014] 松鼠的新家
+
+## 题目描述
+
+松鼠的新家是一棵树，前几天刚刚装修了新家，新家有 $n$ 个房间，并且有 $n-1$ 根树枝连接，每个房间都可以相互到达，且两个房间之间的路线都是唯一的。天哪，他居然真的住在“树”上。
+
+松鼠想邀请小熊维尼前来参观，并且还指定一份参观指南，他希望维尼能够按照他的指南顺序，先去 $a_1$，再去 $a_2$，……，最后到 $a_n$，去参观新家。可是这样会导致重复走很多房间，懒惰的维尼不停地推辞。可是松鼠告诉他，每走到一个房间，他就可以从房间拿一块糖果吃。
+
+维尼是个馋家伙，立马就答应了。现在松鼠希望知道为了保证维尼有糖果吃，他需要在每一个房间各放至少多少个糖果。
+
+因为松鼠参观指南上的最后一个房间 $a_n$ 是餐厅，餐厅里他准备了丰盛的大餐，所以当维尼在参观的最后到达餐厅时就不需要再拿糖果吃了。
+
+## 输入格式
+
+第一行一个正整数 $n$，表示房间个数。第二行 $n$ 个正整数，依次描述 $a_1, a_2,\cdots,a_n$。
+
+接下来 $n-1$ 行，每行两个正整数 $x,y$，表示标号 $x$ 和 $y$ 的两个房间之间有树枝相连。
+
+## 输出格式
+
+一共 $n$ 行，第 $i$ 行输出标号为 $i$ 的房间至少需要放多少个糖果，才能让维尼有糖果吃。
+
+## 输入输出样例 #1
+
+### 输入 #1
+
+```
+5
+1 4 5 3 2
+1 2
+2 4
+2 3
+4 5
+```
+
+### 输出 #1
+
+```
+1
+2
+1
+2
+1
+```
+
+## 说明/提示
+
+对于全部的数据，$2 \le n \le 3 \times 10^5$，$1 \le a_i \le n$。