codeforces 144D Missile Silos spfa

题意：给定一张地图，可以在边和点上设立发射井，其中发射井到首都的距离必须等于l

做法：十万个点，竟然可以求最短路，而且最后时间还只有区区125ms.可能是边太少的原因...

图中的边可以分成两类，一种是最短路上的边，一种是普通边。

若边（u,v）是v最短路上的边，这样要dis[v]>l并且dis[u]<l,,路上才可以造导弹井。

否则，只要当dis[u]<l，且由v点到导弹井的距离大于l，这个位置就是合理的。这里的判断与上面的不同，是因为，这里即使dis[u]<l&&dis[v]<l也是可以造井的

#include<cstdio>
#include<queue>
#include<cstring>
#define LMT 100005
using namespace std;
class line
{
   public:
    int u,v,next;
    int l;
}e[LMT<<1];
int next[LMT],dis[LMT],count[LMT],all,n,m,s,l;
bool vis[LMT];
queue<int>q;
void insert(int u,int v,int ll)
{
    e[all].u=u;e[all].v=v;e[all].next=next[u];e[all].l=ll;next[u]=all++;
    e[all].v=u;e[all].u=v;e[all].next=next[v];e[all].l=ll;next[v]=all++;
}
void spfa(void)
{
    int u,v,x;
    memset(dis,127,sizeof(dis));
    memset(count,0,sizeof(count));
    memset(vis,0,sizeof(vis));
    dis[s]=0;
    vis[s]=1;
    count[s]++;
    q.push(s);
    while(!q.empty())
    {
        u=q.front();
        q.pop();
        count[u]++;
        vis[u]=0;
        if (count[u]>=10*n)break;
        for(x=next[u];x!=-1;x=e[x].next)
        if(dis[e[x].v]>dis[u]+e[x].l)
        {
            v=e[x].v;
            dis[v]=dis[u]+e[x].l;
           if(0==vis[v])
            {
                q.push(v);
                vis[v]=1;
            }
        }
    }
}
int main()
{
    int u,v,ans=0;
    scanf("%d%d%d",&n,&m,&s);
    memset(next,-1,sizeof next);
    for(int i=0;i<m;i++)
    {
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        insert(a,b,c);
    }
    spfa();
    scanf("%d",&l);
    for(int i=1;i<=n;i++)
    if(dis[i]==l)ans++;
    for(int i=0;i<all;i++)
    if(i&1)
    {
        u=e[i].u;
        v=e[i].v;
        if(dis[v]==dis[u]+e[i].l)
        {
            if(dis[u]<l&&dis[v]>l)
            ans++;
        }
        else
        {
            if(dis[u]==dis[v]+e[i].l)
            {
                if(dis[v]<l&&dis[u]>l)
                ans++;
            }
            else
            {
                if(dis[u]<l&&dis[v]+e[i].l-(l-dis[u])>=l)ans++;
                if(dis[v]<l&&dis[u]+e[i].l-(l-dis[v])>=l)ans++;
                if(dis[u]<l&&dis[v]<l&&l-dis[u]+l-dis[v]==e[i].l)ans--;
            }
        }
    }
    printf("%d\n",ans);
    return 0;
}