poj 1984 Navigation Nightmare 並查集

 並查集應用的變形。題目意思是一個圖中，只有上下左右四個方向的邊。給出這樣的一些邊，
求任意指定的2個節點之間的距離。
   有可能當前給出的信息，沒有涉及到要求的2個節點，或者只涉及到了1個節點，那麼肯定
無法確定它們的距離。或者根據已經給出的邊只知道這2個節點在不同的聯通分量裡面，那麼其
距離也是無法確定的，根據題目要求，輸出-1。
   問題是如果能夠確定它們在一個聯通分量裡面，如何確定它們的距離了。
   這個題的關鍵在於，只有上下左右四個方向的邊，假設每個節點都有一個坐標的話，那麼它們
相對於代表該聯通分量節點的坐標肯定是固定的，那麼就不需要考慮圖裡面有環之類的情況了。
這樣就可以很方便的應用並查集來解了。
   利用並查集，給每個節點附加其它信息，即相對於代表該並查集的節點的坐標（x，y）。
在FindSet裡面求出坐標，在UnionSet裡面修改合並後新加入的另外一個集合的根節點的坐標即可。
   代碼如下：

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <algorithm>
using namespace std;

const int MAX_N = 40010;
int nN, nM;
int nSets[MAX_N];
int nX[MAX_N];
int nY[MAX_N];
char szInput[MAX_N][100];

void MakeSets(int nNum)
{
    for (int i = 0; i < nNum; ++i)
    {
        nSets[i] = i;
        nX[i] = nY[i] = 0;
    }
}

int FindSets(int nI)
{
    if (nSets[nI] != nI)
    {
        int nPre = nSets[nI];
        nSets[nI] = FindSets(nSets[nI]);
        nX[nI] += nX[nPre];
        nY[nI] += nY[nPre];
    }
    return nSets[nI];
}

void UnionSets(int nBeg, int nEnd, int dx, int dy)
{
    int nA = FindSets(nBeg);
    int nB = FindSets(nEnd);
    if (nA != nB)
    {
        nSets[nB] = nA;//把集合B合並到集合A中
        nX[nB] = nX[nBeg] + dx - nX[nEnd];//因為方向逆過來了,所以是減去
        nY[nB] = nY[nBeg] + dy - nY[nEnd];
    }
}

int main()
{
    int nBeg, nEnd, nL;
    char szDir[10];

    while (scanf("%d%d%*c", &nN, &nM) == 2)
    {
        MakeSets(nN);
        for (int i = 0; i < nM; ++i)
        {
            fgets(szInput[i], 100, stdin);
        }
        int nK;
        int nF1, nF2, nI;
        scanf("%d", &nK);
        int nCur = 0;
        while (nK--)
        {
            scanf("%d%d%d", &nF1, &nF2, &nI);
            for (int i = nCur; i < nI; ++i)
            {
                sscanf(szInput[i], "%d%d%d%s", &nBeg,
                       &nEnd, &nL, szDir);
                int dx = 0, dy = 0;
                switch (szDir[0])
                {
                    case 'N': dy += nL; break;
                    case 'S': dy -= nL; break;
                    case 'E': dx += nL; break;
                    case 'W': dx -= nL; break;
                }
                UnionSets(nBeg, nEnd, dx, dy);
            }
            nCur = nI;
           
            if (FindSets(nF1) != FindSets(nF2))
            {
                printf("-1\n");
            }
            else
            {
                printf("%d\n", abs(nX[nF1] - nX[nF2])
                        + abs(nY[nF1] - nY[nF2]));
            }
        }
    }

    return 0;