题目来自nyist第115题
描述
南将军统领着N个部队,这N个部队分别驻扎在N个不同的城市。
他在用这N个部队维护着M个城市的治安,这M个城市分别编号从1到M。
现在,小工军师告诉南将军,第K号城市发生了暴乱,南将军从各个部队都派遣了一个分队沿最近路去往暴乱城市平乱。
现在已知在任意两个城市之间的路行军所需的时间,你作为南将军麾下最厉害的程序员,请你编写一个程序来告诉南将军第一个分队到达叛乱城市所需的时间。
注意,两个城市之间可能不只一条路。
输入
第一行输入一个整数T,表示测试数据的组数。(T<20)
每组测试数据的第一行是四个整数N,M,P,Q(1<=N<=100,N<=M<=1000,M-1<=P<=100000)其中N表示部队数,M表示城市数,P表示城市之间的路的条数,Q表示发生暴乱的城市编号。
随后的一行是N个整数,表示部队所在城市的编号。
再之后的P行,每行有三个正整数,a,b,t(1<=a,b<=M,1<=t<=100),表示a,b之间的路如果行军需要用时为t
数据保证暴乱的城市是可达的。
输出
对于每组测试数据,输出第一支部队到达叛乱城市时的时间。每组输出占一行
用队列以及vector数组维护数据,将已经驻扎军队的城市入队,标记到达各个城市的耗时最小值,由于有多个出发点并且题目特性,最小值会不断更新,所以一定修改了最小值就需要入队重新计算周边值(这是平时算法中学习的思路,感觉有可以优化的地方,抽空看一下最优解)。最后输出暴乱城市在数组中的值。代码如下:
#include <cstdio>
#include <cstring>
#include <vector>
#include <queue>
using namespace std;
struct Node{
int b,c;
}temp;
int occupy[1000+5];
queue<int> Que;
//vector<struct Node> G[1000+5];
int main()
{
int T,N,M,P,Q,a,b,c;
int i,j;
scanf("%d",&T);
while(T--)
{
vector<struct Node> G[1000+5];
memset(occupy,-1,sizeof(occupy));
scanf("%d %d %d %d",&N,&M,&P,&Q);
for(i =1;i<=N;i++)
{
scanf("%d",&j);
occupy[j] = 0;
Que.push(j);
}
for(i =0;i<P;i++)
{
scanf("%d %d %d",&a,&b,&c);
temp.b = b;temp.c = c;
G[a].push_back(temp);
temp.b = a;
G[b].push_back(temp);
}
while(!Que.empty())
{
int t = Que.front();Que.pop();
for(i =0 ;i < G[t].size();i++)
{
temp = G[t][i];
c = temp.c;
b = temp.b;
if(occupy[b] >c+occupy[t] || occupy[b] == -1)
{
occupy[b] = c+occupy[t];
Que.push(b);
}
}
}
printf("%d\n",occupy[Q]);
// for(i =1;i <= M;i++)
// G[i].clear();
}
return 0;
}
代码中采用在循环中声明vector的方法,而注释处采用循环情况vector数组的方法,经测验,直接在循环中声明vector要快很多,并且空间也小一些,如下
具体原因还不太清除,占坑后补。