bzoj 5289 [Hnoi2018]排列

http://www.elijahqi.win/archives/3670
首先转化限制条件 变成从所有的$a_i$ 向i连线 因为首先这样画图可以观察出

一定那个排列线选ai的值再选i 如果这图有环则题目无解

考虑如何求最大值
我们知道 相当于这是在一棵树上我们必须先选父亲才能选下面的节点
那么按理来说应该是固定的顺序 但是九省联考的iiidx提示我们
这样贪心显然是不对的
我们考虑如何合并两个已经存在的序列a和b
首先a,b一定已经先各自计算过一部分答案了
那么考虑将式子写出可以发现
如果新组成的序列叫 a在前b在后$w_{ab}$ 假设m1是a序列的长度 m2是b序列的长度
那么假设$w_{ab}>w_{ba}$存在$w_{b}*m1>w_{a}*m2$
即 序列a的平均值越小越先接越好 那么显然用一个堆维护下即可 大概如果没有0 那么一定无解

#include<queue>
#include<cstdio>
#include<cctype>
#include<algorithm>
using namespace std;
#define ll long long
inline char gc(){
    static char now[1<<16],*S,*T;
    if (T==S){T=(S=now)+fread(now,1,1<<16,stdin);if (T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(!isdigit(ch)) {if (ch=='-') f=-1;ch=gc();}
    while(isdigit(ch)) x=x*10+ch-'0',ch=gc();
    return x*f;
}
const int N=500050;
bool visit[N];ll w[N],ans;
int size[N],n,fa[N],fa1[N],h[N],num,cnt;
struct node{
    int size,x;ll w;
    inline friend bool operator <(const node &a,const node &b){return a.w*b.size>b.w*a.size;}
};
struct node1{
    int y,next;
}data[N];
inline int find(int x){
    while(x!=fa1[x]) x=fa1[x]=fa1[fa1[x]];return x;
}
inline void insert1(int x,int y){
    data[++num].y=y;data[num].next=h[x];h[x]=num;
}
inline bool dfs(int x){
    visit[x]=1;++cnt;
    for (int i=h[x];i;i=data[i].next){
        int y=data[i].y;
        if(visit[y]) return 1;if (dfs(y)) return 1;
    }return 0;
}
priority_queue<node>q;
int main(){
    freopen("bzoj5289.in","r",stdin);
    n=read();size[0]=1;
    for (int i=1;i<=n;++i) fa[i]=read(),insert1(fa[i],i);
    if (dfs(0)) {puts("-1");return 0;}if (cnt<=n) {puts("-1");return 0;}
    for (int i=1;i<=n;++i) w[i]=read(),fa1[i]=i,size[i]=1,q.push((node){1,i,w[i]});
    while(!q.empty()){
        static node xx;xx=q.top();q.pop();static int p,x;
        if (xx.size!=size[x=find(xx.x)]) continue;
        p=fa1[x]=find(fa[x]);ans+=size[p]*w[x];size[p]+=size[x];w[p]+=w[x];
        if(p) q.push((node){size[p],p,w[p]});
    }printf("%lld\n",ans);
    return 0;
}