题意:有一个n个节点和m条边的图,每个节点有一个高度,路只能从一个点到一个高度不大于它的另一个点,你可以从你所经过的任何一个点出发去下一个点,求经过的点最多为多少个,且此时经过的最短路为多少?
思路:prim算法中优先队列第一关键词以所到达的节点的高度降序,第二关键词以路程升序
代码:
#include<bits/stdc++.h> #define inf 1000000007 using namespace std; typedef long long ll; int n, m; int h[100005]; struct w { int to, cost; }w; vector<struct w> g[100005]; ll z=0; bool operator<(struct w a,struct w b) { if(h[a.to]!=h[b.to]) { return h[a.to]<h[b.to]; } return a.cost>b.cost; }; int vis[100005], zi=0; void dfs(int v) { memset(vis,0,sizeof(vis)); vis[1]=1; zi=1; priority_queue<struct w> que; for(int i=0;i<g[v].size();i++) { que.push(g[v][i]); } while(!que.empty()) { w=que.top(); que.pop(); if(vis[w.to]!=0) { continue; } z=z+w.cost; zi++; vis[w.to]=1; v=w.to; for(int i=0;i<g[v].size();i++) { if(vis[g[v][i].to]==0) { que.push(g[v][i]); } } } } int main() { scanf("%d%d",&n,&m); for(int i=1;i<=n;i++) { scanf("%d",&h[i]); } for(int i=0;i<m;i++) { int s, e, t; scanf("%d%d%d",&s,&e,&t); w.cost=t; if(h[s]>h[e]) { w.to=e; g[s].push_back(w); } else if(h[s]<h[e]) { w.to=s; g[e].push_back(w); } else { w.to=e; g[s].push_back(w); w.to=s; g[e].push_back(w); } } dfs(1); cout << zi << " " << z << endl; return 0; }