File dữ liệu vào: stdin
File kết quả: stdout
Hạn chế thời gian: 0.5 giây
Hạn chế bộ nhớ: 1024 MB
Cho một cây với trọng số trên mỗi đỉnh và một tập con Q các đỉnh của cây, hãy tìm tập độc lập có tổng
trọng số lớn nhất trong Q.
Dữ liệu vào
Dòng đầu chứa một số T ≤ 20 là số lượng bộ test;
Mỗi test có cấu trúc như sau:
• Dòng đầu chứa 2 số n và m là số đỉnh và số cạnh của đồ thị;
• Dòng thứ 2 chứa n số là trọng số trên mỗi đỉnh;
• Từ dòng 3 đến dòng m + 2, mỗi dòng chứa 2 số là các cặp đỉnh có cạnh nối giữa chúng, dữ liệu
đảm bảo đồ thị đã cho là 1 cây;
• Dòng m + 3 chứa 1 số nguyên q là số truy vấn, sau đó mỗi dòng trong số q dòng tiếp chứa 1 dãy số:
– Số đầu tiên trong dãy k chỉ số lượng của tập đỉnh truy vấn:
– k số tiếp theo chỉ số hiệu từng đỉnh trong tập (đánh số từ 0) là những đỉnh xét đến trong n
đỉnh ban đầu
Kết quả
Mỗi truy vấn in ra 1 kết quả là trọng số tập độc lập lớn nhất trên 1 dòng. Sau mỗi một bộ test thì in ra
một dòng trống.
Ví dụ
stdin stdout
1
3 2
5 4 10
0 2
2 1
1
2 2 1
10
Giải thích
Trong ví dụ trên, có 1 bộ test với cây gồm 3 đỉnh có số hiệu là 0,1,2 và 2 cạnh (0 2) và (2 1). Có 1 truy
vấn chỉ xét trên 2 đỉnh có số hiệu 1 và 2.
Hạn chế
• 50% số test 5 ≤ n ≤ 30, 1 ≤ q ≤ 100;
• 25% số test tiếp theo có 10 ≤ n ≤ 200, 100 ≤ q ≤ 1000;
• 25% số test còn lại có 150 ≤ n ≤ 200, q = 1000.
class Node:
def __init__(self, index, weight):
self.index = index
self.weight = weight
self.children = []
def max_independent_set(root, selected):
if not root:
return 0
if selected[root.index]:
return 0
selected[root.index] = True
exclude_root = 0
for child in root.children:
exclude_root += max_independent_set(child, selected)
selected[root.index] = False
include_root = root.weight
for child in root.children:
include_root += max_independent_set(child, selected)
return max(include_root, exclude_root)
def solve_test_case():
n, m = map(int, input().split())
weights = list(map(int, input().split()))
nodes = [Node(i, weights[i]) for i in range(n)]
for _ in range(m):
u, v = map(int, input().split())
nodes[u].children.append(nodes[v])
nodes[v].children.append(nodes[u])
q = int(input().strip())
for _ in range(q):
query = list(map(int, input().split()))
k = query[0]
selected = [False] * n
for i in range(1, k + 1):
selected[query[i]] = True
result = max_independent_set(nodes[0], selected)
print(result)
# Đọc số lượng bộ test
t = int(input().strip())
# Xử lý từng bộ test
for _ in range(t):
solve_test_case()
print() # In ra dòng trống giữa các bộ test
Bạn Học Tốt Nhé
Tin học là một ngành khoa học chuyên nghiên cứu quá trình tự động hóa việc tổ chức, lưu trữ, xử lý và truyền dẫn thông tin của một hệ thống máy tính cụ thể hoặc trừu tượng. Tin học bao hàm tất cả các nghiên cứu và kỹ thuật có liên quan đến việc mô phỏng, biến đổi và tái tạo thông tin. Hãy tận dụng sức mạnh của tin học để giải quyết các vấn đề và sáng tạo ra những giải pháp mới!
Lớp 8 - Năm thứ ba ở cấp trung học cơ sở, học tập bắt đầu nặng dần và sang năm lại là năm cuối cấp, áp lực lớn dần. Hãy chú ý đến sức khỏe, cân bằng giữa học và nghỉ ngơi để đạt hiệu quả tốt nhất!
Copyright © 2021 HOCTAPSGK