题意
给你一个序列 a a
定义的一个排列 p p 合法需要满足当时不存在 ap[j]=p[k] a p [ j ] = p [ k ]
定义一个排列的权值是 ∑ni=1iwp[i] ∑ i = 1 n i w p [ i ]
求最大权值
你永远不会想到会在正式考试上看到原题 , , 而且还在某校集训上讲过
原题链接 Poj2054 P o j 2054
题解
考虑转化
考虑到如果 aj=k a j = k 那么 k k 一定要排在前面
可以理解为对于 j j 来说需要先选
考虑建出一个图 , , 连边方向表示顺序
这样 [1,n] [ 1 , n ] 每个点的入度都会是 1 1
如果有环那么就无解否则这个图就是一棵以 0 0 为根树
如果是在树上的话也就是说必须要先选父亲才能选儿子
对于一个点 i, i , 如果选到他的时间是 T T 也就是说他在排列中的位置是那么他的贡献就是 Twi T w i
这样我们就成功把这道题转化成那个原题了但是这并没有什么用
考虑怎么求最大值
考虑一种贪心
考虑一个当前权值最小的点 i i
如果 i i 没有父亲那么我们当前一定是选 i i
如果 i i 有父亲那么当 fa[i] f a [ i ] 选了后我们一定会最先选 i i
也就是说在最后的排列中和 i i 是挨在一块的
但是考虑到实际上多次合并后每个节点就是一个序列
考虑一个长度为的序列 a a 和一个长度为的序列 b b
考虑和 ba b a 两种合并后的序列的答案(假设当前在第 i i 位)
如果 Wab>Wba⇒Wam1<Wbm2 W a b > W b a ⇒ W a m 1 < W b m 2
也就是平均权值小的放前面答案会更优
那么我们把平均权值作为合并后的新权值继续操作即可
计算答案的话就把答案拆开来计算
根据上面的式子可以得到把一个序列 b b 放在一个序列后面会产生独立的 Wb×m1 W b × m 1 的贡献 , , 边合并边求和就好了
每次取最小可以用堆实现
因为要修改权值所以你可以用set
你可以用里面的带修改堆 , , <script type="math/tex" id="MathJax-Element-3205">,</script>或者你拿个东西做标记就好了
#include<bits/stdc++.h>
#define fp(i,a,b) for(register int i=a,I=b+1;i<I;++i)
#define fd(i,a,b) for(register int i=a,I=b-1;i>I;--i)
#define go(u) for(register int i=fi[u],v=e[i].to;i;v=e[i=e[i].nx].to)
#define file(s) freopen(s".in","r",stdin),freopen(s".out","w",stdout)
template<class T>inline bool cmax(T&a,const T&b){return a<b?a=b,1:0;}
template<class T>inline bool cmin(T&a,const T&b){return a>b?a=b,1:0;}
using namespace std;
char ss[1<<17],*A=ss,*B=ss;
inline char gc(){return A==B&&(B=(A=ss)+fread(ss,1,1<<17,stdin),A==B)?-1:*A++;}
template<class T>inline void sd(T&x){
char c;T y=1;while(c=gc(),(c<48||57<c)&&c!=-1)if(c==45)y=-1;x=c-48;
while(c=gc(),47<c&&c<58)x=x*10+c-48;x*=y;
}
char sr[1<<21],z[20];int C=-1,Z;
inline void Ot(){fwrite(sr,1,C+1,stdout),C=-1;}
template<class T>inline void we(T x){
if(C>1<<20)Ot();if(x<0)sr[++C]=45,x=-x;
while(z[++Z]=x%10+48,x/=10);
while(sr[++C]=z[Z],--Z);sr[++C]='\n';
}
const int N=5e5+5;
typedef int arr[N];
typedef long long ll;
struct Da{
int u,sz;ll w;
inline bool operator<(const Da b)const{return w*b.sz>b.w*sz;}
};
struct eg{int nx,to;}e[N];
int n,Cnt;arr fi,fa,Fa,sz,vis;ll ans,w[N];priority_queue<Da>q;
void dfs(int u){vis[u]=1;++Cnt;go(u)if(vis[v]){puts("-1"),exit(0);}else dfs(v);}
inline void add(int u,int v){static int ce=0;e[++ce]={fi[u],v},fi[u]=ce;}
int gf(int x){return Fa[x]==x?x:Fa[x]=gf(Fa[x]);}
int main(){
#ifndef ONLINE_JUDGE
file("s");
#endif
sd(n);
fp(i,1,n)sd(fa[i]),add(fa[i],i);
dfs(0);if(Cnt<=n)return puts("-1"),0;
fp(i,0,n)Fa[i]=i,sz[i]=1;
fp(i,1,n)sd(w[i]),q.push(Da{i,1,w[i]});
int u,p;Da s;
while(!q.empty()){
s=q.top();q.pop();
if(sz[u=gf(s.u)]^s.sz)continue;
Fa[u]=p=gf(fa[u]);
ans+=w[u]*sz[p],w[p]+=w[u],sz[p]+=sz[u];
if(p)q.push(Da{p,sz[p],w[p]});
}
printf("%lld\n",ans);
return 0;
}