U66905 zz题
考虑一个点权值被计算了多少次。。。不知
所以对未来承诺，方便直接算上总数！
然后其实是给边定向，即先删除fa和son的哪一个
f[x][j]，会计算j次
无法转移
f[x][j][k]，其中会从子树计算k次。
当边从儿子指向父亲，枚举就是O(n^4)的了，还不能sz剪枝
转移是O(n^4)的
（其实这里记录一个前缀和之类的就行了）
可以用f[i][j]，仅往i子树里选择j个最大值
g[i][j]，往i子树外额外选择j个最大值
然后就可以转移了
注意：
权值有负数，而每个儿子强制必须选的，所以不能累计取max
// luogu-judger-enable-o2 #pragma GCC optimize("O3,Ofast,inline,unroll-all-loops,-ffast-math") #pragma GCC target("avx,sse2,sse3,sse4,popcnt") #include<bits/stdc++.h> #define reg register int #define il inline #define fi first #define se second #define mk(a,b) make_pair(a,b) #define numb (ch^'0') using namespace std; typedef long long ll; template<class T>il void rd(T &x){char ch;x=0;bool fl=false;while(!isdigit(ch=getchar()))(ch=='-')&&(fl=true);for(x=numb;isdigit(ch=getchar());x=x*10+numb);(fl==true)&&(x=-x); } template<class T>il void output(T x){if(x/10)output(x/10);putchar(x%10+'0');} template<class T>il void ot(T x){if(x<0) putchar('-'),x=-x;output(x);putchar(' ');} template<class T>il void prt(T a[],int st,int nd){for(reg i=st;i<=nd;++i) ot(a[i]);putchar('\n');}namespace Miracle{ const int N=401; const ll inf=0x3f3f3f3f3f3f3f3f; int n; struct node{int nxt,to; }e[2*N]; int hd[N],cnt; ll d[N]; void add(int x,int y){e[++cnt].nxt=hd[x];e[cnt].to=y;hd[x]=cnt; } ll h[N][N][N]; ll f[N][N],g[N][N]; int sz[N]; void dfs(int x){ // cout<<" dfs "<<x<<endl;sz[x]=1;for(reg j=1;j<=n;++j){h[x][j][1]=d[x]*j;} // bool fl=false;for(reg i=hd[x];i;i=e[i].nxt){int y=e[i].to;dfs(y); // fl=true;for(reg j=1;j<=n;++j){for(reg k=min(j,sz[x]+sz[y]);k>=1;--k){ll old=h[x][j][k];h[x][j][k]=-0x3f3f3f3f3f3f3f3f;for(reg p=min(sz[x],k-1);p>=1;--p){h[x][j][k]=max(h[x][j][k],h[x][j][p]+f[y][k-p]);}h[x][j][k]=max(h[x][j][k],old+g[y][j]);}}sz[x]+=sz[y];} // cout<<" now "<<x<<endl; // if(!fl){ // cout<<" leaf "<<endl; // for(reg j=1;j<=n;++j){ // h[x][j][1]=d[x]*j; // } // }for(reg j=1;j<=n;++j){ // cout<<" jjj "<<j<<endl; f[x][j]=h[x][j][j];for(reg k=1;k<=sz[x]&&k+j<=n;++k){g[x][j]=max(g[x][j],h[x][j+k][k]);} // cout<<" f "<<f[x][j]<<" g "<<g[x][j]<<" "<<endl; } } int main(){rd(n);for(reg i=1;i<=n;++i) rd(d[i]);int y=0;for(reg x=2;x<=n;++x){rd(y);add(y,x);}memset(h,0xcf,sizeof h);memset(f,0xcf,sizeof f);memset(g,0xcf,sizeof g);dfs(1);ll ans=-0x3f3f3f3f3f3f3f3f;for(reg j=1;j<=n;++j){ans=max(ans,f[1][j]);}printf("%lld",ans);return 0; }} signed main(){ // freopen("data.in","r",stdin); // freopen("my.out","w",stdout); Miracle::main();return 0; }/*Author: *Miracle*Date: 2019/3/29 20:22:41 */
