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

原创 2016年06月02日 10:40:16

点击打开链接

题意:给了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;
}

版权声明:本文为博主原创文章,未经博主允许不得转载。

相关文章推荐

csu 1307 City Tour dijkstra+并查集 最短路 解题报告

Description Alice想要从城市A出发到城市B,由于Alice最近比较穷(不像集训队陈兴老师是个rich second),所以只能选择做火车从A到B。不过Alice很讨厌坐火车,火车上...

hdu5361(最短路+并查集)

链接:点击打开链接 题意:有n个点排成一排,编号从1到n,每个点可以到自己[l, r]距离的位置,花费c,问一号店到其他点的最小花费 代码:#pragma comment(linker, "/ST...

Google中国2014校园招聘笔试Round A China New Grad Test Problem E. Spaceship Defence

Problem The enemy has invaded your spaceship, and only superior tactics will allow you to defend it!...

Codeforces 400D. Dima and Bacteria【并查集+最短路】

D. Dima and Bacteria time limit per test 2 seconds memory limit per test 256 megabytes in...

USACO section 2.4 Cow Tours(并查集+最短路)

Cow Tours Farmer John has a number of pastures on his farm. Cow paths connect some pastures with ...

最短路 并查集/利用边的性质求最短路

题意:N个城市,标号从0到N-1,M条道路,第K条道路(K从0开始)的长度为2^K,求编号为0的城市到其他城市的最短距离 输入:第一行两个正整数N(2 接下来M行两个整数,表示相连的两个城市的编号...
  • Tsaid
  • Tsaid
  • 2012年04月22日 15:49
  • 4944

find the most comfortable road(并查集,最短路)

Problem Description XX星有许多城市,城市之间通过一种奇怪的高速公路SARS(Super Air Roam Structure---超级空中漫游结构)进行交流,每条SARS都对行...

[Usaco2009 Jan]安全路经Travel(最短路树+并查集/树链剖分)

一道数据结构与图论的综合的好题,有着两种解法,可以在这里提交:http://218.28.19.228/cogs/problem/problem.php?pid=279 【题解】 由于1到其他点的...
  • cjk_cjk
  • cjk_cjk
  • 2015年02月07日 01:03
  • 414

并查集代码

  • 2015年12月12日 20:18
  • 1KB
  • 下载

ACM并查集讲解的深化和扩展

  • 2016年07月01日 17:27
  • 1.58MB
  • 下载
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:Google Spaceship Defence 面试题目 并查集+最短路
举报原因:
原因补充:

(最多只允许输入30个字)