小Y上学记——小Y的玩偶
Time Limit:
2000/1000MS (Java/Others)
Memory Limit:
128000/64000KB (Java/Others)
Problem Description
小Y最喜欢拆拆拆了~尽管他不一定能装回去。
小Y有一个很可爱的积木玩偶,是由一块一块积木拼接而成,现在小Y想把这个积木玩偶拆拆拆。
每一块积木玩偶都有一个耐久值,想把一块积木拆出来,小Y需要付出的能量就是和它直接拼接的所有积木的耐久值之和。
小Y很懒的~他想知道把这个玩偶全部拆好,最少需要付出多少能量?
Input
多组数据,每组数据首先是一个整数n表示积木块数。(0<n<=1000)
接下来一行包含n个整数,表示每块积木的耐久值a[i](0<=a[i]<=100000)。
接下来是n行,第i行代表第i块积木的连接情况。
每一行首先是一个整数k,表示这块积木与k块积木相连,接下来是k个整数,代表与这块积木相连的积木标号(标号从0开始)
保证连接情况合法。
Output
对于每组数据,输出一个整数,表示小Y需要的最少能量。
Sample Input
4 10 20 30 40 2 1 3 2 0 2 1 1 1 0 4 100 100 100 100 1 1 3 0 2 3 2 1 3 2 1 2 7 40 10 20 10 20 80 40 4 2 3 4 5 1 4 2 0 3 5 0 2 4 5 6 4 0 1 3 6 2 0 3 2 3 4
Sample Output
40 400 160
Hint
对于第一组数据,首先拆掉第2块积木,需要20能量,然后拆掉第1块积木,需要10能量,接着拆掉第3块积木,需要10能量,最后只剩下第0块了,不需要能量了。总共需要40点能量。
对于第二组数据,无论怎么拆除,都是需要400点能量。
代码:
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn = 1010;
int a[maxn], b[maxn], g[maxn][maxn];
long long ans;
int n;
bool cmp(int i, int j) { return a[i] > a[j]; }
int main() {
while(~scanf("%d", &n)) {
memset(a, 0, sizeof(a));
for(int i = 0; i < n; i++) {
scanf("%d", &a[i]);
b[i] = i;
memset(g[i], 0, sizeof(g[i]));
}
for(int i = 0; i < n; i++) {
int m;
scanf("%d", &m);
for(int j = 0; j < m; j++) {
int x;
scanf("%d", &x);
g[i][x] = 1;
}
}
sort(b, b+n, cmp);
ans = 0;
for(int i = 0; i < n; i++) {
for(int j = 0; j < n; j++) {
if(g[b[i]][j]) {
ans += a[j];
g[b[i]][j] = g[j][b[i]] = 0;
}
}
}
printf("%lld\n", ans);
}
return 0;
}