POJ 1985 Cow Marathon

这是自己写的第一道关于树的直径的题目，因此记录下自己的思维过程。树的直径，众所周知，两次BFS，因此主体思路就出来了；接下来自己就是要考虑如何存储数据，因为每条路径都有长度，于是自己想到用结构体来存储；由于害怕内存超了，自己不敢用二维数组，于是准备用邻接表来存储结点信息。可是自己对邻接表真的不太熟悉，由于这道题的特殊性，每个结点最多只有四个邻接点，所以只要用一个大小为4的数组来记录就可以了。其他具体详见代码注释。

当然自己的代码有许多需要改善的地方，例如有些代码显得冗余重复，导致代码长度显得过长。还有变量名字起得有点长，这也没办法，自己的记性太差，写着写着就不知道变量名是什么意思了，有时居然会起相同的变量名。而且以后再阅读自己的代码的时候肯定不知所云了，呵呵！

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <deque>
#include <algorithm>
using namespace std;
const int N=40005;
int n,m;
bool visited[N];//用来标志结点是否被访问过。
struct farm
{

<pre name="code" class="cpp">//记录与该结点相邻的四个结点的距离，如果不相邻，则显然值为0.

int distance[4];

<pre name="code" class="cpp">//记录与该结点相邻的四个结点的标号，与distance数组对应。

//即distance[i]表示的是和farm_index[i]所记录的标号所代表的farm的距离。

//显然当distance[i]为0时，对应的farm_index[i]也为0。 int farm_index[4];

//记录当前结点到所选的初始结点的距离。
    int len;
}data[N];
void bfs(int x)
{
    deque<int> d;
    for(int i=0;i<4;i++)
    {
        if(data[x].distance[i]!=0)
        {
            int t=data[x].farm_index[i];
            d.push_back(t);

//与x相邻的下一个访问t结点到所设的初始结点（假设为1结点）的距离，

//显然为x到1结点的距离加上t和x之间的距离。
            data[t].len=data[x].len+data[x].distance[i];
        }
    }
    visited[x]=true;//因为结点x已经访问过，所以设为true。
    while(!d.empty())
    {
        int t=d.front();
        d.pop_front();
        for(int i=0;i<4;i++)
        {
            if(data[t].distance[i]!=0)
            {
                int a=data[t].farm_index[i];
                if(visited[a]==false)
                {
                    d.push_back(a);
                    data[a].len=data[t].distance[i]+data[t].len;
                }
            }
        }
        visited[t]=true;//一开始自己这行代码的位置放错了，一定要注意设置哪个结点已经被访问过。
    }
}
int main()
{
    int farm1,farm2,dist;
    char direction;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        for(int i=1;i<=n;i++)
        {
            memset(data[i].distance,0,sizeof(data[i].distance));
            memset(data[i].farm_index,0,sizeof(data[i].farm_index));
            data[i].len=0;
            visited[i]=false;
        }
        while(m--)
        {
            scanf("%d%d%d %c",&farm1,&farm2,&dist,&direction);
            int temp=0;
            switch(direction)
            {
                case 'E':temp=0;break;
                case 'S':temp=1;break;
                case 'W':temp=2;break;
                case 'N':temp=3;break;
            }
            data[farm1].distance[temp]=dist;
            data[farm1].farm_index[temp]=farm2;
            data[farm2].distance[(temp+2)%4]=dist;
            data[farm2].farm_index[(temp+2)%4]=farm1;
        }

//第一次BFS找出最远结点。
        bfs(1);
        int ans_max=0,start=1;
        for(int i=1;i<=n;i++)
        {
            if(data[i].len>ans_max)
            {
                ans_max=data[i].len;
                start=i;//用来标志最远结点的序号。
            }
        }

//再一次的初始化，这里的代码有些冗余，因为完全可以与之前的初始化代码写成一个函数。
        for(int i=1;i<=n;i++)
        {
            data[i].len=0;
            visited[i]=false;
        }

//第二次BFS得出最远距离。
        bfs(start);
        ans_max=0;
        for(int i=1;i<=n;i++)
            if(data[i].len>ans_max)
            ans_max=data[i].len;
        printf("%d\n",ans_max);
    }
    return 0;
}