Google Spaceship Defence 面试题目 并查集+最短路

点击打开链接

题意：给了n个地点，然后每个地点的名称，若名称相同则可以认为这两个地点距离为0，下面m个关系代表两个地点的距离，有向的边，q个询问，两个点的距离，想做的密码是nefu

思路：简单的最短路题目，因为有相同的点，可以用并查集将这些点放到一个集合中，然后建图建的就是父节点就行了，找相同关系我用的Trie树，但是map应该也可以直接做，没用过map只能写Trie树了，剩下的就是最短路模版，easy~~~

#include <queue>
#include <vector>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int inf=0x3f3f3f3f;
const ll INF=0x3f3f3f3f3f3f3f3fll;
const int maxn=400010;
int f[maxn],num[maxn];
struct edge{
    int to,cost;
    edge(){}
    edge(int a,int b){to=a;cost=b;}
};
typedef pair<int,int>P;
vector<edge>G[maxn];
int dis[maxn];
void dijkstra(int st){
    priority_queue<P,vector<P>,greater<P> >que;
    fill(dis,dis+maxn,inf);
    dis[st]=0;
    que.push(P(0,st));
    while(!que.empty()){
        P p=que.top();que.pop();
        int v=p.second;
        if(dis[v]<p.first) continue;
        for(unsigned int i=0;i<G[v].size();i++){
            edge e=G[v][i];
            if(dis[e.to]>dis[v]+e.cost){
                dis[e.to]=dis[v]+e.cost;
                que.push(P(dis[e.to],e.to));
            }
        }
    }
}
int find1(int x){
    if(x==f[x]) return f[x];
    int t=f[x];
    f[x]=find1(f[x]);
    return f[x];
}
void unite(int a,int b){
    int aa=find1(a);
    int bb=find1(b);
    if(aa==bb) return ;
    f[aa]=bb;
    num[bb]+=num[aa];
}
struct Trie{
    Trie *next[96];
    int v;
};
Trie *root;
char str1[30],str2[30];
void creatTrie(char *str,int idx){
    int len=strlen(str);
    Trie *p=root,*q;
    for(int i=0;i<len;i++){
        int id=str[i]-'0';
        if(p->next[id]==NULL){
            q=(Trie *)malloc(sizeof(Trie));
            q->v=0;
            for(int j=0;j<96;j++)
                q->next[j]=NULL;
            p->next[id]=q;
        }
        p=p->next[id];
    }
    p->v=idx;
}
int findtrie(char *str){
    int len=strlen(str);
    Trie *p=root;
    for(int i=0;i<len;i++){
        int id=str[i]-'0';
        p=p->next[id];
        if(p==NULL) return 0;
    }
    return p->v;
}
int dealtrie(Trie *T){
    if(T==NULL) return 0;
    for(int i=0;i<96;i++) if(T->next[i]!=NULL) dealtrie(T->next[i]);
    free(T);
    return 0;
}
int main(){
    int T,n,cas=1,m,u,v,cost,q;
    scanf("%d",&T);
    while(T--){
        scanf("%d",&n);
        for(int i=0;i<maxn;i++) G[i].clear();
        for(int i=0;i<=2*n;i++){
            f[i]=i;num[i]=1;
        }
        int k=1;
        root=(Trie *)malloc(sizeof(Trie));
        for(int i=0;i<96;i++) root->next[i]=NULL;
        for(int i=1;i<=n;i++){
            scanf("%s",str1);
            int ans1=findtrie(str1);
            if(ans1==0) creatTrie(str1,i);
            else unite(ans1,i);
        }
        scanf("%d",&m);
        for(int i=0;i<m;i++){
            scanf("%d%d%d",&u,&v,&cost);
            int tmp;
            int aa=find1(u),bb=find1(v);
            if(aa==bb) tmp=0;
            else tmp=cost;
            G[aa].push_back(edge(bb,tmp));
        }
        scanf("%d",&q);
        printf("Case #%d:\n",cas++);
        while(q--){
            scanf("%d%d",&u,&v);
            int aa=find1(u),bb=find1(v);
            if(aa==bb) printf("0\n");
            else{
                dijkstra(aa);
                if(dis[bb]==inf) printf("-1\n");
                else printf("%d\n",dis[bb]);
            }
        }
        dealtrie(root);
    }
    return 0;
}