题目链接: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);
}
}