WannaflyCamp Day1 F:爬爬爬山

题目描述
爬山是wls最喜欢的活动之一。

在一个神奇的世界里，一共有n座山，m条路。

wls初始有k点体力，在爬山的过程中，他所处的海拔每上升1m，体力会减1点，海拔每下降1m，体力会加一点。

现在wls想从1号山走到n号山，在这个过程中，他的体力不能低于00，所以他可以事先花费一些费用请dls把某些山降低，将一座山降低l米需要花费l∗l的代价，且每座山的高度只能降低一次。因为wls现在就在11号山上，所以这座山的高度不可降低。

wls从1号山到n号山的总代价为降低山的高度的总代价加上他走过的路的总长度。

wls想知道最小的代价是多少。

输入描述
第一行三个整数n，m，k。

接下来一行n个整数，第i个整数hi
​
表示第i座山的高度。

接下来m行，每行三个整数x，y，z表示xy之间有一条长度为z的双向道路。

经过每条路时海拔是一个逐步上升或下降的过程。

数据保证1号山和n号山联通。

输出描述
一行一个整数表示答案。

样例输入 1
4 5 1
1 2 3 4
1 2 1
1 3 1
1 4 100
2 4 1
3 4 1
样例输出 1
6
题解：
初始有k点体力，每升高一米损失一点体力，即最高只能爬到初始状态高度h[1]+k米，若想爬到高于h[1]+k米的山，需要将山高度降到h[1]+k米。将图分为两组，一组为需要降高度的山，并把需要降高度的花费房到边权上，然后dijkstra。

#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<string>
#include<math.h>
#include<cmath>
#include<time.h>
#include<map>
#include<set>
#include<vector>
#include<queue>
#include<algorithm>
#include<numeric>
#include<stack>
#include<bitset>
//#include<unordered_map>
const int inf = 0x3f3f3f3f;
const double EI = 2.71828182845904523536028747135266249775724709369995957496696762772407663035354594571382178525166427;
const double PI = 3.141592653589793238462643383279;
const int maxn=100010;
using namespace std;
typedef long long ll;
int n,m;
ll k;
int head[maxn],tot;
ll dis[maxn];

struct edge{
    int v;
    ll w;
    int Next;
}a[maxn<<2];
bool vis[maxn];
ll h[maxn];
void init(){
    tot=0,memset(head,-1,sizeof(head));
}
void addv(int u,int v,ll w)
{
    a[++tot].v=v;
    a[tot].w=w;
    a[tot].Next=head[u];
    head[u]=tot;
}
priority_queue< pair<ll,int>>q;
void dijkstra(){
    memset(vis,false,sizeof(vis));
    memset(dis,0x3f,sizeof(dis));
    dis[1]=0;
    q.push(make_pair(0,1));
    while(!q.empty()){
        int x=q.top().second;
        q.pop();
        if(vis[x])continue;
        vis[x]=1;
        for(int i=head[x];i!=-1;i=a[i].Next)
        {
            int y=a[i].v;
            ll w=a[i].w;
            if(dis[y]>dis[x]+w){
                dis[y]=dis[x]+w;
                q.push(make_pair(-dis[y],y));
            }
        }
    }
}
int main(void)
{
    //ios::sync_with_stdio(false);
    cin>>n>>m>>k;
    init();
    scanf("%lld",&h[1]);
    k+=h[1];
    for(int i=2;i<=n;i++)
    {
        scanf("%lld",&h[i]);
    }
    int u,v;
    ll w;
    while(m--)
    {
        scanf("%d%d%lld",&u,&v,&w);
        if(h[v]>=k)
        addv(u,v,w+(h[v]-k)*(h[v]-k));
        else addv(u,v,w);
        if(h[u]>=k)
        addv(v,u,w+(h[u]-k)*(h[u]-k));
        else addv(v,u,w);
    }
    dijkstra();
    printf("%lld\n",dis[n]);
    return 0;
}