题目:
题目描述
长假过后,迷宫设计大师必须完成他的工作。旅游公司给他一张矩形的地图。地图由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)代表夫妻两人的位置坐标。
输入数据保证满足条件的只有一种迷宫(即答案唯一)。
输出格式:
对于每个问题,输出一行,其中一个整数表示两个给定方块之间最短路径的长度。
输入输出样例
输入样例#1:
3 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
输出样例#1:
4
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;
}