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 */
?
