NYOJ115城市平乱_单源点最短路径(spfa)

题目链接

第一次敲的时候我用的是vector，看题的时候心里还在想着要判重边，把代码都敲完后发现了问题，用vector没办法判断重边啊~又把数据结构赶紧改成了邻接矩阵，同时也发现了自己容易犯的错误！学了新东西就只用新的，不善于思考两者的优劣势，（工欲善其事必先利其器）这句话自己没做到！！真是大忌啊！！！

还有一点就是今天学会了怎么去计算是否超内存了！这一点对于我做题时候选择数据结构时非常有用

#include<stdio.h>
#include<string.h>
#include<queue>
using namespace std;
#define inf 0x3f3f3f
#define min(a,b) a<b?a:b

int map[1005][1005];
int dis[1005];
int vis[1005];
int spfa(int s,int q,int n)
{
memset(dis,0x3f,sizeof(dis));
memset(vis,0,sizeof(vis));
int v,i,w,u;

queue<int>Q;
Q.push(s);
vis[s]=1;
dis[s]=0;

while(!Q.empty())
{
u=Q.front();
Q.pop();

for(v=1;v<=n;v++)
{
if(map[u][v]<inf)
{
if(dis[v] > dis[u]+map[u][v])
{
dis[v]=dis[u]+map[u][v];
if(!vis[v])
{
vis[v]=1;
Q.push(v);
}
}
}

}
}
return dis[q];
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
int n,m,p,q,i;
int army[105];
memset(army,0,sizeof(army));
memset(map,0x3f,sizeof(map));
scanf("%d%d%d%d",&n,&m,&p,&q);
for(i=1;i<=m;i++)
map[i][i]=0;
for(i=1;i<=n;i++)
{
scanf("%d",&army[i]);
}
for(i=0;i<p;i++)
{
int u,v,w;
scanf("%d%d%d",&u,&v,&w);//需要判重边
map[u][v]=map[v][u]=min(map[u][v],w);

}

int min=1<<28;
for(i=1;i<=n;i++)
{
int res=spfa(army[i],q,m);
if(min>res)
{
min=res;
}
}
printf("%d\n",min);

}
return 0;
}