201903CSPT5 317号子任务

题意：有n个点，m条无向边。n个点中有一些是行星发动机点，现在要求计算出每个点最近的k个行星发动机点的最短路径之和，如果某个点能够达到的行星发动机点不足k个，则求其能到达行星发动机点的最短路径之和。

#include<bits/stdc++.h>
using namespace std;
struct Node
{
	int w;
	int v;
	Node(int w,int v):v(v),w(w){}
};
struct node
{
	int u;
	int d;
	bool operator <(const node y) const
	{
		return d>y.d;
	}
};
vector <Node> V[10010];
vector <int> planet;
int dis[10010],vis[10010],qdis[10010];
int planetdis[10010][10010];
int n,m,k,ans=0;
void Dij(int s)
{
	for(int i=1;i<=n;i++)dis[i]=1999999999;
	dis[s]=0;
	priority_queue <node> q;
	node t;t.d=0;t.u=s;q.push(t);
	while(q.size()!=0)
	{
		int x=q.top().u;//cout<<x<<" "<<endl;
		q.pop();
		if(vis[x]==1)continue;
		vis[x]=1;
		for(int i=0;i<V[x].size();i++)
		{
			int y=V[x][i].v;int w=V[x][i].w;
			if(dis[y]>dis[x]+w)
			{
				dis[y]=dis[x]+w;
				node temp;temp.d=dis[y];temp.u=y;
				q.push(temp);
			}
		}
	}
}
int main()
{
	cin>>n>>m>>k;
	for(int i=1;i<=n;i++)
	{
		int id;
		cin>>id;
		if(id==1)planet.push_back(i);
	}
	for(int i=1;i<=m;i++)
	{
		int u,v,w;
		cin>>u>>v>>w;
		V[u].push_back(Node(w,v));
		V[v].push_back(Node(w,u));
	}
	for(int i=0;i<planet.size();i++)
	{
		memset(vis,0,sizeof(vis));
		Dij(planet[i]);
		for(int j=1;j<=n;j++)
		planetdis[i+1][j]=dis[j];
	}
	/*for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=n;j++)
		cout<<planetdis[i][j]<<" ";
		cout<<endl;
	}*/
	vector<int> q;
	for(int i=1;i<=n;i++)
	{
		ans=0;int num=0;
		memset(qdis,0,sizeof(qdis));
		for(int j=1;j<=planet.size();j++)
		if(planetdis[j][i]!=1999999999)qdis[++num]=planetdis[j][i];
		sort(qdis+1,qdis+num+1);      //for(int l=1;l<=num;l++)cout<<qdis[l]<<" ";
		if(num<=k)
		{
			for(int j=1;j<=num;j++)
			ans+=qdis[j];
		}
		else 
		{
			for(int j=1;j<=k;j++)
			ans+=qdis[j];	
		}
		cout<<ans<<endl;
	}
	return 0;
}