[HNOI/AHOI2018]排列 贪心

题面
题解:
把题面的限制换成中文:
如果排在第k位的下标 = 排在第j位的值 ,那么k < j
换一个描述方式:
一个值为x的数要排在第x个数后面。
再换一个描述方式:
\(fa[i] = a_i\)

仿佛有什么不对劲?
嗯这其实就是一棵树。
并且我们可以发现,这棵树一定以一个虚拟节点0为根,并且有合法排列,也就是树没有环,当且仅当从0开始遍历,可以遍历到所有节点。
且排列合法当且仅当我们在访问一个节点之前,先访问它的父亲,也就相当于在树上走。

因为如果不以0为根,那么由于题面中所说权值大小在\([1, n]\)的区间内,我们根本无法找到进入这样一些关系的"入口",因为不管我们尝试从那个点开始走,都必须要先去这个点的父亲,而要去这个点的父亲,又要先去这个点的父亲的父亲……因为在这棵树中,我们唯一可以直接到达的点是虚拟节点0,所以如果这些关系组成的边无法通向0的话,我们只能在无限个这样的限制中绕圈,永远无法找到入口,即没有合法的解。

因此我们真正的题面其实是:
给定一个n + 1个节点的树,依次取点,满足取儿子之前,必须要取它的父亲。每个点有权值,第i个取的点的权值会被乘上i,求一个合法方案,使得取完所有点后权值之和最大。

根据贪心的原则,一个点的权值越小,就越要优先取。

因此我们考虑整棵树中权值最小的那个节点。

1,如果这个节点没有父亲(或父亲已经被取走),那我们肯定先取它。
2,如果这个点有父亲,那么一旦我们取了它的父亲,我们肯定会马上就取它(此时满足情况1)

综上,我们得到一个结论,如果我们每次考虑权值最小的那个节点,这个节点要么没有父亲(f[i] = 0),要么一定会紧挨着它的父亲取。
所以这个点和它的父亲在最优排列中一定是相邻的,因此我们可以考虑合并这2个节点。
那么实际上,树中的每个节点就代表了一段排列,现在来考虑一段排列怎么评估一段排列的权值

假设有一个长度为\(m_1\)的序列\(a\),和一段长度为\(m_2\)的序列\(b\),我们表示出序列\(ab\)和序列\(ba\)的权值
\[W_{ab} = \sum_{j = 1}^{m_1}(i + j)W_{a_j} + \sum_{j = 1}^{m_2}(i + j + m_1)W_{b_j}\]
\[W_{ab} = \sum_{j = 1}^{m_2}(i + j)W_{b_j} + \sum_{j = 1}^{m_1}(i + j + m_2)W_{a_j}\]

如果排列\(ab\)优于排列\(ba\),那么\(W_{ab} > W_{ba}\),即先选\(a\)更优
那么有:
\[W_{ab} - W_{ba} = m_1W_b - m_2W_a > 0\]
\[\frac{W_b}{m_2} > \frac{W_a}{m_1}\]
因此我们只需要以平均权值作为新的权值来考虑即可。

根据上诉式子,可以得到,如果我们把序列\(b\)放在序列\(a\)后面,可以得到一个独立的\(m_1 \cdot W_b\)的贡献,因此在不断合并节点的过程中统计贡献即可

\(\Delta\)此题有点卡精度,请用long double……

#include<bits/stdc++.h>
using namespace std;
#define R register int
#define AC 501000
#define LL long long
#define ld long double 

int n, rnt;
LL ans, sum[AC];
int v[AC], fa[AC], cnt[AC];//cnt记录每个节点的版本号
int Next[AC], last[AC], belong[AC], Size[AC];//用并查集+双向链表来维护已经合并的序列

struct node{int id, x; ld w;};//当前点版本号+编号+值
struct cmp{bool operator() (node a, node b){return a.w > b.w;}};

struct road{
    int Head[AC], Next[AC], date[AC], tot;
    inline void add(int f, int w){
        date[++ tot] = w, Next[tot] = Head[f], Head[f] = tot;
    }//用父亲向儿子连边
}E;


node _s[AC]; int _top;//check


struct STL_Delete_queue{
    priority_queue<node, vector<node>, cmp> q1;
    void del(int x){++ cnt[x];}//删除编号为x的并查集
    void push(node x){q1.push(x);}
    
    int top()
    {
        while(!q1.empty() && q1.top().id != cnt[q1.top().x]) q1.pop();  
        int x = q1.top().x;
        q1.pop();
        return x;
    }
    
    void check()
    {
        _top = 0;
        while(!q1.empty()) _s[++ _top] = q1.top(), q1.pop();
        for(R i = 1; i <= n; i ++) 
            printf("%d %.3Lf\n", _s[i].x, _s[i].w);
        for(R i = 1; i <= _top; i ++) q1.push(_s[i]);
    }
}q;


inline int read()
{
    int x = 0;char c = getchar();
    while(c > '9' || c < '0') c = getchar();
    while(c >= '0' && c <= '9') x = x * 10 + c - '0', c = getchar();
    return x;
}



int find(int x)
{
    if(belong[x] == x) return x;
    else return belong[x] = find(belong[x]);
}

void dfs1(int x)//先判断是否合法
{
    ++ rnt;//和0相连的点必定形成一个合法的树,其他不合法的环都是独立出来的。
    for(R i = E.Head[x]; i != -1; i = E.Next[i]) 
        dfs1(E.date[i]);//所以只需要记录从0开始遍历,可以遍历到多少点即可
}   

void pre()
{
    n = read();
    for(R i = 0; i <= n; i ++) E.Head[i] = Next[i] = last[i] = -1;//因为有0号节点,因此先全都赋值为-1
    for(R i = 1; i <= n; i ++) fa[i] = read(), E.add(fa[i], i), belong[i] = i;
    for(R i = 1; i <= n; i ++) 
    {
        v[i] = sum[i] = read(), ans += v[i];
        q.push((node){0, i, (ld)v[i]}), Size[i] = 1; //维护集合的信息
    }
}

void work()
{
    for(R i = 1; i <= n; i ++)//因为一共有n + 1个节点,所以会合并n次
    {
        int x = q.top(), fx = find(fa[x]);
        belong[x] = fx, q.del(fx);
        ans += Size[fx] * sum[x], Size[fx] += Size[x], sum[fx] += sum[x];
        if(fx) q.push((node){cnt[fx], fx, (ld)sum[fx] / (ld)Size[fx]});
    }   
    printf("%lld\n", ans);
}

int main()
{
//  freopen("in.in", "r", stdin);
    pre();
    dfs1(0);
    if(rnt != n + 1) printf("-1\n");
    else work();
//  fclose(stdin);
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/ww3113306/p/10277950.html