#1097 : 最小生成树一·Prim算法
时间限制:10000ms
单点时限:1000ms
内存限制:256MB
描述
最近,小Hi很喜欢玩的一款游戏模拟城市开放出了新Mod,在这个Mod中,玩家可以拥有不止一个城市了!
但是,问题也接踵而来——小Hi现在手上拥有N座城市,且已知这N座城市中任意两座城市之间建造道路所需要的费用,小Hi希望知道,最少花费多少就可以使得任意两座城市都可以通过所建造的道路互相到达(假设有A、B、C三座城市,只需要在AB之间和BC之间建造道路,那么AC之间也是可以通过这两条道路连通的)。
提示:不知道为什么Prim算法和Dijstra算法很像呢Σ(っ °Д °;)っ 。
输入
每个测试点(输入文件)有且仅有一组测试数据。
在一组测试数据中:
第1行为1个整数N,表示小Hi拥有的城市数量。
接下来的N行,为一个N*N的矩阵A,描述任意两座城市之间建造道路所需要的费用,其中第i行第j个数为Aij,表示第i座城市和第j座城市之间建造道路所需要的费用。
对于100%的数据,满足N<=10^3,对于任意i,满足Aii=0,对于任意i, j满足Aij=Aji, 0<Aij<10^4.
输出
对于每组测试数据,输出1个整数Ans,表示为了使任意两座城市都可以通过所建造的道路互相到达至少需要的建造费用。
样例输入
5
0 1005 6963 392 1182
1005 0 1599 4213 1451
6963 1599 0 9780 2789
392 4213 9780 0 5236
1182 1451 2789 5236 0
样例输出
4178
#include <bits/stdc++.h>
using namespace std;
#define fi first
#define se second
#define ff(i,a,b) for(int i = a; i <= b; i++)
#define f(i,a,b) for(int i = a; i < b; i++)
typedef pair<int,int> P;
#define ll long long
vector<P> v[100010];
int d[100010],vis[100010],flag[100010],dis[1010][1010];
int n;
int prime()
{
memset(d,0x3f,sizeof(d));
d[0] = 0;
int sum = 0;
f(i,0,n)
{
int mini = 0, minn = 1e9;
ff(j,0,n-1)
{
if(!vis[j] && d[j] < minn)
{
minn = d[j];mini = j;
}
}
vis[mini] = 1;
sum += minn;
ff(j,0,n-1)
{
if(!vis[j] && d[j] > dis[mini][j])
d[j] = dis[mini][j];
}
}
return sum;
}
int main()
{
ios::sync_with_stdio(false);
cin >> n;
ff(i,0,n - 1) ff(j,0,n - 1) cin >> dis[i][j];
cout << prime() << endl;
return 0;
}
优先队列:
#include <bits/stdc++.h>
using namespace std;
#define fi first
#define se second
#define ff(i,a,b) for(int i = a; i <= b; i++)
#define f(i,a,b) for(int i = a; i < b; i++)
typedef pair<int,int> P;
#define ll long long
vector<P> v[100010];
int d[100010],vis[100010],flag[100010];
priority_queue<P,vector<P>, greater<P> > Q;
int prime()
{
memset(d,0x3f,sizeof(d));
Q.push(make_pair(0,0));
vis[0] = 1;
d[0] = 0;
int sum = 0;
while(!Q.empty())
{
P top = Q.top();
Q.pop();
if(!vis[top.se]) sum += d[top.se];
vis[top.se] = 1;
f(i,0,v[top.se].size())
if(!vis[v[top.se][i].fi] && d[v[top.se][i].fi] > v[top.se][i].se)
{
d[v[top.se][i].fi] = v[top.se][i].se;
Q.push(make_pair(d[v[top.se][i].fi],v[top.se][i].fi));
}
}
return sum;
}
int main()
{
ios::sync_with_stdio(false);
int n,x;
cin >> n;
ff(i,0,n - 1) ff(j,0,n - 1) { cin >> x; if(x) v[i].push_back(make_pair(j,x)); }
cout << prime() << endl;
return 0;
}