Yaroslav and Time CodeForces - 302D (最短路)

题目思路

这题要找的是从1到n的最小距离
用最短路就好 可能是最近写并查集 最小生成树写魔怔了
写了好久才发现自己写成最小生成树了
这题需要注意的是不要无脑见双向边
因为在这道题里面我们的边权可能会随着边的终点改变
所以遍历所有点直接建单向边就好了
这个细节卡了我好久

ac代码

#include <stdio.h>
#include <iostream>
#include <algorithm>
#include <math.h>
#include <string.h>
#include <vector>
#include <stack>
#include <queue>
#include  <map>
#include <set>
#include <utility>
#define pi 3.1415926535898
#define ll long long
#define lson rt<<1
#define rson rt<<1|1
#define eps 1e-6
#define ms(a,b) memset(a,b,sizeof(a))
#define legal(a,b) a&b
#define print1 printf("111\n")
using namespace std;
const int maxn = 1e5+5;
const int inf = 0x3f3f3f3f;
const ll llinf = 0x3f3f3f3f3f3f3f3f;
const ll mod = 1000000007;
//998244353

int a[maxn],vis[maxn],len;
int first[maxn];
ll dis[maxn];
int n,d;
struct node1
{
    int x,y;
}e[maxn];

struct node
{
    int to,next;
    ll v;
}t[maxn];

void add(int u,int v,ll w)
{
    t[++len].to=v;
    t[len].v=w;
    t[len].next=first[u];
    first[u]=len;
}


struct point
{
    int id;
    ll val;
    point(int id,ll val)
    {
        this->id=id;
        this->val=val;
    }
    bool operator<(const point &x)const
    {
        return val>x.val;
    }
};

void dij(int s)
{
    ms(vis,0);
    for(int i=0;i<=n+1;i++)dis[i]=llinf;
    priority_queue<point>q;
    q.push(point(s,0));
    dis[s]=0;
    while(!q.empty())
    {
        int rt=q.top().id;
        q.pop();
        if(vis[rt])continue;
        vis[rt]=1;
        for(int i=first[rt];i!=-1;i=t[i].next)
        {
            int id=t[i].to;
            if(!vis[id]&&dis[rt]+t[i].v<dis[id])
            {
                dis[id]=dis[rt]+t[i].v;
                q.push(point(id,dis[id]));
            }
        }
    }
}

int main()
{
    scanf("%d%d",&n,&d);
    ms(first,-1);
    for(int i=2;i<n;i++)
        scanf("%d",&a[i]);
    a[n]=a[1]=0;
    for(int i=1;i<=n;i++)
    {
        scanf("%d%d",&e[i].x,&e[i].y);
    }
    int cot=0;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            ll tem=(abs(e[i].x-e[j].x)+abs(e[i].y-e[j].y))*d-a[j];
            add(i,j,tem);
        }
    }
    dij(1);
    printf("%lld\n",dis[n]);

}

你可能感兴趣的:(Yaroslav and Time CodeForces - 302D (最短路))