微软笔试题

题目三

\quad 给一个无向图，N个顶点，M条边，0为起点，N-1为终点，每条边初始权值为 1。图中除普通节点外有 4 种节点。

• 第一种：走过这种节点后的两条边权值翻倍(Sand)
• 第二种：走过这种节点后的两条边权值减半 (Nitro)
• 第三种：走到这个节点就停止，不能再走了(Cop)
• 第四种：走到这个节点，下一条边的权值+1(Crash)

求节点 0 到 N-1 的最短权值和路径。

输入样例：第一行为点数和边数；第二行为n个字符串表示这n个点的属性；接下来有m行，每行两个节点编号，表示这两个节点有一条无向边连接。
7 8
None Cop Sand None Nitro None None
0 1
0 2
1 2
2 3
2 4
3 6
4 5
5 6

思路：直接爆搜求出所有0到n-1节点的路径(Cop类型的节点不予考虑)，然后再一一计算路径，找出最短权值和路径。

#include <iostream>
#include <vector>
#include <unordered_map>
using namespace std;

const int N = 110;
vector<int> E[N];  // 邻接表存边
unordered_map<int, string> attr;  // 记录节点属性
vector<int> path;  // 存路径
int vis[N];

vector<int> res;
double dis = 1e18;
double calc(vector<int> &path){
	double cost = 0.0;
	int e = path.size() - 1; // 边数
	vector<double> edgeWeight(e, 1.0);  // 初始化各个边的权值为1
	for(int i = 0; i < e; i ++ ){
		int node = path[i];
		if(attr[node] == "Sand"){
			edgeWeight[i] *= 2;
			if(i + 1 < e) edgeWeight[i + 1] *= 2;
		}
		else if(attr[node] == "Nitro"){
			edgeWeight[i] /= 2;
			if(i + 1 < e) edgeWeight[i + 1] /= 2;
		}
		else if(attr[node] == "Crash"){
			edgeWeight[i] += 1;
		}
	}
	for(auto w: edgeWeight) cost += w;
		return cost;
}
void dfs(int u, int end){
	// 当这个点不是终点且属性是Cop时无法通过
	if(u != end && attr[u] == "Cop") return;
	path.push_back(u);
	vis[u] = 1;
	if(u == end){
		double cost = calc(path);  // 计算这条路径的花费
		if(cost < dis){
			dis = cost;
			res = path;
		}
	}
	for(auto v: E[u]){
		if(vis[v]) continue;
		dfs(v, end);
	}
	path.pop_back();
	vis[u] = 0;
}
int main(int argc, char const *argv[])
{
	int n, m; cin >> n >> m;  //点数和边数	
	for(int i = 0; i < n; i ++ ){  // 读入节点属性
		string s; cin >> s;
		attr[i] = s;
	}
	for(int i = 0; i < m; i ++ ){
		int u, v; cin >> u >> v;
		E[u].push_back(v);
		E[v].push_back(u);
	}
	dfs(0, n - 1);  // 起点和终点的所有路径

	// 输出最小花费
	cout << dis << endl;
	// 输出最短路径
	for(auto node: res) cout << node << " ";
	return 0;
}