小Y上学记——小Y的玩偶（贪心）

小Y上学记——小Y的玩偶

Time Limit: 2000/1000MS (Java/Others)    Memory Limit: 128000/64000KB (Java/Others)
Submit Status

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;
}