CDOJ 1146 秋实大哥与连锁快餐店 Prim 最小生成树

嗯，题意就是说，平面坐标系上有n个点，它们可以形成一个完全图，距离就是两点的欧几里德距离

然后其中有一些点是旗舰店，然后现在想连一些边，使得所有的点都与至少一个旗舰店连通，然后问最小的边权和

显然就是最小生成树了

但是这是稠密图（其实我觉得即使稠密图，Kruskal的表现应该也很好，因为并查集部分的操作应该是近乎常数的吧），所以prim会更好

开始时用的Kruskal，直接MLE了，发现是内存限制比较小，然后也就放弃了堆优化 的prim，直接上普通的prim，然后就过了

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <string>
#include <fstream>
#include <list>
#include <stack>
#include <queue>
#include <deque>
#include <algorithm>
#include <map>
#include <set>
#include <vector>
using namespace std;
#define ll long long
#define INF 0x3f3f3f3f
#define maxn 6700
#define maxm 23000000
pair<ll, ll> node[maxn];
int n;
bool root[maxn];
ll dist[maxn];
double Prim()
{
double ans = 0;
bool flag = true;
ll tmp;
for (int i = 0; i < n; ++i)
{
if (!root[i])
continue;
if (flag)
{
for (int j = 0; j < n; ++j)
{
dist[j] = (node[i].first - node[j].first)*(node[i].first - node[j].first) + (node[i].second - node[j].second)*(node[i].second - node[j].second);
}
flag = false;
}
else
{
for (int j = 0; j < n; ++j)
{
tmp = (node[i].first - node[j].first)*(node[i].first - node[j].first) + (node[i].second - node[j].second)*(node[i].second - node[j].second);
if (tmp < dist[j])
dist[j] = tmp;
}
}
}
for (int i = 0; i < n; ++i)
{
int k = -1;
ll minv = 999999999999999;
for (int j = 0; j < n; ++j)
{
if (!root[j] && dist[j] < minv)
{
k = j; minv = dist[j];
}
}
if (k == -1)
break;
root[k] = true;
ans += sqrt(minv);
for (int j = 0; j < n; ++j)
{
if (!root[j])
{
tmp = (node[k].first - node[j].first)*(node[k].first - node[j].first) + (node[k].second - node[j].second)*(node[k].second - node[j].second);
if (tmp < dist[j])
{
dist[j] = tmp;
}
}
}
}
return ans;
}
int main()
{
//freopen("input.txt","r",stdin);
//freopen("output.txt","w",stdout);
//ios::sync_with_stdio(false);
//cin.tie(0); cout.tie(0);
//ifstream in;
//in.open("input.txt", ios::in);
scanf("%d", &n);
int tmp;
for (int i = 0; i < n; ++i)
{
scanf("%lld%lld%d", &node[i].first, &node[i].second, &tmp);
if (tmp > 0)
root[i] = true;
}
double ans;
ans = Prim();
printf("%.2f\n", ans);
//while (1);
return 0;
}