Two Heaps (Codeforces 353B)
Tags: Implement, combinatorics

Valera has 2·n cubes, each cube contains an integer from 10 to 99. He arbitrarily chooses n cubes and puts them in the first heap. The remaining cubes form the second heap. Valera decided to play with cubes. During the game he takes a cube from the first heap and writes down the number it has. Then he takes a cube from the second heap and write out its two digits near two digits he had written (to the right of them). In the end he obtained a single fourdigit integer — the first two digits of it is written on the cube from the first heap, and the second two digits of it is written on the second cube from the second heap.Valera knows arithmetic very well. So, he can easily count the number of distinct fourdigit numbers he can get in the game. The other question is: how to split cubes into two heaps so that this number (the number of distinct fourdigit integers Valera can get) will be as large as possible?

Input: The first line contains integer n (1 ≤ n ≤ 100). The second line contains 2·n space-separated integers ai (10 ≤ ai ≤ 99), denoting the numbers on the cubes.

Output: In the first line print a single number — the maximum possible number of distinct four-digit numbers Valera can obtain. In the second line print 2·n numbers bi (1 ≤ bi ≤ 2). The numbers mean: the i-th cube belongs to the bi-th heap in your division.If there are multiple optimal ways to split the cubes into the heaps, print any of them.

Note: In the first test case Valera can put the first cube in the first heap, and second cube — in second heap. In this case he obtain number 1099. If he put the second cube in the first heap, and the first cube in the second heap, then he can obtain number 9910. In both cases the maximum number of distinct integers is equal to one.In the second test case Valera can obtain numbers 1313, 1345, 2413, 2445. Note, that if he put the first and the third cubes in the first heap, he can obtain only two numbers 1324 and 1345.

Sample Inputs:
1
10 99

2
13 24 13 45

Sample Outputs:
1
2 1

4
1 2 2 1

Solution (Python):
n = 2 * int(input())
t = list(map(int, input().split()))
(q, p, c) = ([0] * n, [0] * 100, [0, 0])
for i in t:
    p[i] += 1
for i in p:
    if i:
        c[i == 1] += 1
k = c[1] // 2
print((c[0] + k) * (c[0] + c[1] - k))
r = sorted([(p[j], j, i) for (i, j) in enumerate(t)])
for i in r[0:n:2]:
    q[i[2]] = '1'
for i in r[1:n:2]:
    q[i[2]] = '2'
print(' '.join(q))

Solution (Python):
n = int(input())
a = list(map(int, input().split()))
g = [[] for i in range(100)]
for i in range(2 * n):
    g[a[i]].append(i)
x = [0, 0]
cur = 1
mus = []
ans = [0] * 2 * n
for i in range(10, 100):
    if len(g[i]) == 1:
        ans[g[i][0]] = cur
        x[cur - 1] += 1
        cur = 3 - cur
    if len(g[i]) >= 2:
        ans[g[i][0]] = 1
        ans[g[i][1]] = 2
        x[0] += 1
        x[1] += 1
        for j in range(2, len(g[i])):
            mus.append(g[i][j])
for i in range(len(mus)):
    ans[mus[i]] = 2 - i % 2
print(x[0] * x[1])
print(*ans)

Solution (Python):
from collections import defaultdict as defdict
n = int(input())
d = defdict(int)
a = list(map(int, input().split()))
for i in a:
    d[i] += 1
b = set((i for i in a if d[i] > 1))
c = [i for i in a if d[i] == 1]
mx = len(b) + len(c) // 2
d = defdict(int)
answ = [None] * (n * 2)
cnt_diff_1 = 0
cnt_diff_2 = 0
for (i, x) in enumerate(a):
    if cnt_diff_1 < mx:
        if x in b:
            if d[x] == 0:
                answ[i] = '1'
                d[x] += 1
                cnt_diff_1 += 1
for (i, x) in enumerate(a):
    if cnt_diff_1 < mx:
        if x not in b:
            answ[i] = '1'
            cnt_diff_1 += 1
cnt1 = cnt_diff_1
cnt2 = 0
for (i, x) in enumerate(a):
    if answ[i] == '1':
        continue
    if x in b:
        if d[x] == 0:
            answ[i] = '2'
            d[x] = 2
            cnt_diff_2 += 1
            cnt2 += 1
        elif d[x] == 1:
            answ[i] = '2'
            d[x] += 1
            cnt_diff_2 += 1
            cnt2 += 1
        elif cnt1 < n:
            answ[i] = '1'
            cnt1 += 1
        else:
            answ[i] = '2'
            cnt2 += 1
    elif cnt2 < n:
        answ[i] = '2'
        cnt2 += 1
        cnt_diff_2 += 1
    else:
        answ[i] = '1'
        cnt1 += 1
        cnt_diff_1 += 1
print(cnt_diff_1 * cnt_diff_2)
print(' '.join(answ))