BZOJ5289 洛谷4437:[HNOI/AHOI2018]排列——题解
https://www.lydsy.com/JudgeOnline/problem.php?id=5289
https://www.luogu.org/problemnew/show/P4437
考慮對(duì)于a[i]=m,a[m]=n,我們令p[j]=i,p[k]=m(一定會(huì)有一對(duì)(j,k)滿足這個(gè)條件的),則我們會(huì)有p[k]=a[p[j]],此時(shí)我們要滿足k<j,也就是a[m]放的位置要比a[i]靠前。
也就是說(shuō)選第m個(gè)之后才能選第i個(gè)。
轉(zhuǎn)換成圖論模型就是m->i <=> a[i]->i。
那么問(wèn)題豁然開朗,首先如果圖中有環(huán)就能判斷無(wú)解了。
如果有解則必然為以0為根的有向樹,則答案為每個(gè)節(jié)點(diǎn)i被拿到的時(shí)間*w[i](前提是i的父親被拿才可以拿i)。
再考慮最大化答案,則我們讓樹中最小值min盡可能早的被拿到就好了。
繼續(xù)貪心,則如果當(dāng)前局面能夠拿到min則一定拿min,換句話將就是拿了min的父親就一定拿min。
那么父親和min之間就成了捆綁關(guān)系,于是將其縮起來(lái),在縮的過(guò)程中更新答案,然后遞歸這個(gè)過(guò)程就好了。
每次找min可以用堆維護(hù),復(fù)雜度O(nlogn)。
(PS:更新答案,每次更新顯然是這個(gè)點(diǎn)前面一些點(diǎn)被選了,于是它一定產(chǎn)生了前面這些點(diǎn)個(gè)數(shù)*w[該點(diǎn)]的價(jià)值)
那么就需要考慮我們縮完的點(diǎn)的w要怎么計(jì)算,對(duì)于兩個(gè)集合a,b要合并,顯然用在計(jì)算上的w=wa+wb,但是用堆排序的時(shí)候就不能這么做了。
自然能想到取平均值,雖然我不會(huì)證明,不過(guò)考量一下發(fā)現(xiàn)差不多。
#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。 +
?+歡迎訪問(wèn)我的博客:http://www.cnblogs.com/luyouqi233/+
+++++++++++++++++++++++++++++++++++++++++++
轉(zhuǎn)載于:https://www.cnblogs.com/luyouqi233/p/9112242.html
創(chuàng)作挑戰(zhàn)賽新人創(chuàng)作獎(jiǎng)勵(lì)來(lái)咯,堅(jiān)持創(chuàng)作打卡瓜分現(xiàn)金大獎(jiǎng)總結(jié)
以上是生活随笔為你收集整理的BZOJ5289 洛谷4437:[HNOI/AHOI2018]排列——题解的全部?jī)?nèi)容,希望文章能夠幫你解決所遇到的問(wèn)題。
- 上一篇: Taro+react开发(78):tar
- 下一篇: [react] 请说说react中Por