洛谷P3258 JLOI2014 松鼠的新家

2022-08-13 23:48:20 字數 2632 閱讀 6435

松鼠的新家是一棵樹,前幾天剛剛裝修了新家,新家有n個房間,並且有n-1根樹枝連線,每個房間都可以相互到達,且倆個房間之間的路線都是唯一的。天哪,他居然真的住在」樹「上。

松鼠想邀請****前來參觀,並且還指定乙份參觀指南,他希望**能夠按照他的指南順序,先去a1,再去a2,......,最後到an,去參觀新家。可是這樣會導致**重複走很多房間,懶惰的**不聽地推辭。可是松鼠告訴他,每走到乙個房間,他就可以從房間拿一塊糖果吃。

**是個饞傢伙,立馬就答應了。現在松鼠希望知道為了保證**有糖果吃,他需要在每乙個房間各放至少多少個糖果。

因為松鼠參觀指南上的最後乙個房間an是餐廳,餐廳裡他準備了豐盛的大餐,所以當**在參觀的最後到達餐廳時就不需要再拿糖果吃了。

輸入格式:

第一行乙個整數n,表示房間個數第二行n個整數,依次描述a1-an接下來n-1行,每行兩個整數x,y,表示標號x和y的兩個房間之間有樹枝相連。

輸出格式:

一共n行,第i行輸出標號為i的房間至少需要放多少個糖果,才能讓**有糖果吃。

輸入樣例#1:

5

1 4 5 3 2

1 22 4

2 34 5

輸出樣例#1:

121

21

2<= n <=300000

題解:分明就是一道樹鏈剖分模板題

可是我還是寫+調花了1個*時(55555……)

好傷心啊,看來這些模板該再練練了,千萬不能考試時又花這麼長時間

**:

1 #include2 #include3 #include

4using

namespace

std;56

const

int maxn = 300005;7

struct

nodepool[2*maxn],*h[maxn];

11int

cnt;

12int

fa[maxn],dep[maxn],son[maxn],sonnum[maxn],vis[maxn];

13int

a[maxn];

1415

void addedge(int u,int

v)20

21void dfs(int

u)34

if(maxnum!=0) son[u]=maxson;

35else son[u]=-1;36

}3738int

w[maxn],rk[maxn],t,top[maxn];

39void dfs2(int

u)49

for(node *p=h[u];p;p=p->next)

50if(!vis[v=p->v] && v!=son[u])56}

5758

struct

treed[maxn*8],*root;

62int

cnt1;

63void build_tree(tree *p,int l,int

r)72

void change(tree *p,int l,int r,int

c)77

if(p->lazy)

82if(l>=p->ch[1]->l) change(p->ch[1

],l,r,c);

83else

if(r<=p->ch[0]->r) change(p->ch[0

],l,r,c);

84else88}

89int query(tree *p,int l,int

r)93

if(p->lazy)

98if(l>=p->ch[1]->l) return query(p->ch[1

],l,r);

99else

if(r<=p->ch[0]->r) return query(p->ch[0

],l,r);

100else

103}

104105

void lca(int x,int

y)112

if(dep[x]>dep[y]) swap(x,y);

113 change(root,rk[x],rk[y],1

);114

}115

intans[maxn];

116117

intmain()

118126

127 root=&d[++cnt1];

128 dep[1]=1

;129 dfs(1

);130 memset(vis,0,sizeof

(vis));

131 top[1]=1;rk[1]=1;t=1;w[1]=1

;132 dfs2(1

);133 build_tree(root,1

,t);

134135

for(i=1;i1

]);136

137for(i=1;i<=n;i++)

142for(i=1;i<=n;i++) printf("

%d\n

",ans[i]);

143144

return0;

145 }

view code

洛谷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,去參觀新家。可是這樣會導...

P3258 JLOI2014 松鼠的新家

松鼠的新家是一棵樹,前幾天剛剛裝修了新家,新家有n個房間,並且有n 1根樹枝連線,每個房間都可以相互到達,且倆個房間之間的路線都是唯一的。天哪,他居然真的住在 樹 上。松鼠想邀請 前來參觀,並且還指定乙份參觀指南,他希望 能夠按照他的指南順序,先去a1,再去a2,最後到an,去參觀新家。可是這樣會導...