1. 程式人生 > >[洛谷 1967]貨車運輸---kruskal+lca(倍增)+遍歷 or kruskal重構樹+lca(倍增)

[洛谷 1967]貨車運輸---kruskal+lca(倍增)+遍歷 or kruskal重構樹+lca(倍增)

題目描述

A 國有 n 座城市,編號從 1 到 n,城市之間有 m 條雙向道路。每一條道路對車輛都有重量限制,簡稱限重。現在有 q 輛貨車在運輸貨物, 司機們想知道每輛車在不超過車輛限重的情況下,最多能運多重的貨物。
輸入輸出格式
輸入格式:

輸入檔名為 truck.in。

輸入檔案第一行有兩個用一個空格隔開的整數 n,m,表示 A 國有 n 座城市和 m 條道

路。 接下來 m 行每行 3 個整數 x、 y、 z,每兩個整數之間用一個空格隔開,表示從 x 號城市到 y 號城市有一條限重為 z 的道路。注意: x 不等於 y,兩座城市之間可能有多條道路 。

接下來一行有一個整數 q,表示有 q 輛貨車需要運貨。

接下來 q 行,每行兩個整數 x、y,之間用一個空格隔開,表示一輛貨車需要從 x 城市運輸貨物到 y 城市,注意: x 不等於 y 。

輸出格式:

輸出檔名為 truck.out。

輸出共有 q 行,每行一個整數,表示對於每一輛貨車,它的最大載重是多少。如果貨

車不能到達目的地,輸出-1。

輸入輸出樣例
輸入樣例#1:

4 3
1 2 4
2 3 3
3 1 1
3
1 3
1 4
1 3

輸出樣例#1:

3
-1
3

說明

對於 30%的資料,0 < n < 1,000,0 < m < 10,000,0 < q< 1,000;

對於 60%的資料,0 < n < 1,000,0 < m < 50,000,0 < q< 1,000;

對於 100%的資料,0 < n < 10,000,0 < m < 50,000,0 < q< 30,000,0 ≤ z ≤ 100,000。

分析

  易知,ans絕對在該圖的最大生成樹上.
  於是,果斷先kurskal將圖處理成樹,那麼給出x,y後只需要在x–>y的路徑上取權值最小即可(保證不超載).
  x–>y的路徑(保證x,y聯通):x–>lca(x,y)–>y {在知道fa的情況下,直接for就行了}
  然後..就沒有然後了(貌似有點暴力,然而資料就是這麼水)

  在翻題解的時候,偶然翻到某位dalao的kruskal重構樹,於是baidu了一下(貌似挺簡單的):
  –方法:按照kruskal求最小生成樹的方式加邊,但每次在加邊時,新建一個節點,然後把兩個聯通塊(其實是兩棵二叉樹)的根節點作為其左右兒子,把邊權賦值給新建節點
  –性質:
   1.是一棵二叉樹;
   2.滿足父節點的值大於等於兒子節點,是一個大頂堆,這是最關鍵的一點;
   3.原圖上任意兩點間路徑最長邊的最小值等於其lca的值;
  根據性質3,可解決圖中兩點路徑最長邊的最小值,即本題的所求
  
  So,與上文類似,除去遍歷步驟,改動一下kruskal即可

程式碼

法一: kruskal+lca(倍增)+遍歷
(貌似有點凌亂)

#include <cstdio>
#include <cstdlib>
#include <algorithm>
#define min_(x,y) x<y?x:y;
#define open(s) freopen(s".in","r",stdin); freopen(s".out","w",stdout);
#define close fclose(stdin); fclose(stdout); 
using namespace std;

int n,m,q;

struct Edge
{
    int to;
    int nxt;
    int w;
};
int cnt;
int head[10005];
Edge edge[20005];
inline void add(int x,int y,int z)//存樹
{
    edge[++cnt]=(Edge){y,head[x],z};
    head[x]=cnt;
}

struct Side//kruskal
{
    int x,y,w;
    bool operator < (const Side &b) const
    {
        return w>b.w;
    }
}side[50005];
int pa[10005];
int size[10005];
inline int find(int x)
{
    return pa[x]=(pa[x]==x)?x:find(pa[x]);
}
inline bool join(int x,int y)
{
    int x1=find(x),y1=find(y);
    if(x1==y1) return 0;
    if(size[x1]>size[y1])
    {
        size[x1]+=size[y1];
        pa[y1]=x1;
    }else
    {
        size[y1]+=size[x1];
        pa[x1]=y1;
    }
    return 1;
}
int dep[10005];//LCA
int w[10005];
int fa[10005][14];
int lg[10005];
inline void swap_(int &x,int &y)
{
    int tmp=x; x=y; y=tmp;
}
inline void get_lg()
{
    for(int i=1;i<=n;++i)
        lg[i]=lg[i-1]+(1<<(lg[i-1]+1)==i);
}
inline void get_dep(int u,int pre)
{
    for(int i=head[u],v;i;i=edge[i].nxt)
    if(edge[i].to!=pre)
    {
        v=edge[i].to;
        dep[v]=dep[u]+1;
        fa[v][0]=u;
        w[v]=edge[i].w;
        get_dep(v,u);
    }
}
inline void get_fa()
{
    for(int j=1;j<=lg[n];++j)
    for(int i=1;i<=n;++i)
    if(j<=lg[dep[i]])
        fa[i][j]=fa[ fa[i][j-1] ][j-1];
}
inline int get_lca(int x,int y)
{
    if(dep[x]<dep[y]) swap_(x,y);
    for(;dep[x]>dep[y];)
        x=fa[x][lg[dep[x]-dep[y]]];
    if(x==y) return x;
    for(int i=lg[dep[x]];i>=0;--i)
    if(fa[x][i]!=fa[y][i])
    {
        x=fa[x][i];
        y=fa[y][i];
    }
    return fa[x][0];
}

inline int read()
{
    int k=1;
    int sum=0;
    char c=getchar();
    for(;'0'>c || c>'9' ;c=getchar())
        if(c=='-') k=-1;
    for(;'0'<=c && c<='9';c=getchar())
        sum=sum*10+c-'0';
    return sum*k;
}

inline void write(int x)
{
    if(x<0) { putchar('-'); x*=-1; }
    if(x>9) write(x/10);
    putchar(x%10+'0');
}

inline void kruskal()
{
    sort(side+1,side+m+1);
    for(int i=1;i<=n;++i) pa[i]=i,size[i]=1;
    int T=n-1;
    for(int i=1,x,y;i<=m && T;++i)
    {
        x=side[i].x; y=side[i].y;
        if(join(x,y))
        {
            --T;
            add(x,y,side[i].w);
            add(y,x,side[i].w);
        }
    }
}

int main()
{
    open("1967");

    n=read();
    m=read();

    for(int i=1,x,y,z;i<=m;++i)
    {
        x=read(); y=read(); z=read();
        side[i]=(Side){x,y,z};
    }
    kruskal();
    get_lg();
    for(int i=1;i<=n;++i)
    if(!dep[i])
        get_dep(i,0);
    get_fa();

    q=read();
    for(int l,r,mid,ans;q;--q)
    {
        l=read(); r=read();
        if(find(l)!=find(r)) { printf("-1\n"); continue; }//不聯通直接處理
        mid=get_lca(l,r);
        ans=0x3f3f3f3f;
        for(;l!=mid;l=fa[l][0]) ans=min_(ans,w[l]);
        for(;r!=mid;r=fa[r][0]) ans=min_(ans,w[r]);//利用fa直接遍歷查詢
        printf("%d\n",ans);
    }

    close;
    return 0;
}

法二: ruskal重構樹+lca(倍增)
(貌似寫炸了,還沒上面的快...)

#include <cstdio>
#include <cstdlib>
#include <algorithm>
#define open(s) freopen(s".in","r",stdin); freopen(s".out","w",stdout);
#define close fclose(stdin); fclose(stdout); 
using namespace std;

int n,m,q;

int fa[20005][14];
int dep[20005];
int w[20005];
int lg[20005];

struct Side
{
    int x,y,w;
    bool operator < (const Side &b) const
    {
        return w>b.w;
    }
}side[50005];
int pa[20005];
inline int find(int x)
{
    return pa[x]=(pa[x]==x)?x:find(pa[x]);
}
inline bool join(int x,int y,int z)
{
    int x1=find(x),y1=find(y);
    if(x1==y1) return 0;
    ++n;  //新增節點
    fa[x1][0]=n; //x1,y1分別作為其左右兒子
    fa[y1][0]=n; 
    pa[x1]=pa[y1]=pa[n]=n; //由於連通了,更新pa
    w[n]=z;//x--y的邊權最為點權
    return 1;
}

inline void swap_(int &x,int &y)
{
    int tmp=x; x=y; y=tmp;
}
inline void get_lg()
{
    for(int i=1;i<=n;++i)
        lg[i]=lg[i-1]+(1<<(lg[i-1]+1)==i);
}
inline int get_dep(int u)
{
    if(!fa[u][0] || dep[u]) return dep[u];
    return dep[u]=get_dep(fa[u][0])+1;
}
inline void get_fa()
{
    for(int j=1;j<=lg[n];++j)
    for(int i=1;i<=n;++i)
    if(j<=lg[dep[i]])
        fa[i][j]=fa[ fa[i][j-1] ][j-1];
}
inline int get_lca(int x,int y)
{
    if(find(x)!=find(y)) return -1;
    if(dep[x]<dep[y]) swap_(x,y);
    for(;dep[x]>dep[y];)
        x=fa[x][lg[dep[x]-dep[y]]];
    if(x==y) return w[x];
    for(int i=lg[dep[x]];i>=0;--i)
    if(fa[x][i]!=fa[y][i])
    {
        x=fa[x][i];
        y=fa[y][i];
    }
    return w[fa[x][0]];
}

inline int read()
{
    int k=1;
    int sum=0;
    char c=getchar();
    for(;'0'>c || c>'9' ;c=getchar())
        if(c=='-') k=-1;
    for(;'0'<=c && c<='9';c=getchar())
        sum=sum*10+c-'0';
    return sum*k;
}

inline void write(int x)
{
    if(x<0) { putchar('-'); x*=-1; }
    if(x>9) write(x/10);
    putchar(x%10+'0');
}

inline void kruskal()
{
    sort(side+1,side+m+1);
    for(int i=1;i<=n;++i) pa[i]=i;
    int T=n-1;
    for(int i=1,x,y;i<=m && T;++i)
    {
        x=side[i].x; y=side[i].y;
        if(join(x,y,side[i].w))
            --T;
    }
}

int main()
{
    open("1967");

    n=read();
    m=read();

    for(int i=1,x,y,z;i<=m;++i)
    {
        x=read(); y=read(); z=read();
        side[i]=(Side){x,y,z};
    }
    kruskal();

    get_lg();
    for(int i=1;i<=n;++i)
    if(!dep[i])
        get_dep(i);
    get_fa();

    q=read();
    for(;q;--q)
    {
        write(get_lca(read(),read()));
        putchar('\n');
    }

    close;
    return 0;
}