[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;
    }
}
原文地址:https://www.cnblogs.com/lautsie/p/3341931.html