nowcoder 牛客練習賽9 B

ACM_e發表於2017-12-29

題目描述

珂朵莉給你一個有根樹，求有多少個子樹滿足其內部節點編號在值域上連續

一些數在值域上連續的意思即其在值域上構成一個連續的區間

輸入描述:

第一行有一個整數n，表示樹的節點數。
接下來n–1行，每行兩個整數x,y,表示存在一條從x到y的有向邊。
輸入保證是一棵有根樹。

輸出描述:

輸出一個數表示答案
示例1

輸入

5
2 3
2 1
2 4
4 5

輸出

5

說明

節點1子樹中編號為1，值域連續
節點3子樹中編號為3，值域連續
節點5子樹中編號為5，值域連續
節點4子樹中編號為4,5，值域連續
節點2子樹中編號為1,2,3,4,5，值域連續

備註:

對於100%的資料，有n <=100000

找到相應節點的最大最小值 和 節點總個數 xjb 搜一下 wa 了一次   陣列開小了

#include<bits/stdc++.h>
using namespace std;
struct ac{
   int mi,mx,sum;
   ac(){
     mi=999999;
     mx=0;
     sum=1;
   }
}a[1000001];
int ans=0;
vector<int>q[100000];
void DFS(int x){
   // cout<<"1"<<endl;
    if(q[x].size()==0){
     //  a[x].sum=1;
       a[x].mi=x;
       a[x].mx=x;
       ans++;
       return ;
    }
    //cout<<q[x].size()<<endl;
    for(int j=0;j<q[x].size();j++){
       // cout<<q[x][j]<<endl;
       DFS(q[x][j]);
       a[x].sum+=a[q[x][j]].sum;
       a[x].mx=max(a[x].mx,a[q[x][j]].mx);
       a[x].mi=min(a[x].mi,a[q[x][j]].mi);
       //cout<<a[q[x][j]].sum<<" "<<q[x][j]<<" "<<ans<<endl;
      /* if(a[x].sum==a[x].mx-a[x].mi){
         ans++;
       //cout<<ans<<endl;
       }*/
       //cout<<a[q[x][j]].sum<<" "<<q[x][j]<<" "<<ans<<endl;
    }
    if(a[x].sum==a[x].mx-a[x].mi+1){
       ans++;
       //cout<<ans<<endl;
    }
    //cout<<a[x].mi<<" "<<a[x].mx<<" "<<a[x].sum<<endl;
}
int main(){
  int n;
  cin>>n;
  int k=0;
  for(int j=0;j<n-1;j++){
      int x,y;
      cin>>x>>y;
      if(j==0) k=x;
      q[x].push_back(y);
      a[x].mi=x;
      a[x].mx=x;
      a[y].mi=y;
      a[y].mx=y;
     // cout<<q[x][0]<<endl;
  }
 // cout<<k<<endl;
  DFS(k);
  cout<<ans<<endl;
  return 0;
}







相關文章