BZOJ5289 洛谷4437：[HNOI/AHOI2018]排列—

https://www.lydsy.com/JudgeOnline/problem.php?id=5289

https://www.luogu.org/problemnew/show/P4437

考虑对于a[i]=m,a[m]=n，我们令p[j]=i,p[k]=m（一定会有一对(j,k)满足这个条件的），则我们会有p[k]=a[p[j]]，此时我们要满足k<j，也就是a[m]放的位置要比a[i]靠前。

也就是说选第m个之后才能选第i个。

转换成图论模型就是m->i <=> a[i]->i。

那么问题豁然开朗，首先如果图中有环就能判断无解了。

如果有解则必然为以0为根的有向树，则答案为每个节点i被拿到的时间*w[i]（前提是i的父亲被拿才可以拿i）。

再考虑最大化答案，则我们让树中最小值min尽可能早的被拿到就好了。

继续贪心，则如果当前局面能够拿到min则一定拿min，换句话将就是拿了min的父亲就一定拿min。

那么父亲和min之间就成了捆绑关系，于是将其缩起来，在缩的过程中更新答案，然后递归这个过程就好了。

每次找min可以用堆维护，复杂度O(nlogn)。

（PS：更新答案，每次更新显然是这个点前面一些点被选了，于是它一定产生了前面这些点个数*w[该点]的价值）

那么就需要考虑我们缩完的点的w要怎么计算，对于两个集合a，b要合并，显然用在计算上的w=wa+wb，但是用堆排序的时候就不能这么做了。

自然能想到取平均值，虽然我不会证明，不过考量一下发现差不多。

#include<cmath>
#include<queue>
#include<vector>
#include<cstdio>
#include<cctype>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<ext/pb_ds/priority_queue.hpp>
using namespace std;
typedef long long ll;
typedef long double dl;
typedef pair<dl,int>pii;
#define fi first
#define se second
typedef __gnu_pbds::priority_queue<pii,greater<pii>,__gnu_pbds::pairing_heap_tag> heap;
const int N=5e5+5;
inline int read(){int X=0,w=0;char ch=0;while(!isdigit(ch)){w|=ch=='-';ch=getchar();}while(isdigit(ch))X=(X<<3)+(X<<1)+(ch^48),ch=getchar();return w?-X:X;
}
struct node{int to,nxt;
}e[N];
int n,cnt,head[N],vis[N],a[N],num,fa[N],size[N];
ll w[N],ans;
heap::point_iterator id[N];
heap q;
inline void add(int u,int v){e[++cnt].to=v;e[cnt].nxt=head[u];head[u]=cnt;
}
bool dfs(int u){vis[u]=1;num++;for(int i=head[u];i;i=e[i].nxt){int v=e[i].to;if(vis[v]||!dfs(v))return 0;}if(!u&&num<=n)return 0;return 1;
}
inline int find(int x){if(x==fa[x])return x;return fa[x]=find(fa[x]);
}
int main(){n=read();for(int i=1;i<=n;i++)a[i]=read(),add(a[i],i);for(int i=1;i<=n;i++)w[i]=read();if(!dfs(0)){puts("-1");return 0;}for(int i=0;i<=n;i++)fa[i]=i,size[i]=1;for(int i=1;i<=n;i++)id[i]=q.push(pii(w[i],i));while(!q.empty()){int u=q.top().se,v=a[u];q.pop();int rt=find(v);ans+=w[u]*size[rt];fa[u]=rt;w[rt]+=w[u];size[rt]+=size[u];if(rt)q.modify(id[rt],pii((dl)w[rt]/size[rt],rt));}printf("%lld\n",ans);return 0;
}

+++++++++++++++++++++++++++++++++++++++++++

+本文作者：luyouqi233。　　　　　　　　　　　　　　+

+欢迎访问我的博客：http://www.cnblogs.com/luyouqi233/+

+++++++++++++++++++++++++++++++++++++++++++

转载于:https://www.cnblogs.com/luyouqi233/p/9112242.html

BZOJ5289 洛谷4437：[HNOI/AHOI2018]排列——题解相关推荐

洛谷4438 [HNOI/AHOI2018]道路
标签:树形DP 题目题目传送门题目描述 W 国的交通呈一棵树的形状.W 国一共有n−1n - 1n−1个城市和nnn个乡村,其中城市从111到n−1n - 1n−1 编号,乡村从111到nnn编号 ...
洛谷P1129 [ZJOI2007] 矩阵游戏题解
洛谷P1129 [ZJOI2007] 矩阵游戏题解题目链接:P1129 [ZJOI2007] 矩阵游戏题意:给定一张有黑白棋子的正方形棋盘,问存不存在解法使得经过若干次交换行或列的操作后,左上角 ...
洛谷P3336 [ZJOI2013]话旧题解
洛谷P3336 [ZJOI2013]话旧题解题目链接:P3336 [ZJOI2013]话旧题意:小林跟着银河队选手去了一趟宇宙比赛,耳濡目染,变得学术起来.回来后,他发现世界大变样了.比丘兽究级 ...
洛谷P4683 [IOI2008] Type Printer 题解
洛谷P4683 [IOI2008] Type Printer 题解题目链接:P4683 [IOI2008] Type Printer 题意: 你需要利用一台可移动的打印机打印出NNN个单词.这种可移 ...
洛谷P3647 [APIO2014] 连珠线题解
洛谷P3647 [APIO2014] 连珠线题解题目链接:P3647 [APIO2014] 连珠线题意: 在达芬奇时代,有一个流行的儿童游戏称为连珠线.当然,这个游戏是关于珠子和线的.线是红色或 ...
洛谷P4170 [CQOI2007]涂色题解
洛谷P4170 [CQOI2007]涂色题解题目链接:P4170 [CQOI2007]涂色题意: 假设你有一条长度为 555 的木板,初始时没有涂过任何颜色.你希望把它的 555 个单位长度分别 ...
洛谷P1868 饥饿的奶牛题解
洛谷P1868 饥饿的奶牛题解题目链接:P1868 饥饿的奶牛题意: 有一条奶牛冲出了围栏,来到了一处圣地(对于奶牛来说),上面用牛语写着一段文字. 现用汉语翻译为: 有 NNN 个区间,每个区 ...
洛谷P1462 通往奥格瑞玛的道路题解
洛谷P1462 通往奥格瑞玛的道路题解题目链接:P1462 通往奥格瑞玛的道路题意:在艾泽拉斯,有 nnn 个城市.编号为 1,2,3,-,n1,2,3,\ldots,n1,2,3,-,n . ...
洛谷 P2015 二叉苹果树题解
洛谷 P2015 二叉苹果树题解洛谷 P2015 题目有一棵苹果树,如果树枝有分叉,一定是分2叉(就是说没有只有1个儿子的结点) 这棵树共有NNN个结点(叶子点或者树枝分叉点),编号为1-NNN ...

BZOJ5289 洛谷4437：[HNOI/AHOI2018]排列——题解

BZOJ5289 洛谷4437：[HNOI/AHOI2018]排列——题解相关推荐

最新文章

热门文章