Arpa's weak amphitheater and Mehrdad's valuable Hoses (Codeforces 741B)
Tags: DP, dsu

Just to remind, girls in Arpa's land are really nice.Mehrdad wants to invite some Hoses to the palace for a dancing party. Each Hos has some weight wi and some beauty bi. Also each Hos may have some friends. Hoses are divided in some friendship groups. Two Hoses x and y are in the same friendship group if and only if there is a sequence of Hoses a1, a2, ..., ak such that ai and ai + 1 are friends for each 1 ≤ i < k, and a1 = x and ak = y.  Arpa allowed to use the amphitheater of palace to Mehrdad for this party. Arpa's amphitheater can hold at most w weight on it. Mehrdad is so greedy that he wants to invite some Hoses such that sum of their weights is not greater than w and sum of their beauties is as large as possible. Along with that, from each friendship group he can either invite all Hoses, or no more than one. Otherwise, some Hoses will be hurt. Find for Mehrdad the maximum possible total beauty of Hoses he can invite so that no one gets hurt and the total weight doesn't exceed w.

Input: The first line contains integers n, m and w (1  ≤  n  ≤  1000, , 1 ≤ w ≤ 1000) — the number of Hoses, the number of pair of friends and the maximum total weight of those who are invited.The second line contains n integers w1, w2, ..., wn (1 ≤ wi ≤ 1000) — the weights of the Hoses.The third line contains n integers b1, b2, ..., bn (1 ≤ bi ≤ 106) — the beauties of the Hoses.The next m lines contain pairs of friends, the i-th of them contains two integers xi and yi (1 ≤ xi, yi ≤ n, xi ≠ yi), meaning that Hoses xi and yi are friends. Note that friendship is bidirectional. All pairs (xi, yi) are distinct.

Output: Print the maximum possible total beauty of Hoses Mehrdad can invite so that no one gets hurt and the total weight doesn't exceed w.

Note: In the first sample there are two friendship groups: Hoses {1, 2} and Hos {3}. The best way is to choose all of Hoses in the first group, sum of their weights is equal to 5 and sum of their beauty is 6.In the second sample there are two friendship groups: Hoses {1, 2, 3} and Hos {4}. Mehrdad can't invite all the Hoses from the first group because their total weight is 12 > 11, thus the best way is to choose the first Hos from the first group and the only one from the second group. The total weight will be 8, and the total beauty will be 7.

Sample Inputs:
3 1 5
3 2 5
2 4 2
1 2

4 2 11
2 4 6 6
6 4 2 1
1 2
2 3

Sample Outputs:
6

7

Solution (Python):
f = lambda : map(int, input().split())
(n, m, s) = f()
wb = [(0, 0)] + list(zip(f(), f()))
t = list(range(n + 1))

def g(x):
    if x == t[x]:
        return x
    t[x] = g(t[x])
    return t[x]
for i in range(m):
    (x, y) = f()
    (x, y) = (g(x), g(y))
    if x != y:
        t[y] = x
p = [[] for j in range(n + 1)]
for i in range(1, n + 1):
    p[g(i)].append(i)
d = [1] + [0] * s
for q in p:
    if len(q) > 1:
        t = [wb[i] for i in q]
        t.append((sum((x[0] for x in t)), sum((x[1] for x in t))))
        t.sort(key=lambda x: x[0])
        for j in range(s, -1, -1):
            if d[j]:
                for (w, b) in t:
                    if j + w > s:
                        break
                    d[j + w] = max(d[j + w], d[j] + b)
    elif len(q) == 1:
        (w, b) = wb[q[0]]
        for j in range(s - w, -1, -1):
            if d[j]:
                d[j + w] = max(d[j + w], d[j] + b)
print(max(d) - 1)

Solution (Python):
f = lambda : map(int, input().split())
(n, m, s) = f()
wb = [(0, 0)] + list(zip(f(), f()))
t = list(range(n + 1))

def g(x):
    if x == t[x]:
        return x
    t[x] = g(t[x])
    return t[x]
for i in range(m):
    (x, y) = f()
    (x, y) = (g(x), g(y))
    if x != y:
        t[y] = x
p = [[] for j in range(n + 1)]
for i in range(1, n + 1):
    p[g(i)].append(i)
d = [1] + [0] * s
for q in p:
    if len(q) > 1:
        t = [wb[i] for i in q]
        t.append((sum((x[0] for x in t)), sum((x[1] for x in t))))
        t.sort(key=lambda x: x[0])
        for j in range(s, -1, -1):
            if d[j]:
                for (w, b) in t:
                    if j + w > s:
                        break
                    d[j + w] = max(d[j + w], d[j] + b)
    elif len(q) == 1:
        (w, b) = wb[q[0]]
        for j in range(s - w, -1, -1):
            if d[j]:
                d[j + w] = max(d[j + w], d[j] + b)
print(max(d) - 1)

Solution (Python):
f = lambda : map(int, input().split())
(n, m, s) = f()
wb = [(0, 0)] + list(zip(f(), f()))
t = list(range(n + 1))

def g(x):
    if x == t[x]:
        return x
    t[x] = g(t[x])
    return t[x]
for i in range(m):
    (x, y) = f()
    (x, y) = (g(x), g(y))
    if x != y:
        t[y] = x
p = [[] for j in range(n + 1)]
for i in range(1, n + 1):
    p[g(i)].append(i)
d = [1] + [0] * s
for q in p:
    if len(q) > 1:
        t = [wb[i] for i in q]
        t.append((sum((x[0] for x in t)), sum((x[1] for x in t))))
        t.sort(key=lambda x: x[0])
        for j in range(s, -1, -1):
            if d[j]:
                for (w, b) in t:
                    if j + w > s:
                        break
                    d[j + w] = max(d[j + w], d[j] + b)
    elif len(q) == 1:
        (w, b) = wb[q[0]]
        for j in range(s - w, -1, -1):
            if d[j]:
                d[j + w] = max(d[j + w], d[j] + b)
print(max(d) - 1)