\(\\\)

\(Description\)

给出一棵以\(1\)为根的\(N\)个节点的树，开始的时候你在\(1\)号节点。

除了\(1\)号节点以外，每个点都有访问次数限制\(t_i\)，即到达该点的次数上限。

除了\(1\)号点每个点还有一个权值\(w_i\)，这个权值可以是负的，每个点被第一次到达时你会被迫得到他的点权，以后该点点权变为\(0\)。

求满足所有次数上限的前提下，从\(1\)号点出发，最后回到\(1\)号点的一条路径所得到的最大点权和，每个点可以经过多次。

同时你还要输出这个最大点权和对应的方案是否唯一。

• \(N\in [1,10^5]\)

\(\\\)

\(Solution\)

• 第一问直接树形\(DP\)就好，从根节点到当前点的路径会消耗一次当前点的访问次数，而每次从子树回溯上来也会消耗一次访问次数，所以对于节点\(u\)，最多只能选\(t_u-1\)棵子树访问。直接\(DFS\)后将子树最大贡献排序，在所有正数答案里选前\(t_u-1\)个子树作为自己的答案。

• 关于方案唯一性的问题，维护一个\(g\)数组表示当前节点最优解是否唯一。转移时只要有一个子树方案数有多种当前节点的方案数就是多种。同时如果下一个要选择的子树（因为访问上限的关系不能选）和当前最后一个选择的子树答案相同，或者答案中选择了包含贡献为\(0\)的子树，方案也不是唯一的。

\(\\\)

\(Code\)

#include<cmath>
#include<cstdio>
#include<cctype>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#define N 100010
#define R register
#define gc getchar
#define inf 200000000
using namespace std;inline int rd(){int x=0; bool f=0; char c=gc();while(!isdigit(c)){if(c=='-')f=1;c=gc();}while(isdigit(c)){x=(x<<1)+(x<<3)+(c^48);c=gc();}return f?-x:x;
}bool g[N];
int n,m,tot,hd[N];
int t[N],f[N],val[N],tmp[N];struct edge{int to,nxt;}e[N<<1];inline void add(int u,int v){e[++tot].to=v; e[tot].nxt=hd[u]; hd[u]=tot;
}inline bool cmp(int x,int y){return f[x]>f[y];}void dfs(int u,int fa){f[u]=val[u];for(R int i=hd[u],v;i;i=e[i].nxt) if((v=e[i].to)!=fa) dfs(v,u);tmp[0]=0;for(R int i=hd[u],v;i;i=e[i].nxt) if((v=e[i].to)!=fa) tmp[++tmp[0]]=v;sort(tmp+1,tmp+1+tmp[0],cmp);int ptr=1,lim=min(tmp[0],t[u]-1);while(ptr<=lim&&f[tmp[ptr]]>=0) f[u]+=f[tmp[ptr]],g[u]|=g[tmp[ptr]],++ptr;if((ptr<=tmp[0]&&ptr>1&&f[tmp[ptr]]==f[tmp[ptr-1]])||(f[tmp[ptr-1]]==0&&ptr>1)) g[u]=1;
}int main(){n=rd();for(R int i=2;i<=n;++i) val[i]=rd();for(R int i=2;i<=n;++i) t[i]=rd();for(R int i=1,u,v;i<n;++i){u=rd(); v=rd(); add(u,v); add(v,u);}val[1]=0; t[1]=inf; dfs(1,0);printf("%d\n",f[1]);puts(g[1]?"solution is not unique":"solution is unique");return 0;
}