题目
题目
一个无向图,各点到点1的最短距离为di,保证满足条件删除m-k条边之后使得到点1的距离仍为di的点数量最多的情况下,输出剩余的k条边的编号(输入顺序即编号)
代码
#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<cctype>
#include<ctime>
#include<iostream>
#include<string>
#include<map>
#include<queue>
#include<stack>
#include<set>
#include<vector>
#include<iomanip>
#include<list>
#include<bitset>
#include<sstream>
#include<fstream>
#include<complex>
#include<algorithm>
#if __cplusplus >= 201103L
#include <unordered_map>
#include <unordered_set>
#endif
#define ll long long
#define int long long
using namespace std;
const int INF = 0x3f3f3f3f3f3f3f3f;
const int N=300010;
struct sut{
int to,next,val;
}edge[N<<1];
int head[N],dir[N],num[N];
bool vis[N];
int tot=1;int n,m,k;
void add_edge(int u,int v,int w){
edge[++tot].to=v;
edge[tot].val=w;
edge[tot].next=head[u];
head[u]=tot;
}
vector<int> v1;
void dij(int x){
for(int i=1;i<=n;i++){
dir[i]=INF;
vis[i]=0;
}
priority_queue<pair<int,int>> q;
dir[x]=0;
q.push(make_pair(-dir[x],x));
while(!q.empty()&&v1.size()<k){
int u=q.top().second;
q.pop();
if(vis[u]) continue;
if(u!=1) v1.push_back(num[u]/2);
vis[u]=1;
for(int i=head[u];i;i=edge[i].next){
int v=edge[i].to;
if(dir[v]>dir[u]+edge[i].val){
dir[v]=dir[u]+edge[i].val;
num[v]=i;
q.push(make_pair(-dir[v],v));
}
}
}
}
signed main(){
ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
cin>>n>>m>>k;
for(int i=1;i<=m;i++){
int u,v,w;
cin>>u>>v>>w;
add_edge(u,v,w);
add_edge(v,u,w);
}
dij(1);
cout<<min(n-1,k)<<endl;
for(int i=0;i<v1.size();i++) cout<<v1[i]<<" ";
return 0;
}