这道题有点像并查集,但其实题目直接问距离,表示是否一个集合案例已经帮我们判断好了。变量名没像其他人那么简单,该有的注释标上了,直接上AC代码。
#include <iostream>
#include <map>
#include <set>
#include <vector>
#define min(a,b) a>b?b:a
using namespace std;
const int MaxNum = 40005;
map<int, int>To[MaxNum];//对于每一个点放置已连接的节点距离键值对
int CaseNum, HouseNum, OrderNum, x, y, dis, rootx, rooty, ans, Endans;
bool visited[MaxNum];//用于dfs判重
set<int>hav;//放置已有节点名,一个案例结束后方便To清零
void dfs(int x, int&y)
{
if (x == y)//如果到目标了
{
Endans = min(ans, Endans);
return;
}
visited[x] = true;
for (auto it = To[x].begin(); it != To[x].end(); it++)//依次进入每个节点
{
if (visited[it->first])continue;//判重
ans += it->second;
dfs(it->first,y);
ans -= it->second;
}
visited[x] = false;
}
int main()
{
memset(visited, false, sizeof(visited));
cin >> CaseNum;
while (CaseNum--)
{
cin >> HouseNum >> OrderNum;
HouseNum--;
while (HouseNum--)//收集房屋信息
{
cin >> x >> y >> dis;
hav.insert(x);
hav.insert(y);
To[x].insert({ y,dis });
To[y].insert({ x,dis });//To表路径
}
while (OrderNum--)
{
Endans = MaxNum*MaxNum;
ans = 0;
cin >> x >> y;
dfs(x, y);
cout << Endans << endl;
}
for (auto it = hav.begin(); it != hav.end(); it++)
{
To[*it].clear();
}
hav.clear();
}
}