题意:有一个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;
}

京公网安备 11010502036488号