CCF CSP认证201903-5 317号子任务

201903-5 317号子任务

题目

思路

暴力混分，每个点来一次dijkstra最短路径算法。

代码如下（30分）

#include<cstdio>
#include<vector>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;
const int inf=1e9;
struct edge{
 	int to,w;
 	bool operator <(const edge&e)const{
  		return w>e.w;
 	}
};
vector<edge> g[10005];
int n,m,k,f[10005],d[10005];
bool vis[10005]={false};

int dijkstra(int x){
 	memset(vis,0,sizeof(vis));
 	fill(d,d+n+1,inf);
 	d[x]=0;
 	int ans=0,cnt=0;
 	priority_queue<edge> q;
 	q.push((edge){x,0});
 	while(!q.empty()){
  		edge now=q.top();
  		q.pop();
  		int u=now.to,w=now.w;
  		if(vis[u]) continue;
  		vis[u]=true;
  		if(f[u]){
   			cnt++;
   			ans+=w;
   			if(cnt==k) return ans;
  		}
  		for(int i=0;i<g[u].size();i++){
   			int v=g[u][i].to,l=g[u][i].w;
   			if(!vis[v]&&d[v]>d[u]+l){
    				d[v]=d[u]+l;
    				q.push((edge){v,d[v]});
   			}
  		}
 	}
 	return ans;
}

int main(){
 	scanf("%d%d%d",&n,&m,&k);
 	for(int i=1;i<=n;i++){
  		scanf("%d",&f[i]);
 	}
 	int u,v,w;
 	for(int i=0;i<m;i++){
  		scanf("%d%d%d",&u,&v,&w);
  		g[u].push_back((edge){v,w});
  		g[v].push_back((edge){u,w});
 	}
 	for(int i=1;i<=n;i++){
  		printf("%d\n",dijkstra(i));
 	}
 	return 0;
}