Description

Input

Output

Sample Input

4
2 3
2 1
2 4

Sample Output

4

Data Constraint

Solution

• 首先，這是一棵有根樹，先要找到它的根是哪個(gè)節(jié)點(diǎn)。

• 由于邊是有向邊，入度為零的點(diǎn)就是根節(jié)點(diǎn)了。

• 接著，如何處理出答案呢？

• 從根節(jié)點(diǎn)開始遍歷這棵樹，做一次 DFS。記錄三個(gè)值：

• Size[i] 表示以 i 為根的子樹的大小（節(jié)點(diǎn)數(shù)）；
• Mx[i] 表示以 i 為根的子樹中節(jié)點(diǎn)標(biāo)號(hào)最大是多少；
• Mn[i] 表示以 i 為根的子樹中節(jié)點(diǎn)標(biāo)號(hào)最小是多少；

• 那么對(duì)于一個(gè)節(jié)點(diǎn) i ，如果滿足：Mx[i]?Mn[i]+1=Size[i]

• 則說明點(diǎn) i 符合條件，答案加一。時(shí)間復(fù)雜度 O(N) 。

Code

#include<cstdio> using namespace std; const int N=100001; int n,tot,ans; int first[N],next[N],en[N]; int size[N],mx[N],mn[N]; bool bz[N]; inline int read() {int X=0,w=1; char ch=0;while(ch<'0' || ch>'9') {if(ch=='-') w=-1;ch=getchar();}while(ch>='0' && ch<='9') X=(X<<3)+(X<<1)+ch-'0',ch=getchar();return X*w; } inline void dfs(int x) {size[mn[x]=mx[x]=x]=1;for(int i=first[x];i;i=next[i]){dfs(en[i]);size[x]+=size[en[i]];if(mx[en[i]]>mx[x]) mx[x]=mx[en[i]];if(mn[en[i]]<mn[x]) mn[x]=mn[en[i]];}if(mx[x]-mn[x]+1==size[x]) ans++; } int main() {int n=read();for(int i=1;i<n;i++){int x=read(),y=read();next[++tot]=first[x];first[x]=tot;bz[en[tot]=y]=true;}for(int i=1;i<=n;i++)if(!bz[i]){dfs(i);break;}printf("%d",ans);return 0; }

總結(jié)

以上是生活随笔為你收集整理的JZOJ 5221. 【GDOI2018模拟7.10】A的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。