判断完全多部图

完全多部图定义如下，给定一个无向图，图中包括n个点和m条边，保证无自环，两个点之间只有一条边。在这个图中，如果所有的点可以被划分为多个类，使得每一类的点两两之间没有连接，不同类别的任意两点之间有边连接，那么就称这个图是完全多部图。

这里的问题就是，给定一些点和边，如何用编程的方法判定这个图是否是完全多部图。

举个例子，下面这个图就是完全多部图，我们可以如此分类：[1, ], [2, 3, 4], [5, ]（数字表示顶点，中括号里的表示同一类点），这样的分类条件下，2、3、4是同一类点，两两之间不相连，但和其他类别的点1、5，任意两点间有连线。

给定的输入如下，第一行有两个数字，n, m，分别代表n个点，m条边；接下来是m行，每行是两个数字，分别代表这条边连接的两个顶点。要求是多部图输出“Yes”，非多部图输出“No”。

样例如下：

输入：
5 7
1 3
1 5
2 3
2 5
3 4
4 5
3 5

输出：
Yes

解题思路：

这里采用类别判定的方法，每次仅看一个类别的点，这些点应满足：类别内的任意两点间无连接；类别内的任意点和类别外的任意点有连接。依次判断每个类，如果所有类都满足这两个条件，则是多部图，否则不是。如何选取一个类呢？选定一个点，所有和这个点不相连的点应该属于同一个类别。

具体代码（基于python2.7）如下：

import sys, collections
# read data, save edges in map_
n,m = sys.stdin.readline().split('\n')[0].split(' ')
n = int(n)
m = int(m)
map_ = collections.defaultdict(lambda:set())
remains = set([i for i in range(1,n+1)])
for im in range(m):
    a1_, a2_ = sys.stdin.readline().split('\n')[0].split(' ')
    a1_, a2_ = int(a1_), int(a2_)
    map_[a1_].add(a2_)
    map_[a2_].add(a1_)
    
res = True
while (res and len(remains) > 0):
    # find one category
    from_ = remains.pop()
    newremains = set()
    for to in map_[from_]:
        map_[to].remove(from_)
        newremains.add(to)
        remains.remove(to)

    # point in one category do not connect one another
    for oa in remains:
        if res:
            for ob in remains:
                if oa != ob:
                    if oa in map_[ob]:
                        res = False
                        break

    # point in one category connects any other type points
    for with_ in remains:
        if res:
            for other_ in newremains:
                if other_ in map_[with_]:
                    map_[with_].remove(other_)
                    map_[other_].remove(with_)
                else:
                    res = False
                    break
    remains = newremains
print 'Yes' if res else 'No'