2753: [SCOI2012]滑雪与时间胶囊

题目链接

题目大意：一个n个点m条边的带权无向图，每个点有一个高度值h。从1号点开始遍历，每次走的边u到v，必须满足h[u]>=h[v]。已知从当前点回到曾经遍历过的任意一个点是不需要走路的。求最多可以遍历多少个点？遍历这些点走的最小路程是多少？

题解：第一问bfs，第二问可以用朱刘算法，但数据太大过不了
Orz
Orz
感觉写题解就是把别人的抄一遍，干脆贴链接好了……
大概就是把相同高度的点分为一层，层内双向边，层与层间单向边，然后就可以kruskal了

我的收获：分层图好神啊，这种思想还是不错的

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define M 1000005
using namespace std;

int n,m,t,cnt;
long long ans;
int head[M],q[M],f[M],h[M];
bool vis[M];

struct node{int u,v,w;}p[M*2];
struct edge{int to,nex;}e[M*2];

void add(int u,int v,int w){p[t+1]=node{u,v,w};e[t].to=v,e[t].nex=head[u],head[u]=t++;}

int fid(int x){return f[x]==x?x:f[x]=fid(f[x]);}

bool cmp(node x,node y){return h[x.v]>h[y.v]||(h[x.v]==h[y.v]&&x.w<y.w);}

void bfs()
{
    int h=0,w=0;cnt=1;
    q[w++]=1;vis[1]=1;
    while(h<w)
    {
        int x=q[h++];
        for (int i=head[x];i!=-1;i=e[i].nex)
          if (!vis[e[i].to]) q[w++]=e[i].to,vis[e[i].to]=1,cnt++;
    }
}

void uniom(int x,int y,int v){
    int p=fid(x),q=fid(y);
    if(p==q) return ;
    f[p]=q,ans+=v;
}

void kruskal()
{
    for(int i=1;i<=n;i++) f[i]=i;
    for(int i=1;i<=t;i++) 
    if(vis[p[i].u]&&vis[p[i].v]) uniom(p[i].u,p[i].v,p[i].w);
}

void work()
{
    bfs();
    kruskal();
    printf("%d %lld\n",cnt,ans);
}

void init()
{
    int x,y,z;t=0;
    memset(head,-1,sizeof(head));
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&h[i]);
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&x,&y,&z);
        if(h[x]>=h[y]) add(x,y,z);
        if(h[y]>=h[x]) add(y,x,z);
    }
    for (int i=1;i<=n;i++) f[i]=i;
    sort(p+1,p+t+1,cmp);
}

int main()
{
    init();
    work();
    return 0;
}