http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=3715
贪心加暴力,可以出奇迹。
之所以这句话那么流行,是因为贪心,暴力之类的算法会让人感到很低级,但又确实能解决很多问题。
哪怕是通过数很低的题目,也有可能是通过贪心,暴力之类的算法解决的。
注意数据规模,不大时可以考虑暴力+贪心,而不是动规或模拟,但要算对时间复杂度,什么样的时间复杂度对应什么样的暴力,或者确实没法暴力。
代码
#include<bits/stdc++.h>
using namespace std;
const int maxn = 110;
const int inf = 0x3f3f3f3f;
int n;
int f[maxn];
int c[maxn];
vector<int>vec[maxn];
int get_ans(int x)
{
int num=x-vec[1].size();
int sum=0;
int cnt=0;
vector<int>sb;
for(int i=2;i<=n;i++) for(int j=0;j<(int)vec[i].size();j++)
{
if((int)vec[i].size()>=x)
{
if(j<=(int)vec[i].size()-x) sum+=vec[i][j],cnt++;
else sb.push_back(vec[i][j]);
}
else sb.push_back(vec[i][j]);
}
if(cnt>=num) return sum;
num-=cnt;
sort(sb.begin(),sb.end());
for(int i=0;i<num;i++) sum+=sb[i];
return sum;
}
void solve()
{
scanf("%d",&n);
for(int i=1;i<=n;i++) vec[i].clear();
for(int i=2;i<=n;i++) scanf("%d",f+i);
for(int i=2;i<=n;i++) scanf("%d",c+i);
for(int i=2;i<=n;i++) vec[f[i]].push_back(c[i]);
for(int i=1;i<=n;i++) sort(vec[i].begin(),vec[i].end());
int ans=inf;
for(int i=(int)vec[1].size();i<n;i++) ans=min(ans,get_ans(i));
printf("%d\n",ans);
}
int main()
{
int T;
scanf("%d",&T);
while(T--) solve();
return 0;
}