G. Rudolf and Subway

黑屿白發表於2024-04-06

原題連結

題解

太巧妙了!!
原題等效於該分層圖,然後廣搜
本題中我用了另一種方法建邊,因為清空太麻煩了

code

#include<bits/stdc++.h>
using namespace std;
int main()
{
    ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
    int t;
    cin>>t;
    while(t--)
    {
        int n,m;
        cin>>n>>m;
        map<int,vector<int> > G;
        map<int,int> dis;
        for(int i=1;i<=m;i++)
        {
            int x,y,w;
            cin>>x>>y>>w;
            w+=n;
            G[x].emplace_back(w);
            G[y].emplace_back(w);
            G[w].emplace_back(x);
            G[w].emplace_back(y);
        }

        int st,ed;
        cin>>st>>ed;
        queue<int> q;
        q.emplace(st);
        dis[st]=1;
        while(q.size())
        {
            int now=q.front();
            q.pop();
            //printf("dis[%d]=%d\n",now,dis[now]);
            if(now==ed)break;
            for(auto next:G[now])
            {
                if(!dis[next])
                {
                    dis[next]=dis[now]+1;
                    q.emplace(next);
                }
            }
        }
        cout<<dis[ed]/2<<endl;
    }
    return 0;
}