Codeforces 144D Missile Silos

题目链接：http://codeforces.com/problemset/problem/144/D

题意：给出n个点，m条边，以及原点，导弹发射井在离原点len(必须是最短路径)，有可能在点上或者在边上，问有几个导弹发射井

思路：首先最短路求出各点到原点的距离，判断出在点上的导弹井，然后在判断在边上的导弹井(边要存放在另一个数组里方便判断……而且不这么做会出莫名的错误，wa了几次)，dijstra会超时要用优先队列优化(然而用的模板有些问题又wa了
)，判断有3种情况(有可能出现1种以上的情况)

1.导弹井靠近w<u,v>的u，首先导弹要在边上，第二就是从点u到导弹井要更近因此

(dist[u]<len) && (len-dist[u]<w) && (w-(len-dist[u])>len-dist[v])

2.导弹井靠近w<u,v>的v，同上

3.刚好在u和v的中间， (w-(len-dist[u])==len-dist[v])也就是(dist[u]+dist[v]+w==len*2)

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <vector>
#include <queue>
#define Maxn 300030
using namespace std;
const int inf=0x3f3f3f3f;
struct Edge
{
    int u,v,w;

}line[Maxn];

struct Node
{
    int dis,pos;
    bool operator < (const Node &node) const
    {
        return dis > node.dis;
    }
};

vector<Edge> list[Maxn];

int vis[Maxn],dist[Maxn],used[Maxn],st;
int n,m,k,sum;
priority_queue<Node> Q;
void init()
{
    sum=0;
    for (int i=0;i<=n;i++) list[i].clear();
    while (!Q.empty()) Q.pop();
    memset(vis,0,sizeof(vis));
    memset(dist,inf,sizeof(dist));
    dist[st]=0;
}




void relax(int u,int v,int w)
{
    if (dist[v] > dist[u] + w)
    {
        dist[v]=dist[u]+w;Q.push((Node){w,v});
    }
}


void dijkstra()
{
    Q.push((Node){0,st});
    while (!Q.empty())
    {
      Node nownode=Q.top();
      int u=nownode.pos;
      Q.pop();
        int tem=nownode.pos;
        if(dist[tem]<nownode.dis) continue;
      for (int i=0;i<list[u].size();i++)
      {
          relax(list[u][i].u,list[u][i].v,list[u][i].w);

      }
    }
}

void addEdge(int u,int v,int w)
{
    list[u].push_back((Edge){u,v,w});
}


int main()
{
    int m,cap,len;
    while (scanf("%d%d%d",&n,&m,&st)!=EOF)
    {
        init();
        for (int i=0;i<m;i++)
        {
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            addEdge(u,v,w);
            addEdge(v,u,w);
            line[i].u=u;
            line[i].v=v;
            line[i].w=w;
        }
        dijkstra();
        scanf("%d",&len);
        int res=0;
        for (int i=0;i<m;i++)
        {
                int u=line[i].u,v=line[i].v,w=line[i].w;
                if ( (dist[u]<len) && (len-dist[u]<w) && (w-(len-dist[u])>len-dist[v]) ) res++;
                if ( (dist[v]<len) && (len-dist[v]<w) && (w-(len-dist[v])>len-dist[u]) ) res++;
                if ( (dist[v]<len) && (dist[u]<len) && (dist[u]+dist[v]+w==len*2) ) res++;

        }
        for (int i=1;i<=n;i++)
        {
            if (dist[i]==len) res++;
        }
        printf("%d\n",res);
    }
}