Problem p03409 (python)

Fixed submission:
#https://abc010.contest.atcoder.jp/tasks/abc010_4
#http://kmjp.hatenablog.jp/entry/2014/06/08/0900

#よくわからない最大フロー用のdfs---------------------------------

num=201 #今回のnの最大値100*2+1

def dfs(cur):
	global vis
	global EE
	vis[cur] = 1

	if cur==num:
		return 1

	for i in range(num+1):
		if vis[i]==0 and EE[cur][i]>0 and dfs(i) > 0:
			EE[cur][i] -= 1
			EE[i][cur] += 1
			return 1
	return 0
#------------------------------------------------------------------------------------

#今回の問題分（インプット～隣接リストの作成）------------------

n=int(input())
s=[]
ss=[]
for i in range(n):
    s.append([int(i) for i in input().split()])
for i in range(n):
    ss.append([int(i) for i in input().split()])
s.sort(key=lambda x:(x[0],x[1]))
ss.sort(key=lambda x:(x[0],x[1]))

pair=[]
for i in range(len(s)):
    right=[]
    for j in range(n):    
        if s[j][0]<ss[i][0] and s[j][1]<ss[i][1]:
            right.append(j)
    pair.append(right)
pair.sort(key=lambda x:len(x))

#print(pair)

#------------------------------------------------------------------------------------

#BFSにぶち込む用のグラフ作成------------------------------------------

EE = [[0 for x in range(num+1)] for y in range(num+11)]
vis = [0] * (num+1)

for i in range(len(pair)): #1から最大100　始点s（番号0）から青い点への辺
	EE[0][i+1] = 1
	EE[i+1][0] = 1

for i in range(n): #101から最大200　赤い点から終点t（番号201）への辺
	EE[i+101][num] = 1

for i in range(len(pair)): #青い点（番号1～100に振り直し）から　赤い点（番号101～200）への辺
    for j in range(len(pair[i])):
        node1, node2 = i+1,pair[i][j]+101
#        EE[node1][node2] = EE[node2][node1] = 1　なんでこれダメなんだろう
        EE[node1][node2] = 1

#風呂～～-----------------------------------------------------------------------

flow=0
while 1:
    for i in range(0, num+1):
        vis[i] = 0
    if dfs(0) == 0:
        break
    flow += 1

print (flow)

#------------------------------------------------------------------------------------

Buggy submission (verdict: Runtime Error):
#http://nobuta05.hatenablog.com/entry/2015/11/12/013938
"""
==================================================
二部グラフの最大マッチングを求めるプログラム

# 条件
G = (U,V,E) |U| = |E|

# 入力
U --> 1 2 3 4 .... 10
V --> a b c d .... j
E -->
1 a
1 d
2 b
2 f
etc...

# 出力(最大マッチングの一例)
edges = [(1,a), (2,b), (3 h), ...]

# 定義
def get_edge(M, u, U)    枝集合Mに含まれる枝のうち、頂点uと接している枝の他方の頂点_(1)を返す関数。そのような枝
                         が複数ある場合、一番最初に見つかった頂点を返す。また、引数にUが指定されている場合、(1)
                         を満たし、かつUに含まれる頂点を返す。

def get_nodes(M)         枝集合Mに含まれる枝に関して、左接続の頂点と右接続の頂点に分け、左接続の頂点集合と右接続
                         の頂点集合の二つを返す関数。

def augpath(Si, Ti, M, E)
                         二部グラフにおいて、増大道があるかどうかを判定・処理するプログラム。

==================================================
"""

def get_edge(M, u, U=None):

    if U == None:
        for x in M[:]:
            if x[0] == u:
                return x[1]
            elif x[1] == u:
                return x[0]

    else:
        for x in M[:]:
            if x[0] == u and x[1] in U:
                return x[1]
            elif x[1] == u and x[0] in U:
                return x[0]
    return None

def get_nodes(M):
    U = set()
    V = set()
    for x in M:
        U.add(x[0])
        V.add(x[1])
    return (U, V)


def augpath(Si, Ti, M, E):
    (U, V) = get_nodes(M)
    Ti1 = set(x[1] for x in E if x[0] in Si and not x[1] in Ti)
    
    if Ti1 == set():
        return (M, None)

    elif Ti1.difference(V) == set():
        Si1 = set(x[0] for x in M if x[1] in Ti1)
        (M, ui1) = augpath(Si1, Ti.union(Ti1), M, E)

        if ui1 == None:
            return (M, None)

        else:
            v = get_edge(M, ui1)
            M.remove((ui1,v))

            ui = get_edge(E, v, U=Si)
            M.append((ui,v))
            
            return (M, ui)

    else:
        v = Ti1.difference(V).pop()
        u = get_edge(E, v, U=Si)
        M.append((u,v))

        return (M, u)

def Hungary(U0, V0, E):
    U_bef = set()
    M = []
    cnt = 0
    length = len(U0)
    
    # CALC
    while True:

        # INIT
        (U, V) = get_nodes(M)
        u = U0[cnt%length]
        cnt = cnt + 1
        S0 = set()
        S0.add(u)
        T0 = set()

        # 終了条件(すべての頂点から始まる増大道を調べて、マッチング数が変わっていなければ、それ以上増えることはない)
        if cnt % length == 0:
            if U.difference(U_bef) == set():
                break
            U_bef = U.copy()

        # 枝刈り(すでにマッチングしている頂点から始まる道は増大道ではない)
        if u in U:
            continue
        
        #CALC
        (M, u) = augpath(S0, T0, M, E)
    
    return M

#今回の問題分---------------------------------------------------------------------

n=int(input())
s=[]
ss=[]
for i in range(n):
    s.append([int(i) for i in input().split()])
for i in range(n):
    ss.append([int(i) for i in input().split()])
s.sort(key=lambda x:(x[0],x[1]))
ss.sort(key=lambda x:(x[0],x[1]))
#print(s,ss)

pair=[]
for i in range(len(s)):
    right=[]
    for j in range(n):    
        if s[j][0]<ss[i][0] and s[j][1]<ss[i][1]:
            right.append(j)
    pair.append(right)
pair.sort(key=lambda x:len(x))

#---------------------------------------------------------------------

E = []
U0 = [x for x in range(len(pair))]
V0 = [x for x in range(n)]


for i in range(len(pair)):
    try:
        for j in range(len(pair[i])):
            node1, node2 = i,pair[i][j]
            E.append((node1, node2))
    except:
        pass

# CALC
try:
    edges = Hungary(U0, V0, E)
except:
    pass

# OUTPUT
try:
    print(len(edges))
else:
    print(0)

Buggy submission (verdict: Runtime Error):
#http://nobuta05.hatenablog.com/entry/2015/11/12/013938
"""
==================================================
二部グラフの最大マッチングを求めるプログラム

# 条件
G = (U,V,E) |U| = |E|

# 入力
U --> 1 2 3 4 .... 10
V --> a b c d .... j
E -->
1 a
1 d
2 b
2 f
etc...

# 出力(最大マッチングの一例)
edges = [(1,a), (2,b), (3 h), ...]

# 定義
def get_edge(M, u, U)    枝集合Mに含まれる枝のうち、頂点uと接している枝の他方の頂点_(1)を返す関数。そのような枝
                         が複数ある場合、一番最初に見つかった頂点を返す。また、引数にUが指定されている場合、(1)
                         を満たし、かつUに含まれる頂点を返す。

def get_nodes(M)         枝集合Mに含まれる枝に関して、左接続の頂点と右接続の頂点に分け、左接続の頂点集合と右接続
                         の頂点集合の二つを返す関数。

def augpath(Si, Ti, M, E)
                         二部グラフにおいて、増大道があるかどうかを判定・処理するプログラム。

==================================================
"""

def get_edge(M, u, U=None):

    if U == None:
        for x in M[:]:
            if x[0] == u:
                return x[1]
            elif x[1] == u:
                return x[0]

    else:
        for x in M[:]:
            if x[0] == u and x[1] in U:
                return x[1]
            elif x[1] == u and x[0] in U:
                return x[0]
    return None

def get_nodes(M):
    U = set()
    V = set()
    for x in M:
        U.add(x[0])
        V.add(x[1])
    return (U, V)


def augpath(Si, Ti, M, E):
    (U, V) = get_nodes(M)
    Ti1 = set(x[1] for x in E if x[0] in Si and not x[1] in Ti)
    
    if Ti1 == set():
        return (M, None)

    elif Ti1.difference(V) == set():
        Si1 = set(x[0] for x in M if x[1] in Ti1)
        (M, ui1) = augpath(Si1, Ti.union(Ti1), M, E)

        if ui1 == None:
            return (M, None)

        else:
            v = get_edge(M, ui1)
            M.remove((ui1,v))

            ui = get_edge(E, v, U=Si)
            M.append((ui,v))
            
            return (M, ui)

    else:
        v = Ti1.difference(V).pop()
        u = get_edge(E, v, U=Si)
        M.append((u,v))

        return (M, u)

def Hungary(U0, V0, E):
    U_bef = set()
    M = []
    cnt = 0
    length = len(U0)
    
    # CALC
    while True:

        # INIT
        (U, V) = get_nodes(M)
        u = U0[cnt%length]
        cnt = cnt + 1
        S0 = set()
        S0.add(u)
        T0 = set()

        # 終了条件(すべての頂点から始まる増大道を調べて、マッチング数が変わっていなければ、それ以上増えることはない)
        if cnt % length == 0:
            if U.difference(U_bef) == set():
                break
            U_bef = U.copy()

        # 枝刈り(すでにマッチングしている頂点から始まる道は増大道ではない)
        if u in U:
            continue
        
        #CALC
        (M, u) = augpath(S0, T0, M, E)
    
    return M

#今回の問題分---------------------------------------------------------------------

n=int(input())
s=[]
ss=[]
for i in range(n):
    s.append([int(i) for i in input().split()])
for i in range(n):
    ss.append([int(i) for i in input().split()])
s.sort(key=lambda x:(x[0],x[1]))
ss.sort(key=lambda x:(x[0],x[1]))
#print(s,ss)

pair=[]
for i in range(len(s)):
    right=[]
    for j in range(n):    
        if s[j][0]<ss[i][0] and s[j][1]<ss[i][1]:
            right.append(j)
    pair.append(right)
pair.sort(key=lambda x:len(x))

#---------------------------------------------------------------------

E = []
U0 = [x for x in range(len(pair))]
V0 = [x for x in range(n)]


for i in range(len(pair)):
    try:
        for j in range(len(pair[i])):
            node1, node2 = i,pair[i][j]
            E.append((node1, node2))
    except:
        pass

# CALC
edges = Hungary(U0, V0, E)

# OUTPUT
print(len(edges))

Buggy submission (verdict: Runtime Error):
#http://nobuta05.hatenablog.com/entry/2015/11/12/013938
"""
==================================================
二部グラフの最大マッチングを求めるプログラム

# 条件
G = (U,V,E) |U| = |E|

# 入力
U --> 1 2 3 4 .... 10
V --> a b c d .... j
E -->
1 a
1 d
2 b
2 f
etc...

# 出力(最大マッチングの一例)
edges = [(1,a), (2,b), (3 h), ...]

# 定義
def get_edge(M, u, U)    枝集合Mに含まれる枝のうち、頂点uと接している枝の他方の頂点_(1)を返す関数。そのような枝
                         が複数ある場合、一番最初に見つかった頂点を返す。また、引数にUが指定されている場合、(1)
                         を満たし、かつUに含まれる頂点を返す。

def get_nodes(M)         枝集合Mに含まれる枝に関して、左接続の頂点と右接続の頂点に分け、左接続の頂点集合と右接続
                         の頂点集合の二つを返す関数。

def augpath(Si, Ti, M, E)
                         二部グラフにおいて、増大道があるかどうかを判定・処理するプログラム。

==================================================
"""

def get_edge(M, u, U=None):

    if U == None:
        for x in M[:]:
            if x[0] == u:
                return x[1]
            elif x[1] == u:
                return x[0]

    else:
        for x in M[:]:
            if x[0] == u and x[1] in U:
                return x[1]
            elif x[1] == u and x[0] in U:
                return x[0]
    return None

def get_nodes(M):
    U = set()
    V = set()
    for x in M:
        U.add(x[0])
        V.add(x[1])
    return (U, V)


def augpath(Si, Ti, M, E):
    (U, V) = get_nodes(M)
    Ti1 = set(x[1] for x in E if x[0] in Si and not x[1] in Ti)
    
    if Ti1 == set():
        return (M, None)

    elif Ti1.difference(V) == set():
        Si1 = set(x[0] for x in M if x[1] in Ti1)
        (M, ui1) = augpath(Si1, Ti.union(Ti1), M, E)

        if ui1 == None:
            return (M, None)

        else:
            v = get_edge(M, ui1)
            M.remove((ui1,v))

            ui = get_edge(E, v, U=Si)
            M.append((ui,v))
            
            return (M, ui)

    else:
        v = Ti1.difference(V).pop()
        u = get_edge(E, v, U=Si)
        M.append((u,v))

        return (M, u)

def Hungary(U0, V0, E):
    U_bef = set()
    M = []
    cnt = 0
    length = len(U0)
    
    # CALC
    while True:

        # INIT
        (U, V) = get_nodes(M)
        u = U0[cnt%length]
        cnt = cnt + 1
        S0 = set()
        S0.add(u)
        T0 = set()

        # 終了条件(すべての頂点から始まる増大道を調べて、マッチング数が変わっていなければ、それ以上増えることはない)
        if cnt % length == 0:
            if U.difference(U_bef) == set():
                break
            U_bef = U.copy()

        # 枝刈り(すでにマッチングしている頂点から始まる道は増大道ではない)
        if u in U:
            continue
        
        #CALC
        (M, u) = augpath(S0, T0, M, E)
    
    return M

#今回の問題分---------------------------------------------------------------------

n=int(input())
s=[]
ss=[]
for i in range(n):
    s.append([int(i) for i in input().split()])
for i in range(n):
    ss.append([int(i) for i in input().split()])
s.sort(key=lambda x:(x[0],x[1]))
ss.sort(key=lambda x:(x[0],x[1]))
#print(s,ss)

pair=[]
for i in range(len(s)):
    right=[]
    for j in range(n):    
        if s[j][0]<ss[i][0] and s[j][1]<ss[i][1]:
            right.append(j)
    pair.append(right)
pair.sort(key=lambda x:len(x))

#---------------------------------------------------------------------

E = []
U0 = [x for x in range(len(pair))]
V0 = [x for x in range(n)]

try:
    for i in range(len(pair)):
        for j in range(len(pair[i])):
            node1, node2 = i,pair[i][j]
            E.append((node1, node2))
except:
    pass

# CALC

edges = Hungary(U0, V0, E)

# OUTPUT
#print('MAX MATCHING ... {0}'.format(len(edges)))

print(len(edges))

Buggy submission (verdict: Runtime Error):
n=int(input())
s=[]
ss=[]
for i in range(n):
    s.append([int(i) for i in input().split()])
for i in range(n):
    ss.append([int(i) for i in input().split()])
s.sort(key=lambda x:(x[0],x[1]))
ss.sort(key=lambda x:(x[0],x[1]))
#print(s,ss)

pair=[]
for i in range(len(s)):
    right=[]
    for j in range(n):    
        if s[j][0]<ss[i][0] and s[j][1]<ss[i][1]:
            right.append(j)
    pair.append(right)
pair.sort(key=lambda x:len(x))

#http://nobuta05.hatenablog.com/entry/2015/11/12/013938
"""
==================================================
二部グラフの最大マッチングを求めるプログラム

# 条件
G = (U,V,E) |U| = |E|

# 入力
U --> 1 2 3 4 .... 10
V --> a b c d .... j
E -->
1 a
1 d
2 b
2 f
etc...

# 出力(最大マッチングの一例)
edges = [(1,a), (2,b), (3 h), ...]

# 定義
def get_edge(M, u, U)    枝集合Mに含まれる枝のうち、頂点uと接している枝の他方の頂点_(1)を返す関数。そのような枝
                         が複数ある場合、一番最初に見つかった頂点を返す。また、引数にUが指定されている場合、(1)
                         を満たし、かつUに含まれる頂点を返す。

def get_nodes(M)         枝集合Mに含まれる枝に関して、左接続の頂点と右接続の頂点に分け、左接続の頂点集合と右接続
                         の頂点集合の二つを返す関数。

def augpath(Si, Ti, M, E)
                         二部グラフにおいて、増大道があるかどうかを判定・処理するプログラム。

==================================================
"""

def get_edge(M, u, U=None):

    if U == None:
        for x in M[:]:
            if x[0] == u:
                return x[1]
            elif x[1] == u:
                return x[0]

    else:
        for x in M[:]:
            if x[0] == u and x[1] in U:
                return x[1]
            elif x[1] == u and x[0] in U:
                return x[0]
    return None

def get_nodes(M):
    U = set()
    V = set()
    for x in M:
        U.add(x[0])
        V.add(x[1])
    return (U, V)


def augpath(Si, Ti, M, E):
    (U, V) = get_nodes(M)
    Ti1 = set(x[1] for x in E if x[0] in Si and not x[1] in Ti)
    
    if Ti1 == set():
        return (M, None)

    elif Ti1.difference(V) == set():
        Si1 = set(x[0] for x in M if x[1] in Ti1)
        (M, ui1) = augpath(Si1, Ti.union(Ti1), M, E)

        if ui1 == None:
            return (M, None)

        else:
            v = get_edge(M, ui1)
            M.remove((ui1,v))

            ui = get_edge(E, v, U=Si)
            M.append((ui,v))
            
            return (M, ui)

    else:
        v = Ti1.difference(V).pop()
        u = get_edge(E, v, U=Si)
        M.append((u,v))

        return (M, u)
    
    

def Hungary(U0, V0, E):
    U_bef = set()
    M = []
    cnt = 0
    length = len(U0)
    
    # CALC
    while True:

        # INIT
        (U, V) = get_nodes(M)
        u = U0[cnt%length]
        cnt = cnt + 1
        S0 = set()
        S0.add(u)
        T0 = set()

        # 終了条件(すべての頂点から始まる増大道を調べて、マッチング数が変わっていなければ、それ以上増えることはない)
        if cnt % length == 0:
            if U.difference(U_bef) == set():
                break
            U_bef = U.copy()

        # 枝刈り(すでにマッチングしている頂点から始まる道は増大道ではない)
        if u in U:
            continue
        
        #CALC
        (M, u) = augpath(S0, T0, M, E)
    
    return M



E = []
U0 = [x for x in range(len(pair))]
V0 = [x for x in range(n)]

for i in range(len(pair)):
    for j in range(len(pair[i])):
        node1, node2 = i,pair[i][j]
        E.append((node1, node2))

# CALC

edges = Hungary(U0, V0, E)

# OUTPUT
#print('MAX MATCHING ... {0}'.format(len(edges)))

print(len(edges))