BZOJ.5289.[AHOI/HNOI2018]排列(贪心 heap)
\(Kelin\)写的挺清楚的...
要求如果\(a_{p_j}=p_k\),\(k\lt j\),可以理解为\(k\)要在\(j\)之前选。
那么对于给定的\(a_j=k\),我们可以连边\(k\to j\)建一张图。如果图有环,那么无解;否则这是一棵以\(0\)为根的树。
限制就变成了,选一个点前必须要选其父亲;如果第\(k\)个选点\(i\),\(i\)的贡献是\(k\cdot w_i\)。
然后怎么做呢...
假设当前最小的数是\(x\),那么如果\(fa[x]\)选过,现在一定选\(x\);否则如果有一次选了\(fa[x]\),下一次一定选\(x\)。
考虑到按顺序选点实际是构成一个序列,且上面的\(x\)一定和\(fa[x]\)在序列中挨在一起。所以考虑把\(x,fa[x]\)合并成一个序列,再考虑其它的。
之前是比点权,现在我们要比较序列与序列之间谁放在后面更优。
考虑两个序列\(a,b\),设\(a\)中有\(t_1\)个点,\(b\)中有\(t_2\)个点,假设现在已经选了\(k\)个点,那么\(ab,ba\)的贡献分别为$$W_{ab}=\sum_{i=1}{t_1}(k+i)w_{a_i}+\sum_{i=1}{t_2}(k+i+t_1)w_{b_i}\W_{ba}=\sum_{i=1}{t_2}(k+i)w_{b_i}+\sum_{i=1}{t_1}(k+i+t_2)w_{a_i}\W_{ab}-W_{ba}=t_1\sum w_{b_i}-t_2\sum w_{a_i}$$
那么\(W_{ab}>W_{ba}\Rightarrow \frac{\sum w_{b_i}}{t_2}>\frac{\sum w_{a_i}}{t_1}\),即平均数更小的放到前面更优。
所以我们就可以拿堆维护了,每次取出最小的,把它与父节点所在序列合并。这里可以用带修改堆修改父节点的状态,也可以打个标记删掉父节点的状态(其实判一下\(sz\)是否等于当前状态下的\(sz\)就可以了)。
答案就在一个序列被放到后面的时候统计就可以了。
另外堆也可以用\(STL\)里的\(heap\)。只需要#include <algorithm>
。
常数应该是比priority_queue
小一些的?可能是我写的丑...
priority_queue
:
//30268kb 2620ms
#include <queue>
#include <cstdio>
#include <cctype>
#include <cstdlib>
#include <algorithm>
//#define gc() getchar()
#define MAXIN 500000
#define gc() (SS==TT&&(TT=(SS=IN)+fread(IN,1,MAXIN,stdin),SS==TT)?EOF:*SS++)
typedef long long LL;
const int N=5e5+5;
int Cnt,fa[N],Enum,H[N],nxt[N],to[N],F[N],sz[N];
LL w[N];
char IN[MAXIN],*SS=IN,*TT=IN;
struct Node
{
int x,sz; LL val;
bool operator <(const Node &x)const
{
return val*x.sz>x.val*sz;
}
};
inline int read()
{
int now=0;register char c=gc();
for(;!isdigit(c);c=gc());
for(;isdigit(c);now=now*10+c-48,c=gc());
return now;
}
inline void AE(int u,int v)
{
to[++Enum]=v, nxt[Enum]=H[u], H[u]=Enum;
}
void DFS(int x)
{
static bool vis[N];
++Cnt, vis[x]=1;
for(int i=H[x]; i; i=nxt[i])
if(!vis[to[i]]) DFS(to[i]);
else {puts("-1"); exit(0);}
}
int Find(int x)
{
return x==F[x]?x:F[x]=Find(F[x]);
}
int main()
{
// freopen("perm.in","r",stdin);
// freopen("perm.out","w",stdout);
const int n=read();
for(int i=1; i<=n; ++i) AE(fa[i]=read(),i);
DFS(0);
if(Cnt<=n) return puts("-1"),0;
std::priority_queue<Node> q;
for(int i=0; i<=n; ++i) F[i]=i, sz[i]=1;//from 0
for(int i=1; i<=n; ++i) q.push((Node){i,1,w[i]=read()});
LL ans=0;
while(!q.empty())
{
Node tmp=q.top(); q.pop();
int x=Find(tmp.x);
if(tmp.sz!=sz[x]) continue;
int r=Find(fa[x]);
F[x]=r, ans+=tmp.val*sz[r], sz[r]+=sz[x], w[r]+=w[x];
if(r) q.push((Node){r,sz[r],w[r]});
}
printf("%lld\n",ans);
return 0;
}
heap
:
//25792kb 2716ms
#include <cstdio>
#include <cctype>
#include <cstdlib>
#include <algorithm>
//#define gc() getchar()
#define MAXIN 500000
#define gc() (SS==TT&&(TT=(SS=IN)+fread(IN,1,MAXIN,stdin),SS==TT)?EOF:*SS++)
typedef long long LL;
const int N=5e5+5;
int Cnt,fa[N],Enum,H[N],nxt[N],to[N],F[N],sz[N];
LL w[N];
char IN[MAXIN],*SS=IN,*TT=IN;
struct Node
{
int x,sz; LL val;
bool operator <(const Node &x)const
{
return val*x.sz>x.val*sz;
}
}q[N];
inline int read()
{
int now=0;register char c=gc();
for(;!isdigit(c);c=gc());
for(;isdigit(c);now=now*10+c-48,c=gc());
return now;
}
inline void AE(int u,int v)
{
to[++Enum]=v, nxt[Enum]=H[u], H[u]=Enum;
}
void DFS(int x)
{
static bool vis[N];
++Cnt, vis[x]=1;
for(int i=H[x]; i; i=nxt[i])
if(!vis[to[i]]) DFS(to[i]);
else {puts("-1"); exit(0);}
}
int Find(int x)
{
return x==F[x]?x:F[x]=Find(F[x]);
}
int main()
{
// freopen("perm.in","r",stdin);
// freopen("perm.out","w",stdout);
const int n=read();
for(int i=1; i<=n; ++i) AE(fa[i]=read(),i);
DFS(0);
if(Cnt<=n) return puts("-1"),0;
for(int i=0; i<=n; ++i) F[i]=i, sz[i]=1;//from 0
for(int i=1; i<=n; ++i) q[i]=(Node){i,1,w[i]=read()};
LL ans=0; int cnt=n;
std::make_heap(q+1,q+1+cnt);
while(cnt)
{
Node tmp=q[1]; std::pop_heap(q+1,q+1+cnt), --cnt;
int x=Find(tmp.x);
if(tmp.sz!=sz[x]) continue;
int r=Find(fa[x]);
F[x]=r, ans+=tmp.val*sz[r], sz[r]+=sz[x], w[r]+=w[x];
if(r) q[++cnt]=(Node){r,sz[r],w[r]}, std::push_heap(q+1,q+1+cnt);
}
printf("%lld\n",ans);
return 0;
}
BZOJ.5289.[AHOI/HNOI2018]排列(贪心 heap)的更多相关文章
- BZOJ.5288.[AHOI/HNOI2018]游戏(思路 拓扑)
BZOJ LOJ 洛谷 考虑如何预处理每个点能到的区间\([l,r]\). 对于\(i,i+1\)的一扇门,如果钥匙在\(i\)的右边,连边\(i\to i+1\),表示从\(i\)出发到不了\(i+ ...
- BZOJ.5290.[AHOI/HNOI2018]道路(树形DP)
BZOJ LOJ 洛谷 老年退役选手,都写不出普及提高DP= = 在儿子那统计贡献,不是在父亲那统计啊!!!(这样的话不写这个提高DP写记忆化都能过= =) 然后就令\(f[x][a][b]\)表示在 ...
- BZOJ.5287.[AHOI HNOI2018]毒瘤(虚树 树形DP)
BZOJ LOJ 洛谷 设\(f[i][0/1]\)表示到第\(i\)个点,不选/选这个点的方案数.对于一棵树,有:\[f[x][0]=\prod_{v\in son[x]}(f[v][0]+f[v] ...
- BZOJ.5285.[AHOI/HNOI2018]寻宝游戏(思路 按位计算 基数排序..)
BZOJ LOJ 洛谷 话说vae去年的专辑就叫寻宝游戏诶 只有我去搜Mystery Hunt和infinite corridor了吗... 同样按位考虑,假设\(m=1\). 我们要在一堆\(01\ ...
- BZOJ.5286.[AHOI/HNOI2018]转盘(线段树)
BZOJ LOJ 洛谷 如果从\(1\)开始,把每个时间\(t_i\)减去\(i\),答案取决于\(\max\{t_i-i\}\).记取得最大值的位置是\(p\),答案是\(t_p+1+n-1-p=\ ...
- 5289: [Hnoi2018]排列
5289: [Hnoi2018]排列 链接 分析: 首先将题意转化一下:每个点向a[i]连一条边,构成了一个以0为根节点的树,要求选一个拓扑序,点x是拓扑序中的第i个,那么价值是i*w[x].让价值最 ...
- 【BZOJ5289】[HNOI2018]排列(贪心)
[BZOJ5289][HNOI2018]排列(贪心) 题面 BZOJ 洛谷 题解 这个限制看起来不知道在干什么,其实就是找到所有排列\(p\)中,\(p_k=x\),那么\(k<j\),其中\( ...
- loj #2509. 「AHOI / HNOI2018」排列
#2509. 「AHOI / HNOI2018」排列 题目描述 给定 nnn 个整数 a1,a2,…,an(0≤ai≤n),以及 nnn 个整数 w1,w2,…,wn.称 a1,a2,…,an 的 ...
- Loj #2495. 「AHOI / HNOI2018」转盘
Loj #2495. 「AHOI / HNOI2018」转盘 题目描述 一次小 G 和小 H 原本准备去聚餐,但由于太麻烦了于是题面简化如下: 一个转盘上有摆成一圈的 \(n\) 个物品(编号 \(1 ...
随机推荐
- cf869C组合计数问题
如果在两个区域里连点,两个区域内选的点数一定要相等 即a中选出i个点,必须与b中选出i个点相连 连接种类数为 然后我们再来看,如果ab中有两点相连,其中一点再与c相连会出事吗? 很显然不会对答案产生 ...
- C++ Primer 笔记——标准库类型string
1.如果使用等号初始化一个变量,实际上执行的是拷贝初始化,编译器吧等号右侧的初始值拷贝到新创建的对象中去:如果不使用等号则执行的是直接初始化. std::string str = "Test ...
- 出现xml错误的时候都是配置文件的名字没有改造成的
The error may exist in com/bjpowernode/dao/PlayerDao.xml
- 重新认识python
为什么这样说呢,我几个月前就开始学python,但是一直都没有进步,还就只是会一些其它语言的共性的问题,也就是新学习的约等于0. 后来一直找一些适合自己的教材,通过同学找到了一个学长的教程. 开始了新 ...
- Notepad++文件自动更新
- 一脸懵逼学习HBase---基于HDFS实现的。(Hadoop的数据库,分布式的,大数据量的,随机的,实时的,非关系型数据库)
1:HBase官网网址:http://hbase.apache.org/ 2:HBase表结构:建表时,不需要指定表中的字段,只需要指定若干个列族,插入数据时,列族中可以存储任意多个列(即KEY-VA ...
- CentOS6.9安装socat
yum -y install epel-release yum -y install socat
- javascript 对象(四)
一.对象概述 对象中包含一系列的属性,这些属性是无序的.每个属性都有一个字符串key和对应的value. var obj={x:1,y:2}; obj.x; obj.y; 1.为什么属性的key必须是 ...
- [转]xshell使用技巧
https://yq.aliyun.com/articles/44721 xshell是我用过的最好用的ssh客户端工具,没有之一.这个软件完全免费,简单易用,可以满足通过ssh管理linux vps ...
- Hbase的写入负载均衡
在写Hbase的时候,会担心数据分布在各个region上不均匀(与预分区无关): 这个时候可能想到的方式:hash.加盐等(当然是可以的,但是读取的时候咋办呢? 哪天写一个这样的文章) 本文采用新方式 ...