题目链接
题目大意:我看不懂题,看别人博客说题意就是所有S到A的连接的最小距离。
思路:最小生成树(我用的kruskal)+广搜;
本题坑点(太恶心了):①读完nm要用gets(),用getchar还不行。②开的数组还必须比给的范围大很多,要不给报WA(不报RE,无语了)③用G++提交TLE的代码和WA的代码可以在C++上AC(浪费好长时间)
具体做法:用广搜求出每个A到其他A和S的距离。然后套kruskal的板子就行了,注意POJ的那些坑就行了。
代码:
#include<iostream>
#include<cstdlib>
#include<sstream>
#include<cstdio>
#include<stack>
#include<cstdio>
#include<map>
#include<set>
#include<queue>
#include<cstring>
#include<cmath>
#include<vector>
#include<algorithm>
using namespace std;
typedef long long LL;
#define me(a,b) memset(a,b,sizeof(a))
#define inf 0x3f3f3f3f
const int N=2111122;
const int M=20005;
const int H = 750;//比给的范围要大才行。
int p[M],v[H][H],vis[H][H],stemp[H][H];
int f1[4] = {0,0,1,-1};
int f2[4] = {1,-1,0,0};
int v_1[H][H];
char s[H][H];
int sum,o,n,m,cond;
struct node {
int x,y;
};
struct node_1 {
int l,r,ans;
}a[M];
bool cmp(node_1 a,node_1 b) {
return a.ans < b.ans;
}
int find(int root) {
return root == p[root]?root:p[root] = find(p[root]);
}
void kruskal() {
int i,ct=0;
for(i = 1 ; i <= cond ; i++)p[i] = i;
sort(a+1,a+1+o,cmp);
//for(i = 1 ; i <= o ; i ++) cout << a[i].ans << " " << a[i].l << " " << a[i].r << endl;
for(i = 1 ; i <= o ; i++) {
int x = find(a[i].l);
int y = find(a[i].r);
if(x != y) {
p[x] = y;
ct ++;
sum += a[i].ans;
if(ct == cond-1)break;
}
}
}
queue<node> q;
void bfs(int l,int r) {
memset(v_1,0,sizeof(v_1));
//v[l][r] = 1;
v_1[l][r] = 1;
int fx = l;
int fy = r;
int i;
//memset(v,0,sizeof(v));
me(stemp,0);
while(!q.empty())
q.pop();
node t;
t.x = l;
t.y = r;
q.push(t);
while(!q.empty()) {
node g = q.front();
q.pop();
for(i = 0 ; i < 4 ; i ++) {
int dx = g.x + f1[i];
int dy = g.y + f2[i];
if(dx>=0&&dx<m&&dy>=0&&dy<n) {
if(s[dx][dy] != '#'&&v_1[dx][dy]==0) {
v_1[dx][dy] = 1;
t.x = dx;
t.y = dy;
stemp[dx][dy] = stemp[g.x][g.y]+1;
q.push(t);
if(s[dx][dy]!=' ') {
//if(!v[dx][dy]) {
a[++o].l = vis[dx][dy];
a[o].r = vis[fx][fy];
a[o].ans = stemp[dx][dy];
//v[dx][dy] = 1;
//}
}
}
}
}
}
}
int main() {
int tt,i,j;
char ch[13];
scanf("%d",&tt);
while(tt--) {
scanf("%d%d",&n,&m);
gets(ch);//为什么getchar不行,为什么为什么?求大佬解释。
cond = 0;
sum=0;
o=0;
for(i = 0 ; i < m ; i++) gets(s[i]);
for(i = 0 ; i < m ; i++) {
for( j = 0 ; j < n; j++) {
if(s[i][j] == 'S'||s[i][j] == 'A') {
vis[i][j] = ++cond;
}
}
}
if(cond <= 1) {//特判了一下,没啥diao用。
cout << 0 << endl;
return 0;
}
for(i = 0 ; i < m ; i++) {
for( j = 0 ; j < n; j++) {
if(s[i][j] == 'S'||s[i][j] == 'A') {
bfs(i,j);
}
}
}
kruskal();
printf("%d\n",sum);
}
}