[topcoder]BestRoads

http://community.topcoder.com/stat?c=problem_statement&pm=10172&rd=13515

http://community.topcoder.com/tc?module=Static&d1=match_editorials&d2=srm424

这道题目是和最小生成树有关。记得最小生成树的一个算法（容易实现的那个），Kruskal，要用到并查集。想想就是以边为主，那么才可能形成多个边的集合，然后再合并。
并查集的n次合并查找的复杂度是o(n)。TopCoder的思想似乎主要是快速实现解决问题，对复杂度的要求一般，所以这里简单的链式实现就够用了，复杂度是一次合并o(n)（果然比之前好的数据结构复杂度大大增加啊）。
另外返回一个空的数组是直接return new int[]{}就行了。

public class BestRoads
{
    public int[] numberOfRoads(String[] roads, int M)
    {
        int N = roads.length;
        char[][] matrix = new char[N][N];
        for (int i = 0; i < N; i++)
        {
            matrix[i] = roads[i].toCharArray();
        }

        int id[] = new int[N]; // simple joint-set
        for (int i = 0; i < N; i++)
        {
            id[i] = i;
        }
        int deg[] = new int[N];
        int CmpCnt = 0;
        for (int i = 0; i < N; i++)
        {
            for (int j = i+1; j < N; j++)
            {
                if (matrix[i][j] == 'N' || id[i] == id[j]) continue;
                int idi = id[i]; int idj = id[j];
                matrix[i][j] = 'N'; matrix[j][i] = 'N';
                deg[i]++; deg[j]++;
                M--; CmpCnt++;
                for (int t = 0; t < N; t++)
                {
                    if (id[t] == idi) id[t] = idj;
                }
            }
        }
        if (CmpCnt != N-1) return new int[]{}; // not connected;

        for (int i = 0; i < N && M > 0; i++)
        {
            for (int j = i+1; j < N && M > 0; j++)
            {
                if (matrix[i][j] == 'Y')
                {
                    matrix[i][j] = 'N';
                    matrix[j][i] = 'N';
                    deg[i]++; deg[j]++;
                    M--;
                }
            }
        }
        if (M > 0) return new int[]{};
        return deg;
    }
}