Codeforces Round #383 (Div. 2) D. Arpa's weak amphitheater and Mehrdad's valuable Hoses —— DP(01背包)

题目链接:http://codeforces.com/contest/742/problem/D

D. Arpa's weak amphitheater and Mehrdad's valuable Hoses
time limit per test

1 second

memory limit per test

256 megabytes

input

standard input

output

standard output

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.

Codeforces Round #383 (Div. 2) D. Arpa's weak amphitheater and Mehrdad's valuable Hoses —— DP(01背包)

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 nm and w (1  ≤  n  ≤  1000, Codeforces Round #383 (Div. 2) D. Arpa's weak amphitheater and Mehrdad's valuable Hoses —— DP(01背包), 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 ≤ nxi ≠ yi),
meaning that Hoses xiand 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.

Examples
input
3 1 5
3 2 5
2 4 2
1 2
output
6
input
4 2 11
2 4 6 6
6 4 2 1
1 2
2 3
output
7
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.

题解:

1.通过并查集,得出每一组有哪些人,并且将这些人的信息重新归类。

2.对于每一组,要么全选,要么只选一人,要么都不选。那么将全选的又看成一个“人 ”,并放入这个集合中,所以就变成了01背包了。

3.代码中,cnt为集合的个数,c[i]为集合i的元素个数,x[i][j]、y[i][j]为i集合中第j个元素的体重和颜值。

类似的题:http://blog.csdn.net/dolfamingo/article/details/73438052

写法一:

 #include<bits/stdc++.h>
//#define LOCAL
using namespace std;
typedef long long LL;
const double eps = 1e-;
const int INF = 2e9;
const LL LNF = 9e18;
const int mod = 1e9+;
const int maxn = 1e3+; int n, m, w, a[maxn], b[maxn];
int fa[maxn], x[maxn][maxn], y[maxn][maxn], c[maxn], cnt, dp[maxn][maxn], vis[maxn]; int find(int x) { return (x==fa[x])?x:x=find(fa[x]); } void init()
{
scanf("%d%d%d",&n,&m,&w);
for(int i = ; i<=n; i++)
scanf("%d",&a[i]);
for(int i = ; i<=n; i++)
scanf("%d",&b[i]);
for(int i = ; i<=n; i++)
fa[i] = i; for(int i = ; i<=m; i++)
{
int u, v;
scanf("%d%d",&u, &v);
u = find(u);
v = find(v);
if(u!=v)
fa[u] = v;
} for(int i = ; i<=n; i++)
{
int f = find(i);
if(!vis[f]) vis[f] = ++cnt;
f = vis[f]; //将f赋值为集合的序号
++c[f]; //集合内元素的个数
x[f][c[f]] = a[i];
y[f][c[f]] = b[i];
} for(int i = ; i<=cnt; i++) //全部都取,相当于每个集合增加一个元素
{
int X = , Y = ;
for(int j = ; j<=c[i]; j++)
{
X += x[i][j];
Y += y[i][j];
}
++c[i];
x[i][c[i]] = X;
y[i][c[i]] = Y;
}
} void solve()
{
for(int i = ; i<=cnt; i++)
for(int j = ; j<=w; j++)
{
for(int k = ; k<=c[i]; k++) //Remember!!!
dp[i][j] = dp[i-][j]; for(int k = ; k<=c[i]; k++)
if(j>=x[i][k])
dp[i][j] = max(dp[i][j], dp[i-][j-x[i][k]]+y[i][k]);
} cout<< dp[cnt][w]<<endl;
} int main()
{
#ifdef LOCAL
freopen("input.txt", "r", stdin);
// freopen("output.txt", "w", stdout);
#endif
init();
solve();
}

写法二:

 #include<bits/stdc++.h>
//#define LOCAL
using namespace std;
typedef long long LL;
const double eps = 1e-;
const int INF = 2e9;
const LL LNF = 9e18;
const int mod = 1e9+;
const int maxn = 1e3+; int n, m, w, a[maxn], b[maxn];
int fa[maxn], x[maxn][maxn], y[maxn][maxn], c[maxn], cnt, dp[maxn], vis[maxn]; int find(int x) { return (x==fa[x])?x:x=find(fa[x]); } void init()
{
scanf("%d%d%d",&n,&m,&w);
for(int i = ; i<=n; i++)
scanf("%d",&a[i]);
for(int i = ; i<=n; i++)
scanf("%d",&b[i]);
for(int i = ; i<=n; i++)
fa[i] = i; for(int i = ; i<=m; i++)
{
int u, v;
scanf("%d%d",&u, &v);
u = find(u);
v = find(v);
if(u!=v)
fa[u] = v;
} for(int i = ; i<=n; i++)
{
int f = find(i);
if(!vis[f]) vis[f] = ++cnt;
f = vis[f]; //将f赋值为集合的序号
++c[f]; //集合内元素的个数
x[f][c[f]] = a[i];
y[f][c[f]] = b[i];
} for(int i = ; i<=cnt; i++) //全部都取,相当于每个集合增加一个元素
{
int X = , Y = ;
for(int j = ; j<=c[i]; j++)
{
X += x[i][j];
Y += y[i][j];
}
++c[i];
x[i][c[i]] = X;
y[i][c[i]] = Y;
}
} void solve()
{
for(int i = ; i<=cnt; i++) //01背包
for(int j = w; j>=; j--)
for(int k = ; k<=c[i]; k++)
if(j>=x[i][k])
dp[j] = max(dp[j], dp[j-x[i][k]]+y[i][k]); cout<< dp[w]<<endl;
} int main()
{
#ifdef LOCAL
freopen("input.txt", "r", stdin);
// freopen("output.txt", "w", stdout);
#endif
init();
solve();
}
上一篇:人工神经网络入门(4) —— AFORGE.NET简介


下一篇:Loadrunner和JMeter并发对比