Problem:


Score : 700 points

Problem Statement

In the State of Takahashi in AtCoderian Federation, there are N cities, numbered 1, 2, ..., N.
M bidirectional roads connect these cities.
The i-th road connects City A_i and City B_i.
Every road connects two distinct cities.
Also, for any two cities, there is at most one road that directly connects them.

One day, it was decided that the State of Takahashi would be divided into two states, Taka and Hashi.
After the division, each city in Takahashi would belong to either Taka or Hashi.
It is acceptable for all the cities to belong Taka, or for all the cities to belong Hashi.
Here, the following condition should be satisfied:

Any two cities in the same state, Taka or Hashi, are directly connected by a road.

Find the minimum possible number of roads whose endpoint cities belong to the same state.
If it is impossible to divide the cities into Taka and Hashi so that the condition is satisfied, print -1.

Constraints

2 \leq N \leq 700

0 \leq M \leq N(N-1)/2

1 \leq A_i \leq N

1 \leq B_i \leq N

A_i \neq B_i

If i \neq j, at least one of the following holds: A_i \neq A_j and B_i \neq B_j.

If i \neq j, at least one of the following holds: A_i \neq B_j and B_i \neq A_j.

Input

Input is given from Standard Input in the following format:

N M
A_1 B_1
A_2 B_2
:
A_M B_M

Output

Print the answer.

Sample Input 1

5 5
1 2
1 3
3 4
3 5
4 5

Sample Output 1

4

For example, if the cities 1, 2 belong to Taka and the cities 3, 4, 5 belong to Hashi, the condition is satisfied.
Here, the number of roads whose endpoint cities belong to the same state, is 4.

Sample Input 2

5 1
1 2

Sample Output 2

-1

In this sample, the condition cannot be satisfied regardless of which cities belong to each state.

Sample Input 3

4 3
1 2
1 3
2 3

Sample Output 3

3

Sample Input 4

10 39
7 2
7 1
5 6
5 8
9 10
2 8
8 7
3 10
10 1
8 10
2 3
7 4
3 9
4 10
3 4
6 1
6 7
9 5
9 7
6 9
9 4
4 6
7 5
8 3
2 5
9 2
10 7
8 6
8 9
7 3
5 3
4 5
6 3
2 10
5 10
4 2
6 2
8 4
10 6

Sample Output 4

21



Sample input:
5 5
1 2
1 3
3 4
3 5
4 5


Sample output:
4

Submission status: Runtime Error

Solution code:
def main():
    n, m = map(int, input().split())
    ab = [list(map(int, input().split())) for _ in [0]*m]

    n_hozon = n
    g = [set() for _ in [0]*n]
    [g[a-1].add(b-1) for a, b in ab]
    [g[b-1].add(a-1) for a, b in ab]
    s = set([i for i in range(n)])

    # 題意：できるだけ辺が少なくなるように2つの完全グラフに分けたい
    # まず、2つの完全グラフに分けられるのかを調べる必要がある。
    # これが不可能であれば-1を出力して即座に終了。

    # 分けられるのであれば、できるだけ街の数を均等にしたい。
    # 以下のアルゴリズムを考える。
    # まずお互い繋がっていない都市の組A,Bを選ぶ。
    # すると、Aにしか繋がっていない街、Bにしか繋がっていない街、
    # 両方に繋がっている街の3グループに分かれる。
    # このうち、最後のグループについて、
    # 最初に戻って同様の処理を行う。
    # 街がなくなるか、両方に繋がっている街のグループが完全グラフに
    # なった時点で終了。

    # 上記のアルゴリズムにより、(A,B)の列と最後に残った街の数が求められる。
    # これをもとに、動的計画法で考えられる街の振り分け方を全て洗い出す。
    # 最も均等に近く振り分けた場合について、完全グラフの辺数を計算して
    # 出力すれば問題が解ける。
    a_list, b_list = [], []
    while True:
        #print(n, m)
        # print(s)
        # print(g)
        # 完全グラフかどうか
        if m == n*(n-1)//2:
            break
        a = [-1, 10**10]
        for i in s:
            l = len(g[i])
            if l < a[1]:
                a = [i, l]
        a = a[0]
        b = [-1, 10**10]
        for i in s-g[a]-{a}:
            l = len(g[i])
            if l < b[1]:
                b = [i, l]
        b = b[0]
        a_set, b_set = {a}, {b}
        s.remove(a)
        s.remove(b)
        n -= 2
        remain = set()
        for i in s:
            flag_a = True
            for j in a_set:
                if j not in g[i]:
                    flag_a = False
                else:
                    g[i].remove(j)
                    g[j].remove(i)
                    m -= 1
            flag_b = True
            for j in b_set:
                if j not in g[i]:
                    flag_b = False
                else:
                    g[i].remove(j)
                    g[j].remove(i)
                    m -= 1
            #print(i, flag_a, flag_b)
            if flag_a == flag_b == False:
                print(-1)
                return
            elif flag_a == False or flag_b == False:
                # print(remain)
                q = {i}
                flag = flag_a  # True→Aに入れる
                while q:
                    #print(q, "q")
                    qq = set()
                    while q:
                        i = q.pop()
                        if flag:
                            a_set.add(i)
                        else:
                            b_set.add(i)
                        for j in remain:
                            if j not in g[i]:
                                qq.add(j)
                            else:
                                g[i].remove(j)
                                g[j].remove(i)
                                m -= 1
                    for i in qq:
                        remain.remove(i)
                    flag ^= True
                    q = qq
            else:
                remain.add(i)
            # print(g)
        for i in (a_set | b_set)-{a}-{b}:
            s.remove(i)
            n -= 1

        a_list.append(len(a_set))
        b_list.append(len(b_set))

    m = n
    n = n_hozon
    k = len(a_list)

    if k == 1:
        dp = [False]*(n+1)
        a = a_list[0]
        b = b_list[0]
        ans = 10**10
        for i in range(m+1):
            ans = min(ans, (a+i)*(a+i-1)//2+(n-a-i)*(n-a-i-1)//2)
        print(ans)
        return

    dp = [False]*(n+1)
    dp[0] = True

    for i in range(k):
        a, b = a_list[i], b_list[i]
        maxab = max(a, b)
        dp2 = [False]*(n+1)
        for j in range(n-maxab+1):
            dp2[j+a] |= dp[j]
            dp2[j+b] |= dp[j]
        dp = dp2

    dp2 = [False]*(n+1)
    for j in range(m+1):
        for i in range(n-j+1):
            dp2[i+j] |= dp[i]
    ans = 10**10

    for i in range(n+1):
        if dp2[i] is False:
            continue
        j = n-i
        ans = min(ans, i*(i-1)//2+j*(j-1)//2)
    print(ans)


main()

Submission status: Runtime Error

Solution code:
def main():
    n, m = map(int, input().split())
    ab = [list(map(int, input().split())) for _ in [0]*m]

    n_hozon = n
    g = [set() for _ in [0]*n]
    [g[a-1].add(b-1) for a, b in ab]
    [g[b-1].add(a-1) for a, b in ab]
    s = set([i for i in range(n)])

    # 題意：できるだけ辺が少なくなるように2つの完全グラフに分けたい
    # まず、2つの完全グラフに分けられるのかを調べる必要がある。
    # これが不可能であれば-1を出力して即座に終了。

    # 分けられるのであれば、できるだけ街の数を均等にしたい。
    # 以下のアルゴリズムを考える。
    # まずお互い繋がっていない都市の組A,Bを選ぶ。
    # すると、Aにしか繋がっていない街、Bにしか繋がっていない街、
    # 両方に繋がっている街の3グループに分かれる。
    # このうち、最後のグループについて、
    # 最初に戻って同様の処理を行う。
    # 街がなくなるか、両方に繋がっている街のグループが完全グラフに
    # なった時点で終了。

    # 上記のアルゴリズムにより、(A,B)の列と最後に残った街の数が求められる。
    # これをもとに、動的計画法で考えられる街の振り分け方を全て洗い出す。
    # 最も均等に近く振り分けた場合について、完全グラフの辺数を計算して
    # 出力すれば問題が解ける。
    a_list, b_list = [], []
    while True:
        #print(n, m)
        # print(s)
        # print(g)
        # 完全グラフかどうか
        if m == n*(n-1)//2:
            break
        a = [-1, 10**10]
        for i in s:
            l = len(g[i])
            if l < a[1]:
                a = [i, l]
        a = a[0]
        b = [-1, 10**10]
        for i in s-g[a]-{a}:
            l = len(g[i])
            if l < b[1]:
                b = [i, l]
        b = b[0]
        a_set, b_set = {a}, {b}
        s.remove(a)
        s.remove(b)
        n -= 2
        remain = set()
        for i in s:
            flag_a = True
            for j in a_set:
                if j not in g[i]:
                    flag_a = False
                else:
                    g[i].remove(j)
                    g[j].remove(i)
                    m -= 1
            flag_b = True
            for j in b_set:
                if j not in g[i]:
                    flag_b = False
                else:
                    g[i].remove(j)
                    g[j].remove(i)
                    m -= 1
            if flag_a == flag_b == False:
                print(-1)
                return
            elif flag_a == False:
                b_set.add(i)
                temp = []
                for j in remain:
                    if j not in g[i]:
                        temp.append(j)
                    else:
                        g[i].remove(j)
                        g[j].remove(i)
                        m -= 1
                for j in temp:
                    remain.remove(j)
                    a_set.add(j)
            elif flag_b == False:
                a_set.add(i)
                temp = []
                for j in remain:
                    if j not in g[i]:
                        temp.append(j)
                    else:
                        g[i].remove(j)
                        g[j].remove(i)
                        m -= 1
                for j in temp:
                    remain.remove(j)
                    b_set.add(j)
            else:
                remain.add(i)
        remove_list = []
        for i in (a_set | b_set)-{a}-{b}:
            s.remove(i)
            for j in g[i]:
                remove_list.append((i, j))
            n -= 1
        for i, j in remove_list:
            g[i].remove(j)
            g[j].remove(i)
            m -= 1
        a_list.append(len(a_set))
        b_list.append(len(b_set))

    m = n
    n = n_hozon
    k = len(a_list)

    if k == 1:
        dp = [False]*(n+1)
        a = a_list[0]
        b = b_list[0]
        ans = 10**10
        for i in range(m+1):
            ans = min(ans, (a+i)*(a+i-1)//2+(n-a-i)*(n-a-i-1)//2)
        print(ans)
        return

    dp = [False]*(n+1)
    dp[0] = True

    for i in range(k):
        a, b = a_list[i], b_list[i]
        maxab = max(a, b)
        dp2 = [False]*(n+1)
        for j in range(n-maxab+1):
            dp2[j+a] |= dp[j]
            dp2[j+b] |= dp[j]
        dp = dp2

    dp2 = [False]*(n+1)
    for j in range(m+1):
        for i in range(n-j+1):
            dp2[i+j] |= dp[i]
    ans = 10**10

    for i in range(n+1):
        if dp2[i] is False:
            continue
        j = n-i
        ans = min(ans, i*(i-1)//2+j*(j-1)//2)
    print(ans)


main()

Submission status: Runtime Error

Solution code:
def main():
    n, m = map(int, input().split())
    n_hozon = n
    N, M = [n], [m]

    ab = [list(map(int, input().split())) for _ in [0]*m]
    g = [set() for _ in [0]*n]
    [g[a-1].add(b-1) for a, b in ab]
    [g[b-1].add(a-1) for a, b in ab]
    s = set([i for i in range(n)])

    # 題意：できるだけ辺が少なくなるように2つの完全グラフに分けたい
    # まず、2つの完全グラフに分けられるのかを調べる必要がある。
    # これが不可能であれば-1を出力して即座に終了。

    # 分けられるのであれば、できるだけ街の数を均等にしたい。
    # 以下のアルゴリズムを考える。
    # まずお互い繋がっていない都市の組A,Bを選ぶ。
    # すると、Aにしか繋がっていない街、Bにしか繋がっていない街、
    # 両方に繋がっている街の3グループに分かれる。
    # このうち、最後のグループについて、
    # 最初に戻って同様の処理を行う。
    # 街がなくなるか、両方に繋がっている街のグループが完全グラフに
    # なった時点で終了。

    # 上記のアルゴリズムにより、(A,B)の列と最後に残った街の数が求められる。
    # これをもとに、動的計画法で考えられる街の振り分け方を全て洗い出す。
    # 最も均等に近く振り分けた場合について、完全グラフの辺数を計算して
    # 出力すれば問題が解ける。
    a_list, b_list = [], []
    while True:
        n, m = N[0], M[0]
        #print(n, m)
        # print(s)
        # print(g)
        # 完全グラフかどうか
        if m == n*(n-1)//2:
            break
        a = [-1, 10**10]
        for i in s:
            l = len(g[i])
            if l < a[1]:
                a = [i, l]
        a = a[0]
        b = [-1, 10**10]
        for i in s-g[a]-{a}:
            l = len(g[i])
            if l < b[1]:
                b = [i, l]
        b = b[0]
        a_set, b_set = {a}, {b}
        s.remove(a)
        s.remove(b)
        n -= 2
        for i in s:
            flag_a = True
            for j in a_set:
                if j not in g[i]:
                    flag_a = False
                else:
                    g[i].remove(j)
                    g[j].remove(i)
                    m -= 1
            flag_b = True
            for j in b_set:
                if j not in g[i]:
                    flag_b = False
                else:
                    g[i].remove(j)
                    g[j].remove(i)
                    m -= 1
            if flag_a == flag_b == False:
                print(-1)
                return
            elif flag_a == False:
                b_set.add(i)
            elif flag_b == False:
                a_set.add(i)
        remove_list = []
        for i in (a_set | b_set)-{a}-{b}:
            s.remove(i)
            for j in g[i]:
                remove_list.append((i, j))
            n -= 1
        for i, j in remove_list:
            g[i].remove(j)
            g[j].remove(i)
            m -= 1
        a_list.append(len(a_set))
        b_list.append(len(b_set))
        N, M = [n], [m]

    n = n_hozon
    if sum(a_list+b_list)+m != n:
        print(1/0)
    k = len(a_list)

    dp = [False]*(n+1)
    dp[0] = True

    for i in range(k):
        a, b = a_list[i], b_list[i]
        maxab = max(a, b)
        dp2 = [False]*(n+1)
        for j in range(n-maxab+1):
            dp2[j+a] |= dp[j]
            dp2[j+b] |= dp[j]
        dp = dp2

    dp2 = [False]*(n+1)
    for j in range(m+1):
        for i in range(n-j+1):
            dp2[i+j] |= dp[i]
    ans = 10**10

    for i in range(n+1):
        if dp2[i] is False:
            continue
        j = n-i
        ans = min(ans, i*(i-1)//2+j*(j-1)//2)
    print(ans)


main()

Submission status: Runtime Error

Solution code:
def main():
    n, m = map(int, input().split())
    n_hozon = n
    N, M = [n], [m]

    ab = [list(map(int, input().split())) for _ in [0]*m]
    g = [set() for _ in [0]*n]
    [g[a-1].add(b-1) for a, b in ab]
    [g[b-1].add(a-1) for a, b in ab]
    s = set([i for i in range(n)])

    # 題意：できるだけ辺が少なくなるように2つの完全グラフに分けたい
    # まず、2つの完全グラフに分けられるのかを調べる必要がある。
    # これが不可能であれば-1を出力して即座に終了。

    # 分けられるのであれば、できるだけ街の数を均等にしたい。
    # 以下のアルゴリズムを考える。
    # まずお互い繋がっていない都市の組A,Bを選ぶ。
    # すると、Aにしか繋がっていない街、Bにしか繋がっていない街、
    # 両方に繋がっている街の3グループに分かれる。
    # このうち、最後のグループについて、
    # 最初に戻って同様の処理を行う。
    # 街がなくなるか、両方に繋がっている街のグループが完全グラフに
    # なった時点で終了。

    # 上記のアルゴリズムにより、(A,B)の列と最後に残った街の数が求められる。
    # これをもとに、動的計画法で考えられる街の振り分け方を全て洗い出す。
    # 最も均等に近く振り分けた場合について、完全グラフの辺数を計算して
    # 出力すれば問題が解ける。
    a_list, b_list = [], []
    while True:
        n, m = N[0], M[0]
        #print(n, m)
        # print(s)
        # print(g)
        # 完全グラフかどうか
        if m == n*(n-1)//2:
            break
        a = [-1, 10**10]
        for i in s:
            l = len(g[i])
            if l < a[1]:
                a = [i, l]
        a = a[0]
        b = [-1, 10**10]
        for i in s-g[a]-{a}:
            l = len(g[i])
            if l < b[1]:
                b = [i, l]
        b = b[0]
        a_set, b_set = {a}, {b}
        s.remove(a)
        s.remove(b)
        n -= 2
        for i in s:
            flag_a = True
            for j in a_set:
                if j not in g[i]:
                    flag_a = False
                else:
                    g[i].remove(j)
                    g[j].remove(i)
                    m -= 1
            flag_b = True
            for j in b_set:
                if j not in g[i]:
                    flag_b = False
                else:
                    g[i].remove(j)
                    g[j].remove(i)
                    m -= 1
            if flag_a == flag_b == False:
                print(-1)
                return
            elif flag_a == False:
                b_set.add(i)
            elif flag_b == False:
                a_set.add(i)
        remove_list = []
        for i in (a_set | b_set)-{a}-{b}:
            s.remove(i)
            for j in g[i]:
                remove_list.append((i, j))
            n -= 1
        for i, j in remove_list:
            g[i].remove(j)
            g[j].remove(i)
            m -= 1
        a_list.append(len(a_set))
        b_list.append(len(b_set))
        N, M = [n], [m]

    n = n_hozon
    k = len(a_list)
    if k != 1:
        print(1/0)
    dp = [False]*(n+1)
    dp[0] = True

    for i in range(k):
        a, b = a_list[i], b_list[i]
        maxab = max(a, b)
        dp2 = [False]*(n+1)
        for j in range(n-maxab+1):
            dp2[j+a] |= dp[j]
            dp2[j+b] |= dp[j]
        dp = dp2

    dp2 = [False]*(n+1)
    for j in range(m+1):
        for i in range(n-j+1):
            dp2[i+j] |= dp[i]
    ans = 10**10

    for i in range(n+1):
        if dp2[i] is False:
            continue
        j = n-i
        ans = min(ans, i*(i-1)//2+j*(j-1)//2)
    print(ans)


main()

Submission status: Runtime Error

Solution code:
def main():
    n, m = map(int, input().split())
    n_hozon = n
    N, M = [n], [m]

    ab = [list(map(int, input().split())) for _ in [0]*m]
    g = [set() for _ in [0]*n]
    [g[a-1].add(b-1) for a, b in ab]
    [g[b-1].add(a-1) for a, b in ab]
    s = set([i for i in range(n)])

    # 題意：できるだけ辺が少なくなるように2つの完全グラフに分けたい
    # まず、2つの完全グラフに分けられるのかを調べる必要がある。
    # これが不可能であれば-1を出力して即座に終了。

    # 分けられるのであれば、できるだけ街の数を均等にしたい。
    # 以下のアルゴリズムを考える。
    # まずお互い繋がっていない都市の組A,Bを選ぶ。
    # すると、Aにしか繋がっていない街、Bにしか繋がっていない街、
    # 両方に繋がっている街の3グループに分かれる。
    # このうち、最後のグループについて、
    # 最初に戻って同様の処理を行う。
    # 街がなくなるか、両方に繋がっている街のグループが完全グラフに
    # なった時点で終了。

    # 上記のアルゴリズムにより、(A,B)の列と最後に残った街の数が求められる。
    # これをもとに、動的計画法で考えられる街の振り分け方を全て洗い出す。
    # 最も均等に近く振り分けた場合について、完全グラフの辺数を計算して
    # 出力すれば問題が解ける。
    a_list, b_list = [], []
    while True:
        n, m = N[0], M[0]
        #print(n, m)
        # print(s)
        # print(g)
        # 完全グラフかどうか
        if m == n*(n-1)//2:
            break
        a = [-1, 10**10]
        for i in s:
            l = len(g[i])
            if l < a[1]:
                a = [i, l]
        a = a[0]
        b = [-1, 10**10]
        for i in s-g[a]-{a}:
            l = len(g[i])
            if l < b[1]:
                b = [i, l]
        b = b[0]
        a_set, b_set = {a}, {b}
        s.remove(a)
        s.remove(b)
        n -= 2
        for i in s:
            flag_a = True
            for j in a_set:
                if j not in g[i]:
                    flag_a = False
                else:
                    g[i].remove(j)
                    g[j].remove(i)
                    m -= 1
            flag_b = True
            for j in b_set:
                if j not in g[i]:
                    flag_b = False
                else:
                    g[i].remove(j)
                    g[j].remove(i)
                    m -= 1
            if flag_a == flag_b == False:
                print(-1)
                return
            elif flag_a == False:
                b_set.add(i)
            elif flag_b == False:
                a_set.add(i)
        remove_list = []
        for i in (a_set | b_set)-{a}-{b}:
            s.remove(i)
            for j in g[i]:
                remove_list.append((i, j))
            n -= 1
        for i, j in remove_list:
            g[i].remove(j)
            g[j].remove(i)
            m -= 1
        a_list.append(len(a_set))
        b_list.append(len(b_set))
        N, M = [n], [m]

    n = n_hozon
    k = len(a_list)
    if k > 1:
        print(1/0)
    dp = [False]*(n+1)
    dp[0] = True

    for i in range(k):
        a, b = a_list[i], b_list[i]
        maxab = max(a, b)
        dp2 = [False]*(n+1)
        for j in range(n-maxab+1):
            dp2[j+a] |= dp[j]
            dp2[j+b] |= dp[j]
        dp = dp2

    dp2 = [False]*(n+1)
    for j in range(m+1):
        for i in range(n-j+1):
            dp2[i+j] |= dp[i]
    ans = 10**10

    for i in range(n+1):
        if dp2[i] is False:
            continue
        j = n-i
        ans = min(ans, i*(i-1)//2+j*(j-1)//2)
    print(ans)


main()

Submission status: Runtime Error

Solution code:
def main():
    n, m = map(int, input().split())
    n_hozon = n
    N, M = [n], [m]

    ab = [list(map(int, input().split())) for _ in [0]*m]
    g = [set() for _ in [0]*n]
    [g[a-1].add(b-1) for a, b in ab]
    [g[b-1].add(a-1) for a, b in ab]
    s = set([i for i in range(n)])

    # 題意：できるだけ辺が少なくなるように2つの完全グラフに分けたい
    # まず、2つの完全グラフに分けられるのかを調べる必要がある。
    # これが不可能であれば-1を出力して即座に終了。

    # 分けられるのであれば、できるだけ街の数を均等にしたい。
    # 以下のアルゴリズムを考える。
    # まずお互い繋がっていない都市の組A,Bを選ぶ。
    # すると、Aにしか繋がっていない街、Bにしか繋がっていない街、
    # 両方に繋がっている街の3グループに分かれる。
    # このうち、最後のグループについて、
    # 最初に戻って同様の処理を行う。
    # 街がなくなるか、両方に繋がっている街のグループが完全グラフに
    # なった時点で終了。

    # 上記のアルゴリズムにより、(A,B)の列と最後に残った街の数が求められる。
    # これをもとに、動的計画法で考えられる街の振り分け方を全て洗い出す。
    # 最も均等に近く振り分けた場合について、完全グラフの辺数を計算して
    # 出力すれば問題が解ける。
    a_list, b_list = [], []
    while True:
        n, m = N[0], M[0]
        #print(n, m)
        # print(s)
        # print(g)
        # 完全グラフかどうか
        if m == n*(n-1)//2:
            break
        a = [-1, 10**10]
        for i in s:
            l = len(g[i])
            if l < a[1]:
                a = [i, l]
        a = a[0]
        b = [-1, 10**10]
        for i in s-g[a]-{a}:
            l = len(g[i])
            if l < b[1]:
                b = [i, l]
        b = b[0]
        a_set, b_set = {a}, {b}
        s.remove(a)
        s.remove(b)
        n -= 2
        for i in s:
            flag_a = True
            for j in a_set:
                if j not in g[i]:
                    flag_a = False
                else:
                    g[i].remove(j)
                    g[j].remove(i)
                    m -= 1
            flag_b = True
            for j in b_set:
                if j not in g[i]:
                    flag_b = False
                else:
                    g[i].remove(j)
                    g[j].remove(i)
                    m -= 1
            if flag_a == flag_b == False:
                print(-1)
                return
            elif flag_a == False:
                b_set.add(i)
            elif flag_b == False:
                a_set.add(i)
        remove_list = []
        for i in (a_set | b_set)-{a}-{b}:
            s.remove(i)
            for j in g[i]:
                remove_list.append((i, j))
            n -= 1
        for i, j in remove_list:
            g[i].remove(j)
            g[j].remove(i)
            m -= 1
        a_list.append(len(a_set))
        b_list.append(len(b_set))
        N, M = [n], [m]
    print(1/0)
    n = n_hozon
    k = len(a_list)
    dp = [False]*(n+1)
    dp[0] = True

    for i in range(k):
        a, b = a_list[i], b_list[i]
        maxab = max(a, b)
        dp2 = [False]*(n+1)
        for j in range(n-maxab+1):
            dp2[j+a] |= dp[j]
            dp2[j+b] |= dp[j]
        dp = dp2

    dp2 = [False]*(n+1)
    for j in range(m+1):
        for i in range(n-j+1):
            dp2[i+j] |= dp[i]
    ans = 10**10

    for i in range(n+1):
        if dp2[i] is False:
            continue
        j = n-i
        ans = min(ans, i*(i-1)//2+j*(j-1)//2)
    print(ans)


main()

Submission status: Runtime Error

Solution code:

"""

m,k 個に分けた場合 mC2+kC2が答え
分けられるもののうち、差が最小となるようにする。

aとbがつながってない
a = 0とすると b = 1
bとcがつながってない c = 0

ぐる0,ぐる1で分かれる
x,y個に分かれた時、abs(x-y)として差を最小にする。

dpで求めればいいか。


[親、所属(0or1)]でunion-find
"""

import math
import sys

def nCr(n,r):
    if n-r < 0:
        return 0
    return math.factorial(n) // (math.factorial(n-r) * math.factorial(r))

def wantp(n):

    wpl = []
    while n != p[n]:
        wpl.append(n)
        n = p[n]

    for i in wpl:
        p[i] = n
    return n


N,M = map(int,input().split())

link = [[False] * N for i in range(N)]

for i in range(M):

    A,B = map(int,input().split())
    A -= 1
    B -= 1

    link[A][B] = True
    link[B][A] = True


#まず普通にUnion-Findする
p = [i for i in range(N)]
rank = [1] * N
for i in range(N):

    for j in range(N):

        if not link[i][j]:
            ip = wantp(i)
            jp = wantp(j)

            if ip != jp:

                if rank[ip] > rank[jp]:
                    p[jp] = ip
                elif rank[ip] < rank[jp]:
                    p[ip] = jp
                else:
                    p[ip] = jp
                    rank[jp] += 1
                    

dic = {} #親がキーで配列内に属す頂点を入れる(これで分類)

for i in range(N):
    np = wantp(i)
    if np not in dic:
        dic[np] = []
    dic[np].append(i)


side = [-1] * N #0と1サイドで分ける

for i in dic: #整合性の確認

    side[i] = 0
    for j in dic[i]:

        for k in dic[i]:
            if j != k and (not link[j][k]) and side[k] != -1:

                if side[j] == -1:
                    side[j] = side[k] ^ 1
                elif side[j] == side[k]:
                    print (-1)
                    sys.exit()


dic2 = {} #キーは先頭[0の数-1の数を入れる]

for i in range(N):
    np = wantp(i)
    if np not in dic2:
        dic2[np] = 0

    if side[i] == 0:
        dic2[np] += 1
    else:
        dic2[np] -= 1

allnum = [False] * 1401
allnum[0] = [True]

for i in dic2:
    dic2[i] = abs(dic2[i])

#dpで中心に一番近い数を求める

for i in dic2:
    nn = dic2[i]
    nall = [False] * 1401
    
    for j in range(1401):
        if allnum[j]:
            nall[j + nn] = True
            nall[j - nn] = True

    allnum = nall.copy()

ret = 0
while True:
    if allnum[ret] or allnum[-1 * ret]:
        break
    ret += 1

a = N
b = 0

while abs(a-b) != ret:
    a -= 1
    b += 1

ans = 0
print (nCr(a,2) + nCr(b,2))

Submission status: Runtime Error

Solution code:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef vector<int> vi;
typedef vector<ll> vll;
#define pb push_back
#define mp make_pair
#define eps 1e-9
#define INF 2000000000
#define LLINF 1000000000000000ll
#define sz(x) ((int)(x).size())
#define fi first
#define sec second
#define all(x) (x).begin(),(x).end()
#define sq(x) ((x)*(x))
#define rep(i,n) for(int (i)=0;(i)<(int)(n);(i)++)
#define repn(i,a,n) for(int (i)=(a);(i)<(int)(n);(i)++)
#define EQ(a,b) (abs((a)-(b))<eps)
template<class T> void chmin(T& a,const T& b){if(a>b)a=b;}
template<class T> void chmax(T& a,const T& b){if(a<b)a=b;}
int e[705][705];
vector<int> g[705];
int col[705];
int cnt[3];
int used[705];
int dp[705][705];
bool binary_graph(int v,int c){
	col[v]=c;
	cnt[c+1]++;
	used[v]=true;
	bool ret = true;
	for(int i=0;i<g[v].size();i++){
		int u = g[v][i];
		if(used[u]){
			if(col[u]==c){
				ret=false;
			}
		}else{
			if(!binary_graph(u,-c))ret=false;
		}
	}
	return ret;
}
int N,M;
int main(){
	cin >> N >> M;
	for(int i=0;i<N;i++){
		e[i][i]=1;
	}
	for(int i=0;i<M;i++){
		int A,B;
		cin >> A >> B;
		A--;B--;
		e[A][B]=e[B][A]=1;
	}
	for(int i=0;i<N;i++){
		for(int j=0;j<N;j++){
			if(e[i][j]==0){
				g[i].pb(j);
			}
		}
	}
	bool res = true;
	vector<int> vec;
	for(int i=0;i<N;i++){
		if(!used[i]){
			cnt[0]=cnt[2]=0;
			res &= binary_graph(i,1);
			//cout << res << endl;
			//cout << cnt[0] << ' ' << cnt[2] << endl;
			if(res)vec.pb(abs(cnt[0]-cnt[2]));
		}
	}
	if(!res){
		cout << -1 << endl;
		return 0;
	}
	dp[0][0]=1;
	for(int i=0;i<vec.size();i++){
		for(int j=0;j<=700;j++){
			if(dp[i][j]==0)continue;
			if(j-vec[i]>=0)dp[i+1][j-vec[i]]=1;
			dp[i+1][j+vec[i]]=1;
		}
	}
	int K = -1;
	for(int i=0;i<=700;i++){
		if(dp[vec.size()][i]==1){
			K=i;
			break;
		}
	}
	//cout << K << endl;
	int C = (N+K)/2,D = (N-K)/2;
	cout << C*(C-1)/2+D*(D-1)/2 << endl;
	return 0;
}

Submission status: Runtime Error

Solution code:
import bisect

N,M = map(int,input().split())
bridge = [[True for _ in range(N)] for _ in range(N)]
for i in range(M):
    a,b = map(int,input().split())
    bridge[a-1][b-1] = False
    bridge[b-1][a-1] = False
connect = [[] for _ in range(N)]
for i in range(N):
    for j in range(N):
        if bridge[i][j] and i!=j:
            connect[i].append(j)
flg = False
used = [False for i in range(N)]
seplist = []
for i in range(N):
    if not used[i]:
        consider = [i]
        g = [0 for i in range(N)]
        g[i] = 1
        while consider != []:
            j = consider.pop()
            used[j] = True
            for k in connect[j]:
                if g[k] == 0:
                    g[k] = -1 if g[j] == 1 else 1
                    consider.append(k)
                elif g[k]*g[j]==1:
                    flg = True
                    break
                
            if flg:
                break
        if flg:
            break
        else:
            seplist.append([g.count(1),g.count(-1)])
            

if flg:
    print(-1)
else:
    canmake = set()
    for s in seplist:
        if canmake == set():
            canmake.add(s[0])
            canmake.add(s[1])
        else:
            c = canmake.copy()
            for j in c:
                canmake.add(s[0]+j)
                canmake.add(s[1]+j)
    cc = list(canmake)
    cc.sort()
    pos = bisect.bisect_right(cc,N//2)
    left = cc[pos]
    right = N-left
    print(left*(left-1)//2+right*(right-1)//2)