poj 3159 差分约束 用栈代替队列

在别人的和擦粉约束列表里搜到的这题，看完题目果断感觉是水题

根据s[b]-s[a]<=c来建图

最后是求s[n]-s[1]的最大值，从1出发走一遍到n的最短路径就好

但是超时了，果断不能接受。。。。

别人的做法:用栈来代替spfa的队列，或者用dijkstra+heap来做。。晕倒，总之，又学到了一点。。。

View Code

#include<stdio.h>
#include<stack>
#include<string.h>
using namespace std;
const int INF = 9999999;
struct NODE{
    int v,w;
    int next;
}list[150010];
int tot;
int head[50010];
int dis[50010];
int vis[50010];
stack<int> Q;
int n,m;
void init()
{
    int i;tot=0;
    memset(head,-1,sizeof(head));
}
void add(int s,int t,int w)
{
    list[tot].v=t;
    list[tot].w=w;
    list[tot].next=head[s];
    head[s]=tot++;
}
void spfa(int src)
{
    int i,j,u;
    while(!Q.empty())  Q.pop();
    memset(vis,0,sizeof(vis));
    for(i=0;i<=50000;i++)
        dis[i]=INF;
    Q.push(src);
    vis[src]=1;
    dis[src]=0;
    while(!Q.empty())
    {
        u=Q.top();
        Q.pop();
        vis[u]=0;
        for(j=head[u];j!=-1;j=list[j].next)
        {
            int e=list[j].v;
            if(dis[e]>list[j].w+dis[u])
            {
                dis[e]=list[j].w+dis[u];
                if(!vis[e])
                {
                    Q.push(e);
                    vis[e]=1;
                }
            }
        }
    }
    printf("%d\n",dis[n]);
}
int main()
{
    int i,a,b,c;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        init();
        for(i=0;i<m;i++)
        {
            scanf("%d%d%d",&a,&b,&c);
            add(a,b,c);
        }
        spfa(1);
    }
}

dijkstra+heap

View Code

#include<stdio.h>
#include<queue>
#include<string.h>
using namespace std;
const int INF = 9999999;
struct NODE{
    int v,w;
    int next;
}list[150010];
int tot;
int head[50010];
int n,m;
void init()
{
    tot=0;
    memset(head,-1,sizeof(head));
}
void add(int s,int t,int w)
{
    list[tot].v=t;
    list[tot].w=w;
    list[tot].next=head[s];
    head[s]=tot++;
}
struct PP{
    int len,num;
    bool operator<(const PP &b) const{
        return len > b.len;
    }
}dis[30010];
priority_queue<PP> Q;
int vis[30010];
void dijkstra()
{
    while(!Q.empty()) Q.pop();
    int i,j,k;
    PP tmp;
    memset(vis,0,sizeof(vis));
    for(i=1;i<=n;i++)
    {
        dis[i].num=i;
        dis[i].len=INF;
    }
    dis[1].len=0;
    dis[1].num=1;
    Q.push(dis[1]);
    while(!Q.empty())
    {
        tmp=Q.top();Q.pop();
        int now=tmp.num;
        if(vis[now]) continue;
        vis[now]=1;
        for(i=head[now];i!=-1;i=list[i].next)
        {
            int len=list[i].w;
            int to=list[i].v;
            if(dis[to].len>dis[now].len+len)
            {
                dis[to].len=dis[now].len+len;
                Q.push(dis[to]);
            }
        }
    }
    printf("%d\n",dis[n].len);
}
int main()
{
    int i,a,b,c;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        init();
        for(i=0;i<m;i++)
        {
            scanf("%d%d%d",&a,&b,&c);
            add(a,b,c);
        }
        dijkstra();
    }
}