题意:给你N个城市的坐标,城市之间存在公路,但是由于其中一些道路损坏了,需要维修,维修的费用与公路长成正比(公路是直的)。但现有M条公路是完整的,不需要维修,下面有M行,表示不需要维修的道路两端的城市,问最短费用。
直接做?哇无限wa。。。。。。
最后也不知道为啥,改了改数组大小啊,输入输出啊,什么的玄学优化就a了。。。。。。
然后一样的代码在c++会t。。。。。。
链接:poj 1751
#include <iostream>
#include <cstring>
#include <string>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstdlib>
#include <limits>
#include <vector>
#include <sstream>
#define ll long long
using namespace std;
const int maxn = 2200 + 10;
const int maxm = 1001000;
const int inf = 0x3f3f3f3f;
int fa[maxn];
int rnk[maxn];
int num[maxn];
int n, m;
struct node {
int a, b;
int c;
friend bool operator < (node x, node y) {
return x.c < y.c;
}
}q[maxm];
void init() {
for(int i = 0; i <= n; i++) {
fa[i] = i;
rnk[i] = 0;
}
}
int getf(int x) {
if(x != fa[x]) {
fa[x] = getf(fa[x]);
}
return fa[x];
}
void unions(int x, int y) {
x = getf(x);
y = getf(y);
if(x == y) return;
if(rnk[x] < rnk[y]) {
fa[x] = y;
}
else {
fa[y] = x;
if(rnk[x] == rnk[y]) rnk[x]++;
}
}
bool same(int x, int y) {
return getf(x) == getf(y);
}
void kruskal() {
sort(q + 1, q + 1 + m);
int ans = 0;
int nedge = 0;
for(int i = 1; i <= m && nedge != n - 1; i++) {
if(getf(q[i].a) != getf(q[i].b)) {
unions(q[i].a, q[i].b);
if(q[i].c != 0) {
printf("%d %d\n", q[i].a, q[i].b);
}
nedge++;
}
}
//if(nedge < n - 1) ans = -1;
//return ans;
}
int x[maxn], y[maxn];
int dis(int i, int j) {
return (x[i] - x[j]) * (x[i] - x[j]) + (y[i] - y[j]) * (y[i] - y[j]);
}
int main()
{
int T, kcase = 0;
while(~scanf("%d", &n)) {
m = 0;
for(int i = 1; i <= n; i++) {
scanf("%d %d", &x[i], &y[i]);
for(int j = 1; j < i; j++) {
int res = dis(i, j);
q[++m].a = i;
q[m].b = j;
q[m].c = res;
}
}
int k;
scanf("%d", &k);
int a, b;
init();
for(int i = 1; i <= k; i++) {
scanf("%d %d", &a, &b);
unions(a, b);
}
kruskal();
}
return 0;
}