1. 程式人生 > >牛客訓練賽(最短路)

牛客訓練賽(最短路)

這個是求關鍵點到其他關鍵點的最小距離。。感覺比較經典??然而做法比較沙雕。。

很容易想到從超級大源點向關鍵點連邊然後跑dij,可是會出現就是關鍵點的最小距離會連他自己算在內,只要解決這個問題這題就完了。。

因此在求最短路的時候可以再增加一個標號,即這個最短路是從哪個關鍵點得到的。。然後對同個關鍵點來的直接按照普通最短路做就可以。。然而我們只需要排除標號是自己本身的情況,因此在幾個標號的最短路中,我們只需要找最短和次短的即可。。

所以在跑dij轉移的時候存二元組,然後標號相同的合併,不同的取最小,然後做個小分類討論就行了。。

感覺很傻逼。。為什麼不來做這道題。。。

/**
 *        ┏┓    ┏┓
 *        ┏┛┗━━━━━━━┛┗━━━┓
 *        ┃       ┃  
 *        ┃   ━    ┃
 *        ┃ >   < ┃
 *        ┃       ┃
 *        ┃... ⌒ ...  ┃
 *        ┃       ┃
 *        ┗━┓   ┏━┛
 *          ┃   ┃ Code is far away from bug with the animal protecting          
 *          ┃   ┃   神獸保佑,程式碼無bug
 *          ┃   ┃           
 *          ┃   ┃        
 *          ┃   ┃
 *          ┃   ┃           
 *          ┃   ┗━━━┓
 *          ┃       ┣┓
 *          ┃       ┏┛
 *          ┗┓┓┏━┳┓┏┛
 *           ┃┫┫ ┃┫┫
 *           ┗┻┛ ┗┻┛
 */ 
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<queue>
#include<cmath>
#include<map>
#include<stack>
#include<set>
#include<bitset>
#include<stdlib.h>
#include<assert.h>
#define inc(i,l,r) for(int i=l;i<=r;i++)
#define dec(i,l,r) for(int i=l;i>=r;i--)
#define link(x) for(edge *j=h[x];j;j=j->next)
#define mem(a) memset(a,0,sizeof(a))
#define ll long long
#define eps 1e-8
#define succ(x) (1<<x)
#define lowbit(x) (x&(-x))
#define sqr(x) ((x)*(x))
#define mid (x+y>>1)
#define NM 200005 
#define nm 400005
#define pi 3.1415926535897931
const ll inf=1e9+7;
using namespace std;
ll read(){
    ll x=0,f=1;char ch=getchar() ;
    while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
    while(isdigit(ch))x=x*10+ch-'0',ch=getchar();
    return f*x;
}



struct edge{int t,v;edge*next;}e[nm],*h[NM],*o=e;
void add(int x,int y,int v){o->t=y;o->v=v;o->next=h[x];h[x]=o++;}
int n,m,p,_x,_y,_t,a[NM];
struct tmp{
    int x;ll d;
    bool operator<(const tmp&o)const{return d>o.d;}
}d[NM][3];
priority_queue<tmp>q;



void dij(){
    while(!q.empty()){
	int t=q.top().x;q.pop();
	inc(k,1,2)if(d[t][k].x)
	    link(t){
		if(d[j->t][1].x==d[t][k].x){
		    if(d[j->t][1].d>d[t][k].d+j->v)
			q.push(tmp{j->t,d[j->t][1].d=d[t][k].d+j->v});
		}else if(d[j->t][2].x==d[t][k].x){
		    if(d[j->t][2].d>d[t][k].d+j->v){
			q.push(tmp{j->t,d[j->t][2].d=d[t][k].d+j->v});
			if(d[j->t][1].d>d[j->t][2].d)swap(d[j->t][1],d[j->t][2]);
		    }
		}else if(!d[j->t][1].x||d[j->t][1].d>d[t][k].d+j->v){
		    d[j->t][2]=d[j->t][1];d[j->t][1]=d[t][k];
		    q.push(tmp{j->t,d[j->t][1].d+=j->v});
		}else if(!d[j->t][2].x||d[j->t][2].d>d[t][k].d+j->v){
		    d[j->t][2]=d[t][k];
		    q.push(tmp{j->t,d[j->t][2].d+=j->v});
		}
	    }
    }
}


int main(){
    n=read();m=read();p=read();
    inc(i,1,p){
	a[i]=read();d[a[i]][1]=tmp{a[i],0};q.push(tmp{a[i],0});
    }
    inc(i,1,m){_x=read();_y=read();_t=read();add(_x,_y,_t);add(_y,_x,_t);}
    dij();
    //inc(i,1,n){printf("%d %d::%d %d\n",d[i][1].x,d[i][1].d,d[i][2].x,d[i][2].d);}
    inc(i,1,p)if(d[a[i]][1].x!=a[i])printf("%lld ",d[a[i]][2].d);else printf("%lld ",d[a[i]][2].d);
    return 0*printf("\n");
}

Metropolis

時間限制:C/C++ 1秒,其他語言2秒 空間限制:C/C++ 524288K,其他語言1048576K 64bit IO Format: %lld

題目描述

魔方國有n座城市,編號為。城市之間通過n-1條無向道路連線,形成一個樹形結構。 在若干年之後,其中p座城市發展成了大都會,道路的數量也增加到了m條。 大都會之間經常有貿易往來,因此,對於每座大都會,請你求出它到離它最近的其它大都會的距離。

輸入描述:

第一行三個整數n,m,p (1 ≤ n,m ≤ 2*105,2 ≤ p ≤ n),第二行p個整數表示大都會的編號 (1≤ xi≤ n)。接下來m行每行三個整數ai,bi,li表示一條連線ai和bi,長度為li的道路 (1 ≤ ai,bi ≤ n,1 ≤ li ≤ 109)。
保證圖是連通的。

輸出描述:

輸出一行p個整數,第i個整數表示xi的答案。

示例1

輸入

複製

5 6 3
2 4 5
1 2 4
1 3 1
1 4 1
1 5 4
2 3 1
3 4 3

輸出

複製

3 3 5