松鼠的新家是一棵樹，前幾天剛剛裝修了新家，新家有n個房間，並且有n-1根樹枝連線，每個房間都可以相互到達，且倆個房間之間的路線都是唯一的。天哪，他居然真的住在」樹「上。
松鼠想邀請****前來參觀，並且還指定乙份參觀指南，他希望**能夠按照他的指南順序，先去a1，再去a2，......，最後到an，去參觀新家。可是這樣會導致**重複走很多房間，懶惰的**不停地推辭。可是松鼠告訴他，每走到乙個房間，他就可以從房間拿一塊糖果吃。
**是個饞傢伙，立馬就答應了。現在松鼠希望知道為了保證**有糖果吃，他需要在每乙個房間各放至少多少個糖果。
因為松鼠參觀指南上的最後乙個房間an是餐廳，餐廳裡他準備了豐盛的大餐，所以當**在參觀的最後到達餐廳時就不需要再拿糖果吃了。
輸入格式：
第一行乙個整數n，表示房間個數第二行n個整數，依次描述a1-an接下來n-1行，每行兩個整數x，y，表示標號x和y的兩個房間之間有樹枝相連。
輸出格式：
一共n行，第i行輸出標號為i的房間至少需要放多少個糖果，才能讓**有糖果吃。
輸入樣例#1： 複製
51 4 5 3 2
1 22 4
2 34 5
輸出樣例#1： 複製
12121
2<= n <=300000
//記錄從哪乙個房間出發，update(出發點，終點,add)
//因為從房間出發的時候不需要糖果，到達的時候才需要糖果
//所以update的時候會在出發點多放一塊糖果
//那我們就update(s,s,-1),把多放的那塊減去
//第乙個房間是起點，提前在for迴圈外邊放上一塊糖。
//last記錄起點是誰
//這樣，在每次update之前都先把last上的糖--，然後update（區間）
//最後讓終點的糖--，因為題目中說最後乙個點是餐廳，不用放糖
#include
#include
#include
#include
using
namespace
std;
const
int n=3e5+5
;int
n;int
a[n];
inthead[n],num_edge;
struct
edge
edge[n
<<1
];struct
node
node[n];
struct
tree
tree[n
<<2
];typedef tree*tree;
tree now_node=tree,root;
inline
intread()
inline
void add_edge(int u,int
v)void dfs1(intu)}
intbound;
void dfs2(int u,int
top)
}node[u].t=bound;
}void build(tree &root,int l,int
r)inline
void
pushdown(tree root)
}void update(const tree &root,int l,int r,int
add)
pushdown(root);
if(r<=root->mid)
update(root->lson,l,r,add);
else
if(l>root->mid)
update(root->rson,l,r,add);
else
//root->sum=root->lson->sum+=root->rson->sum; 不用pushup
}int query(const tree &root,int
pos)
inline
void modify(int x,int
y)
else
}if(node[x].dep>node[y].dep)
update(root,node[y].s,node[x].s,1);
else
update(root,node[x].s,node[y].s,1);
}int
main()
dfs1(1);
dfs2(
1,1);
build(root,
1,n);
int last=a[1
]; update(root,node[last].s,node[last].s,
1); //
起點放糖
for(int i=2;i<=n;++i)
update(root,node[last].s,node[last].s,-1); //
終點的糖--
for(int i=1;i<=n;++i) //
查詢每個房間要放多少糖
printf("
%d\n
",query(root,node[i].s));
return0;
}
P3258 JLOI2014 松鼠的新家
傳送門 此題樹剖可過 然而可以樹上差分為什麼要樹剖.對於一條路徑 a，b 只要把 val a val b val lca a,b val fa lca a,b 那麼求每個點的經過次數就求一下子樹 val 的和就好了 但是要注意，每一條路徑的終點是下一條路徑的起點，那些點會被多算一次 所以求完子樹和以...
洛谷P3258 JLOI 2014松鼠的新家
題目 樹上差分 樹上點差分，注意會出現路徑端點多記錄的情況，這時需要在最後輸出的時候輸出子樹的差分陣列的和 1，而不是在處理原資料的時候減1。並且a n 不需要糖果，最後也減去就行。include define n 1001011 using namespace std struct edg e n...
洛谷P3258 JLOI2014 松鼠的新家
松鼠的新家是一棵樹，前幾天剛剛裝修了新家，新家有n個房間，並且有n 1根樹枝連線，每個房間都可以相互到達，且倆個房間之間的路線都是唯一的。天哪，他居然真的住在 樹 上。松鼠想邀請 前來參觀，並且還指定乙份參觀指南，他希望 能夠按照他的指南順序，先去a1，再去a2，最後到an，去參觀新家。可是這樣會導...