Description
顺利潜入勺林寺几天后,方丈给了西瓜一个光荣而艰巨的任务——打扫寺庙中的道路。同时给了西瓜一张勺林寺的地图。
西瓜这才知道,勺林寺中总共有n座房子,但道路只有n-1条,这n-1条道路连接了寺中的所有房子,即保证在任何两座房子都能沿着道路抵达。
好在西瓜人缘不错,他知道每座房子中都有个自己的朋友,只要给他们每个人打个电话,让他到自己这里来,顺便把路也扫了,即给某座房子中的朋友打过电话后,可认为从该房子到西瓜所在的位置之间所有的道路都会被打扫干净。
同时西瓜还知道,这n-1条道路中有一些路昨天已经被人打扫过了不需要再打扫一遍。
现在西瓜想知道,自己最少要给多少个朋友打电话才能完成方丈给的任务。
西瓜在编号为1的房子中。
西瓜这才知道,勺林寺中总共有n座房子,但道路只有n-1条,这n-1条道路连接了寺中的所有房子,即保证在任何两座房子都能沿着道路抵达。
好在西瓜人缘不错,他知道每座房子中都有个自己的朋友,只要给他们每个人打个电话,让他到自己这里来,顺便把路也扫了,即给某座房子中的朋友打过电话后,可认为从该房子到西瓜所在的位置之间所有的道路都会被打扫干净。
同时西瓜还知道,这n-1条道路中有一些路昨天已经被人打扫过了不需要再打扫一遍。
现在西瓜想知道,自己最少要给多少个朋友打电话才能完成方丈给的任务。
西瓜在编号为1的房子中。
Input
输入包含多组数据
每组数据第一行包含一个n(2 ≤ n ≤ 10^5),表示有n座房子
之后n-1行,每行三个数字a,b,c表示在房子a和房子b之间有一条路,c等于1表示路已被打扫,c等于2表示路未被打扫。
每组数据第一行包含一个n(2 ≤ n ≤ 10^5),表示有n座房子
之后n-1行,每行三个数字a,b,c表示在房子a和房子b之间有一条路,c等于1表示路已被打扫,c等于2表示路未被打扫。
Output
每组数据输出一个整数k,表示最少要给k个朋友打电话
Sample Input
Original | Transformed |
5 1 2 2 1 3 2 1 4 2 1 5 2
Sample Output
Original | Transformed |
4
解题思路:
bfs + dfs
代码:
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <cstdio>
#include <string>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <functional>
using namespace std;
#define mod 7
#define INF 0x3f3f3f3f
#define lson (rt << 1)
#define rson (rt << 1 | 1)
#define Clear(a) memset(a, 0, sizeof(a))
#define Max(a, b) ( (a) > (b) ? (a) : (b) )
#define Min(a, b) ( (a) < (b) ? (a) : (b) )
typedef long long LL;
typedef pair<int, int > pi;
typedef struct node{
int to;
int w;
node(){}
node(int a, int b){
to = a; w = b;
}
}Edge;
typedef struct nn{
int lev;
int id;
}SS;
const int maxn = 1e5 + 5;
const int dir[8][2] = {1,2, 2,1, -1,2, -2,1, 1,-2, 2,-1, -1,-2, -2,-1};
int ans;
SS s[maxn];
int vis[maxn], num[maxn];
vector<Edge> vec[maxn];
bool cmp(SS a, SS b){
if(a.lev == b.lev){
return a.id < b.id;
}else{
return a.lev > b.lev;
}
}
void dfs(int p){
vis[p] = 1;
if(p == 1) return;
int len = vec[p].size();
for(int i = 0; i < len; ++i){
Edge v = vec[p][i];
if(num[v.to] < num[p])
dfs(v.to);
}
}
void bfs(){
queue<int> q;
memset(s, 0, sizeof(s));
while(!q.empty()) q.pop();
q.push(1);
s[1].id = 1;
s[1].lev = 1;
while(!q.empty()){
int now = q.front();
q.pop();
int len = vec[now].size();
for(int i = 0; i < len; ++i){
Edge v = vec[now][i];
if(!s[v.to].lev){
s[v.to].id = v.to;
s[v.to].lev = s[now].lev + 1;
q.push(v.to);
}
}
}
}
int main()
{
int n, a, b, c;
while(~scanf("%d", &n)){
for(int i = 0; i <= n; ++i) vec[i].clear();
for(int i = 1; i < n; ++i){
scanf("%d%d%d", &a, &b, &c);
vec[a].push_back(node(b, c));
vec[b].push_back(node(a, c));
}
bfs();
for(int i = 1; i <= n; ++i) num[i] = s[i].lev;
sort(s + 1, s + n + 1, cmp);
memset(vis, 0, sizeof(vis));
int ans = 0;
for(int i = 1; i <= n; ++i){
if(vis[s[i].id]) continue;
int len = vec[s[i].id].size();
for(int j = 0; j < len; ++j){
Edge tmp = vec[s[i].id][j];
if(tmp.w == 2){
++ans;
dfs(s[i].id);
}
}
}
printf("%d\n", ans);
}
return 0;
}