#include #include #include #include using MAP = std::unordered_map>>; int INF = 2100000000; void setup_map(MAP& map, int line_cnt,const int& node_cnt) { int start, goal, cost; while (line_cnt--) { std::cin >> start >> goal >> cost; map[start].push_back(std::make_pair(goal, cost)); map[goal].push_back(std::make_pair(start, cost)); } for (int i = 1; i < node_cnt + 1; i++) { std::sort(map[i].begin(), map[i].end(), [](const std::pair& a, const std::pair& b) {return a < b; }); } } int find_node_num_has_min_cost(const std::vector& costs, const std::vector& visit) { int min_cost = INF, min_idx = -1; for (int i = 1; i < costs.size(); i++) { if (!visit[i]) { if (costs[i] < min_cost) { min_cost = costs[i]; min_idx = i; } } } return min_idx; } std::vector get_dis_info(const int& target, const int& goal1, const int& goal2, const int& goal3, MAP& map, const int& node_cnt) { std::vector node_cost(node_cnt + 1, INF); std::vector visit(node_cnt + 1, false); node_cost[target] = 0; int now = target; bool find1 = false, find2 = false, find3 = false; if (target == goal1) find1 = true; if (target == goal2) find2 = true; if (target == goal3) find3 = true; while (!find1 || !find2 || !find3) { visit[now] = true; for (const std::pair& line_info : map[now]) { if (!visit[line_info.first]) { if (node_cost[line_info.first] > node_cost[now] + line_info.second) { node_cost[line_info.first] = node_cost[now] + line_info.second; } } } now = find_node_num_has_min_cost(node_cost, visit); if (now == -1) break; if (now == goal2) find2 = true; if (now == goal1) find1 = true; if (now == goal3) find3 = true; } std::vector res = {node_cost[goal1],node_cost[goal2],node_cost[goal3]}; return res; } int main(void) { std::ios::sync_with_stdio(0); std::cin.tie(0); int node_cnt, line_cnt; std::cin >> node_cnt >> line_cnt; if (line_cnt == 0) { std::cout << -1; return 0; } MAP node_map; setup_map(node_map, line_cnt,node_cnt); int must1, must2; std::cin >> must1 >> must2; std::vector m1_dis_info = get_dis_info(must1, 1, must2, node_cnt, node_map, node_cnt); std::vector m2_dis_info = get_dis_info(must2, 1, must1, node_cnt, node_map, node_cnt); if (m1_dis_info[1] == INF) std::cout << -1; else { if (m1_dis_info[0] != INF && m2_dis_info[2] != INF) std::cout << m1_dis_info[0] + m1_dis_info[1] + m2_dis_info[2]; else if (m2_dis_info[0] != INF && m1_dis_info[2] != INF) std::cout << m2_dis_info[0] + m2_dis_info[1] + m1_dis_info[2]; else std::cout << -1; } return 0; }

제출 #11240

제출 정보

제출 결과

제출 코드

찾은 반례

입력:

오답 출력:

정답 출력:

입력:

오답 출력:

정답 출력:

입력:

오답 출력:

정답 출력: