【SCOI2012】【BZOJ2753】滑雪与时间胶囊

2753: [SCOI2012]滑雪与时间胶囊
Time Limit: 50 Sec Memory Limit: 128 MB
Submit: 1524 Solved: 536
[Submit][Status][Discuss]
Description
a180285非常喜欢滑雪。他来到一座雪山，这里分布着M条供滑行的轨道和N个轨道之间的交点（同时也是景点），而且每个景点都有一编号i（1<=i<=N）和一高度Hi。a180285能从景点i 滑到景点j 当且仅当存在一条i 和j 之间的边，且i 的高度不小于j。 与其他滑雪爱好者不同，a180285喜欢用最短的滑行路径去访问尽量多的景点。如果仅仅访问一条路径上的景点，他会觉得数量太少。于是a180285拿出了他随身携带的时间胶囊。这是一种很神奇的药物，吃下之后可以立即回到上个经过的景点（不用移动也不被认为是a180285 滑行的距离）。请注意，这种神奇的药物是可以连续食用的，即能够回到较长时间之前到过的景点（比如上上个经过的景点和上上上个经过的景点）。 现在，a180285站在1号景点望着山下的目标，心潮澎湃。他十分想知道在不考虑时间
胶囊消耗的情况下，以最短滑行距离滑到尽量多的景点的方案（即满足经过景点数最大的前提下使得滑行总距离最小）。你能帮他求出最短距离和景点数吗？
Input
输入的第一行是两个整数N，M。
接下来1行有N个整数Hi，分别表示每个景点的高度。
接下来M行，表示各个景点之间轨道分布的情况。每行3个整数，Ui，Vi，Ki。表示
编号为Ui的景点和编号为Vi的景点之间有一条长度为Ki的轨道。
Output

输出一行，表示a180285最多能到达多少个景点，以及此时最短的滑行距离总和。
Sample Input

3 3

3 2 1

1 2 1

2 3 1

1 3 10

Sample Output
3 2

HINT

【数据范围】

对于30%的数据，保证 1<=N<=2000

对于100%的数据，保证 1<=N<=100000

对于所有的数据，保证 1<=M<=1000000，1<=Hi<=1000000000，1<=Ki<=1000000000。

Source

没想到2012年四川还在出这种水题…
然而就是水题也坑了我好多次= =
各种冰茶几写残QAQ
妈妈我还是剁手吧
就是个最小生成树= =
只不过填进去的边是有向边(因为有高度的问题)

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define MAXN 100010
#define MAXM 2000010
using namespace std;
int n,m;
int h[MAXN];
int tp;
int u,v,w;
int f[MAXN];
bool vis[MAXN];
int queue[MAXN],head,tail;
long long sum;
struct Edge
{
    int u,v,w;
    Edge *next;
    bool operator <(const Edge& a)const{
        if (h[v]!=h[a.v]) return h[v]>h[a.v];
        return w<a.w;
    }
}e[MAXM],*prev[MAXN];
void insert(int u,int v,int w)
{
    e[++tp].v=v;e[tp].u=u;e[tp].next=prev[u];prev[u]=&e[tp];e[tp].w=w;
}
int find(int x)
{
    if (f[x]!=x) return f[x]=find(f[x]);
    return f[x];
}
int Union(int x,int y)
{
    int a=find(x),b=find(y);
    if (a!=b) 
    {
        f[b]=a;
        return 1;
    }
    return 0;
}
int main()
{
    scanf("%d%d",&n,&m);
    memset(vis,0,sizeof(vis));
    for (int i=1;i<=n;i++)  scanf("%d",&h[i]);
    for (int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&u,&v,&w);
        if (h[u]<=h[v]) insert(v,u,w);
        if (h[v]<=h[u]) insert(u,v,w);
    }
    for (int i=1;i<=n;i++)  f[i]=i;
    queue[tail++]=1;
    while (head<tail)
    {
        int x=queue[head++];
        vis[x]=1;
        for (Edge *i=prev[x];i;i=i->next)
            if (!vis[i->v]) queue[tail++]=i->v,vis[i->v]=1;
    }
    sort(e+1,e+tp+1);
    for (int i=1;i<=tp;i++)
        if (vis[e[i].u]&&vis[e[i].v]&&Union(e[i].u,e[i].v)) sum+=e[i].w;
    printf("%d %lld\n",tail,sum);
}