BZOJ 4152: [AMPPZ2014]The Captain

本来可以考虑每两个点之间连一条边 O(n^2)

但是如果两个点之间连了一条x的边,中间还有一个点,那么|x_i - x_a|+|x_a - x_i+1| = |x_i - x_i+1|

可以只连i 到a的x的边和a到i+1的x的边

y同理只连排序后的相邻的边

用dijkstra 跑一遍最短路。稳定的时间复杂度O(nlogn) 可以过

为什么不用spfa呢。。。因为时间复杂度不稳定,理论上是O(mE)但是这道TLE了。。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <algorithm>
#include <cstdlib>
using namespace std;
#define MAXN 200010
#define INF 0x7fffffff
long long ans;
int head[MAXN];
bool inq[MAXN];
long long dis[MAXN];
int l,n;
struct node{
    int x,y,id;
}a[MAXN];
struct point{
    int y,z,next;
}edge[4*MAXN];
void add(int x,int y,int z)
{
    l++;
    edge[l].y=y;edge[l].z=z;edge[l].next=head[x];
    head[x]=l;
}
int cmp1(node a,node b)
{
    return a.x<b.x;
}
int cmp2(node a,node b)
{
    return a.y<b.y;
}
struct line{
    int x,dis;
    bool operator < (const line &o) const
    {
        return dis > o.dis;
    }
};
void dijkstra()
{
    priority_queue<line> q;
    for (int i=1;i<=n;i++)
        dis[i]=INF;
    dis[1]=0;
    q.push((line){1,0});
    while (!q.empty())
    {
        line t=q.top();q.pop();
        if (dis[t.x]!=t.dis) continue;
        int p=head[t.x];
        while (p)
        {
            if (dis[edge[p].y]>dis[t.x]+edge[p].z)
            {
                dis[edge[p].y]=dis[t.x]+edge[p].z;
                q.push((line){edge[p].y,dis[edge[p].y]});
            }
            p=edge[p].next;
        }
    }
    printf("%d",dis[n]);
}
int main()
{
    scanf("%d", &n);
    for (int i=1;i<=n;i++)
    {
        scanf("%d%d", &a[i].x, &a[i].y);
        a[i].id=i;
    }
    sort(a+1,a+n+1,cmp1);
    for (int i=1;i<n;i++)
    {
        add(a[i].id,a[i+1].id,a[i+1].x-a[i].x);
        add(a[i+1].id,a[i].id,a[i+1].x-a[i].x);
    }
    sort(a+1,a+n+1,cmp2);
    for (int i=1;i<n;i++)
    {
        add(a[i].id,a[i+1].id,a[i+1].y-a[i].y);
        add(a[i+1].id,a[i].id,a[i+1].y-a[i].y);
    }
    dijkstra();
}

--------------------------------------

c++还是没学好,有两个元素的优先队列不会写。。。

你可能感兴趣的:(BZOJ 4152: [AMPPZ2014]The Captain)