1. 程式人生 > >2018/11/02 膜你賽 設計迷宮

2018/11/02 膜你賽 設計迷宮

題目:

題目描述
長假過後,迷宮設計大師必須完成他的工作。旅遊公司給他一張矩形的地圖。地圖由N*M個小方塊組成。旅遊公司將把一對夫婦放在迷宮中的兩個不同的小方塊上,讓他們互相尋找。
原本的地圖上是空曠曠的一片,大師要做的就是在一些小方塊之間建造一些牆從而把地圖變成迷宮。
建造的迷宮有一個條件,無論這對夫妻出現在何處,他們之間只有且僅有一條路。 
這對他來說並不困難,但他是一個善解人意的人。
他也知道在兩個相鄰廣場之間建造每面牆的成本是不同的。
所以他想設計一個迷宮,讓旅遊公司花費最少的錢來建造它並滿足上訴條件。
建成迷宮後,你得到了 Q 個詢問包含夫妻兩人將被放置的位置資訊。
你需要弄 清楚它們之間最短路徑的長度(一個小方格為一個單位)。

輸入輸出格式
輸入格式:
輸入的第一行包含兩個整數 N和M。
 接下來輸入的 N×M 行給出迷宮中每個小方塊的資訊,它們的資訊按順序輸入, 輸入順序為 (
1,1),(1,2),(1,3)(1,m)(2,1)(2,2)(2,3)(3,1)……(n,m) 每行包含兩個字元 D 和 R 和兩個整數,D 後面的整數代表建造該小方格下邊的牆的花費,R 後面的整數代表建造該小方格右邊的牆的花費。 如果邊是邊界,缺少的路徑將被替換為 X 0。 下一行包含一個整數 Q 表示有Q個詢問。 接下來q 行,每一行兩個整數對(x1,y1)(x2,y2)代表夫妻兩人的位置座標。 輸入資料保證滿足條件的只有一種迷宮(即答案唯一)。 輸出格式: 對於每個問題,輸出一行,其中一個整數表示兩個給定方塊之間最短路徑的長度。 輸入輸出樣例 輸入樣例#13 3 D 1
R 9 D 7 R 8 D 4 X 0 D 2 R 6 D 12 R 5 D 3 X 0 X 0 R 10 X 0 R 11 X 0 X 0 3 1 1 3 3 1 2 3 2 2 2 3 1 輸出樣例#14 2 2 說明 對於30%的資料: 0<=N,M<=5 對於100%的資料:0<=N,M<=500

思路:
最大生成樹+LCA。
把加邊改為刪邊,然後跑最小生成樹求出路徑。
再在路徑上跑lca求最短距離。

注意:n,m不要弄反,邊的最大值為500000。

程式碼:

#include<bits/stdc++.h>
using namespace std;

#define
read(x) scanf("%d",&x)
#define maxn 500000 struct Edge{ int x,y,z; Edge(){} Edge(int xx,int yy,int zz) { x=xx,y=yy,z=zz; } bool operator < (const Edge& oth) const { return z<oth.z; } }; int n,m,cc=0; Edge e[maxn+5]; int getid(int x,int y) { return (x-1)*m+y; } void readin() { read(n),read(m); for(int i=1; i<=n; i++) { for(int j=1; j<=m; j++) { for(int k=1; k<=2; k++) { char x; int y; while((~scanf("%c",&x))&&!isalpha(x)); read(y); if(x=='X') continue; if(x=='D') e[++cc]=Edge(getid(i,j),getid(i+1,j),y); if(x=='R') e[++cc]=Edge(getid(i,j),getid(i,j+1),y); } } } } vector<int> tr[maxn+5]; int fa[maxn+5]; int find(int x) { if(fa[x]) return fa[x]=find(fa[x]); return x; } void kruskal() { int s=0; for(int i=cc;i>=1;i--) { if(s==n*m-1) return ; int fa1=find(e[i].x),fa2=find(e[i].y); if(fa1==fa2) continue; fa[fa1]=fa2; s++; tr[e[i].x].push_back(e[i].y); tr[e[i].y].push_back(e[i].x); } return ; } int d[maxn+5],anc[maxn+5][30]; void dfs(int x,int fa) { d[x]=d[fa]+1; anc[x][0]=fa; for(int i=1;i<=25;i++) { anc[x][i]=anc[anc[x][i-1]][i-1]; } for(int i=0;i<tr[x].size();i++) { int y=tr[x][i]; if(y==fa) continue; dfs(y,x); } } int findLCA(int x,int y) { if(d[x]<d[y]) swap(x,y); for(int i=25;i>=0;i--) { if(d[anc[x][i]]>=d[y]) x=anc[x][i]; } if(x==y) return x; for(int i=25;i>=0;i--) { if(anc[x][i]!=anc[y][i]) x=anc[x][i],y=anc[y][i]; } return anc[x][0]; } int main() { readin(); sort(e+1,e+cc+1); kruskal(); dfs(1,0); int Q; read(Q); while(Q--) { int x1,y1,x2,y2; read(x1),read(y1),read(x2),read(y2); int x=getid(x1,y1),y=getid(x2,y2); int lca=findLCA(x,y); printf("%d\n",d[x]+d[y]-d[lca]*2); } return 0; }