186. [USACO Oct08] 牧场旅行

 186. [USACO Oct08] 牧场旅行

            输入文件：pwalk.in   输出文件：pwalk.out   简单对比

时间限制：1 s   内存限制：128 MB

n个被自然地编号为1..n奶牛（1<=n<=1000）正在同样被方便的编号为1..n的n个牧场中吃草。更加自然而方便的是，第i个奶牛就在第i个牧场中吃草。

其中的一些对牧场被总共的n-1条双向通道的一条连接。奶牛可以通过通道。第i条通道连接的两个牧场是A_i和B_i（1<=A_i<=N;1<=B_i<=N）其长度是L_i（1<=L_i<=10000）。

通道只会连接两个不同的牧场，所以这些通道使得整个牧场构成了一棵树。

奶牛们是好交际的希望能够经常的访问别的奶牛。急切地，它们希望你能通过告诉它们Q（1<=Q<=1000）对牧场的路径来帮助他们安排旅行。（这里将有Q个询问，p1,p2(1<=p1<=n；1<=p1<=n)）

分数：200

问题名称：pwalk

输入格式：

第1行：两个用空格隔开的整数：n和Q

第2..n行：第i+1行包含三个用空格隔开的整数：A_i,B_i和L_i

第n+1..N+Q行：每行包含两个用空格隔开的整数，代表两个不同的牧场，p1和p2

输入样例（file pwalk.in）：

4 2

2 1 2

4 3 2

1 4 3

1 2

3 2

输出格式：

第1..Q行：行i包含第i个询问的答案。

输出样例：

2

7

输出说明：

询问1：牧场1和牧场2的路径长度为2。 询问2：3->4->1->2;总长为7。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;

const long MAXN=9999;
const long maxn=0x7fffffff;

bool visit[MAXN];
long dis[MAXN],n,m,u,v,len,t,a,e;
long head[MAXN],num_edge;
queue<int>que;

struct Edge{
    long pre;
    long to;
    long len;
}edge[MAXN];

void in(){
    num_edge=0;
    for(long i=1;i<=m;i++)    {
        cin>>u>>v>>len;
        edge[++num_edge].pre=head[u];
        edge[num_edge].to=v;
        edge[num_edge].len=len;
        head[u]=num_edge;
        swap(u,v);
        edge[++num_edge].pre=head[u];
        edge[num_edge].to=v;
        edge[num_edge].len=len;
        head[u]=num_edge;
    }
}

void spfa(){
    memset(visit,false,sizeof(visit));
    fill(dis,dis+MAXN,maxn);
    dis[a]=0;
    visit[a]=true;
    que.push(a); 
    while(!que.empty()){
        long k=que.front();
        que.pop();
        visit[k]=false;
        for(long j=head[k];j!=-1;j=edge[j].pre){
            long w=edge[j].len;
            if(dis[edge[j].to]>dis[k]+w){
                dis[edge[j].to]=dis[k]+w;
                if(!visit[edge[j].to]){
                    visit[edge[j].to]=true;
                    que.push(edge[j].to);
                }
            }
        }
    }
    cout<<dis[e]<<endl;
}

int main(){
    freopen("pwalk.in","r",stdin);
    freopen("pwalk.out","w",stdout);
    memset(head,-1,sizeof(head));
    cin>>n>>t;
    m=n-1;
    in();
    for(int i=1;i<=t;i++){
          cin>>a>>e;
          spfa();
      }
    fclose(stdin);
    fclose(stdout);
    return 0;
}