码迷,mamicode.com
首页 > 其他好文 > 详细

爬爬爬山

时间:2019-04-22 23:05:04      阅读:268      评论:0      收藏:0      [点我收藏+]

标签:ref   class   fine   har   ems   contest   opera   back   end   

传送门:

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define re register
#define pb push_back
#define mp make_pair
#define fi first
#define se second
const int N=1e6;
const int P=1e9+7;
void read(int &a)
{
    a=0;
    int d=1;
    char ch;
    while(ch=getchar(),ch>9||ch<0)
        if(ch==-)
            d=-1;
    a=ch-0;
    while(ch=getchar(),ch>=0&&ch<=9)
        a=a*10+ch-0;
    a*=d;
}
void write(int x)
{
    if(x<0)
        putchar(45),x=-x;
    if(x>9)
        write(x/10);
    putchar(x%10+0);
}
int head[N];
struct note
{
    int v,dis,next;
}edge[N];
int num;
void add(int u,int v,int dis)
{
    edge[++num].next=head[u];
    edge[num].v=v;
    edge[num].dis=dis;
    head[u]=num;
}
struct node
{
    ll cost,now,nowh,pos;
    bool operator < (const node & x) const {return x.cost<cost;}
};
priority_queue <node> q;
ll dis[N];
int h[N];
inline void dijstra()
{
    memset(dis,0x3f,sizeof(dis));
    dis[1]=0;
    while(!q.empty())
    {
        node p=q.top();
        q.pop();
        int u=p.pos;
        if(dis[u]<p.cost)
            continue;
        for(re int i=head[u];i;i=edge[i].next)
        {
            int v=edge[i].v;
            int w=edge[i].dis;
            ll co=min(0ll,p.now-(h[v]-p.nowh));
            if(dis[v]>dis[u]+w+co*co)
            {
                dis[v]=dis[u]+w+co*co;
                q.push({dis[v],max(0ll,p.now-(h[v]-p.nowh)),h[v]+co,v});
            }
        }
    }
}
int main()
{
    int n,m,k;
    read(n);
    read(m);
    read(k);
    for(re int i=1;i<=n;i++)
        read(h[i]);
    for(re int i=1;i<=m;i++)
    {
        int u,v,w;
        read(u);
        read(v);
        read(w);
        add(u,v,w);
        add(v,u,w);
    }
    q.push({0,k,h[1],1});
    dijstra();
    cout<<dis[n]<<endl;
    return 0;
}

 

爬爬爬山

标签:ref   class   fine   har   ems   contest   opera   back   end   

原文地址:https://www.cnblogs.com/acm1ruoji/p/10753466.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!