【2019复旦机试】（工研院）C 通信代价 [DFS]

C 通信代价 [DFS]

题目描述
给定一个无向图，保证是一棵树，定义两个结点 ab 之间的通信代价为 ab 路径上的边的数目，节点 a 的通信代价为 a 到其他所有节点的通信代价之和。在一行中输出所有结点的通信代价。输入的一行数字表示节点数，下面每行各有两个数字a和b，表示节点a到节点b有直接路径。
样例输入：
4
1 2
2 3
2 4
输出：
5 3 5 5
 

这题想了想，无向图，没有环，所以a到b的路径只有唯一一条！DFS

思路：

main函数：

一个for循环，从节点i   从1到n：

        cost = 0；维护从节点i到其他未访问过的节点的路径总和；

        temp = 0；维护从节点i到某一节点j的路径数，找到新的时候，cost += temp;

        DFS（i , 其他参数）

DFS函数：

void findShortestPath(vector<vector<int>>& grid, vector<bool>& visit,int& start, int& cost, int& temp)

        grid为图的矩阵存储（根据题意0行0列无意义
        visit为访问数组，初始化全为false，达到一个节点，变为true 
        start为当前位置

        cost为目前记录到的到其他节点的总的路径数 
        temp记录的是到某一节点的单一路径数
        每当走到一个之前未曾访问的节点temp += 1; cost += temp; visit = true;
        对start可以到达的节点但是还未访问过的节点进行DFS（）
        回溯一个节点时，visit变false，temp -= 1; 

        

DFS代码如下：

/*
DFS
grid为图的矩阵存储（根据题意0行0列无意义
visit为访问数组，初始化为false，达到一个节点，变为true 
当前位置为start， 目前记录到的到其他节点的总的路径数为 cost, 
temp记录的是到某一节点的单一路径数
每当走到一个之前未曾访问的节点temp += 1; cost += temp; visit = true; 
接着DFS
回溯一个节点时，visit变false，temp -= 1; 
*/ 
 
void findShortestPath(vector<vector<int>>& grid, vector<bool>& visit,int& start, int& cost, int& temp){
	for(int i = 1; i < grid[start].size(); i++){
		if((!visit[i]) && grid[start][i] == 1){
			visit[i] = true;
			temp += 1;
			cost += temp;
			findShortestPath(grid, visit, i, cost, temp);
			visit[i] = false;
			temp -= 1;
		}
	}
} 

整体代码：

#include<iostream>
#include<vector>
#include<map>
#include<cmath>
using namespace std;
/*
DFS
grid为图的矩阵存储（根据题意0行0列无意义
visit为访问数组，初始化为false，达到一个节点，变为true 
当前位置为start， 目前记录到的到其他节点的总的路径数为 cost, 
temp记录的是到某一节点的单一路径数
每当走到一个之前未曾访问的节点temp += 1; cost += temp; visit = true; 
接着DFS
回溯一个节点时，visit变false，temp -= 1; 
*/ 
 
void findShortestPath(vector<vector<int>>& grid, vector<bool>& visit,int& start, int& cost, int& temp){
	for(int i = 1; i < grid[start].size(); i++){
		if((!visit[i]) && grid[start][i] == 1){
			visit[i] = true;
			temp += 1;
			cost += temp;
			findShortestPath(grid, visit, i, cost, temp);
			visit[i] = false;
			temp -= 1;
		}
	}
} 
int main(){
	int n;
	cin >> n;
	vector<vector<int>> grid = vector<vector<int>>(n + 1, vector<int>(n + 1, 0));
	//结果集final——cost
    vector<int> final_cost;

    //键入矩阵存储
	for(int i = 1; i < n; i++){
		int a,b;
		cin >> a >> b;
		grid[a][b] = 1;
		grid[b][a] = 1;
	}
    //从1~n号节点开始DFS，每个结点到其他结点的所有路径和保存在cost里，中间到某结点结果为temp
	for(int i = 1; i <= n; i++){
		vector<bool> visit = vector<bool>(n+1, false);
		visit[i] = true;
		int cost = 0, temp = 0, start = i;
		findShortestPath(grid, visit, start, cost, temp);
		final_cost.push_back(cost);	
	}
	 
	for(int i = 0; i < final_cost.size(); i++){
		cout << final_cost[i] ;
		 
	}
	

	return 0;
	
	
}

运行结果：