【2019集训队互测】学习轨迹（贪心）

传送门

---

题解：

首先没有限制的话排个序就完事了。

如果有限制的话，显然我们需要干的事情就是做出修正。

按照权值从小到大排序。

考虑什么情况下非法，权值小的依赖权值大的。

这个时候的修正非常明显，把这个区间拎出来，无依赖的升序，然后有依赖的降序排在后面。

这样的区间可以通过一次差分前缀和找出来。

容易注意到有的时候我们把若干个区间合并可以得到更优的答案。

简单讨论可以发现，如果不是全部合并，必然是前缀合并一段，中间全部不合并，后缀合并一段，找到这个合并的位置即可。

注意权值从大到小是等价的，我们还要把权值乘上-1再做一遍。

---

代码：

#include<bits/stdc++.h>
#define ll long long
#define re register
#define cs const

namespace IO{
	
inline char gc(){
	static cs int Rlen=1<<22|1;static char buf[Rlen],*p1,*p2;
	return (p1==p2)&&(p2=(p1=buf)+fread(buf,1,Rlen,stdin),p1==p2)?EOF:*p1++;
}template<typename T>T get_integer(){
	char c;bool f=false;while(!isdigit(c=gc()))f=c=='-';T x=c^48;
	while(isdigit(c=gc()))x=((x+(x<<2))<<1)+(c^48);return f?-x:x;
}inline int gi(){return get_integer<int>();}
inline ll gl(){return get_integer<ll>();}

char obuf[30000007],*oh=obuf,ch[23];
template<typename T>void print(T a,char c){
	if(a<0)*oh++='-',a=-a;int tl=0;
	do ch[++tl]=a%10;while(a/=10);
	while(tl)*oh++=ch[tl--]^48;*oh++=c;
}struct obuf_flusher{~obuf_flusher(){fwrite(obuf,1,oh-obuf,stdout);}}Flusher;

}using namespace IO;

using std::cerr;
using std::cout;
using pli=std::pair<ll,int>;
#define fi first
#define se second

cs int N=1e6+7;

ll w[N];int n,m,bel[N];
int id[N],ps[N],L[N],R[N]; 
ll ans[2];int Ans[2][N];
int sm[N];pli f[N];

void work(int o){
	for(int re i=1;i<=n;++i)ps[id[i]]=i;
	memset(sm+1,0,sizeof(int)*n);int ct=0;
	for(int re i=m+1;i<=n;++i)
		if(ps[i]<ps[bel[i]])++sm[ps[i]],--sm[ps[bel[i]]];
	for(int re i=2;i<=n;++i)sm[i]+=sm[i-1];
	for(int re i=n;i;--i)
		if(id[i]<=m)Ans[o][++ct]=id[i];
	for(int re i=1;i<=n;++i)
		if(id[i]>m)Ans[o][++ct]=id[i];
	for(int re i=2;i<=n;++i)
		ans[o]+=std::abs(w[Ans[o][i]]-w[Ans[o][i-1]]);
	ct=0;for(int re i=1;i<=n;++i)
		if(!sm[i-1])R[ct]=i-1,L[++ct]=i;
	R[ct]=n;if(ct>1){
		f[1]={w[id[R[1]]]-2*w[id[1]],1};
		for(int i=2;i<=ct;++i)
			f[i]=std::min<pli>(
				{w[id[R[i]]]-2*w[id[1]],i},
				{f[i-1].fi+(w[id[R[i]]]-w[id[L[i]]])*2,f[i-1].se}
			);
		ll tmp=1e18+7;int pl=0,pr=0;
		for(int re i=1;i<ct;++i){
			ll vl=f[i].fi+2*w[id[n]]-w[id[R[i]+1]];
			if(vl<tmp)tmp=vl,pr=i+1,pl=f[i].se;
		}if(ans[o]>tmp){
			ans[o]=tmp;int tl=0;
			for(int re i=R[pl];i;--i)
				if(id[i]<=m)Ans[o][++tl]=id[i];
			for(int re i=1;i<=R[pl];++i)
				if(id[i]>m)Ans[o][++tl]=id[i];
			for(int re i=pl+1;i<pr;++i){
				for(int re j=R[i];j>=L[i];--j)
					if(id[j]<=m)Ans[o][++tl]=id[j];
				for(int re j=L[i];j<=R[i];++j)
					if(id[j]>m)Ans[o][++tl]=id[j];
			}
			for(int re i=L[pr];i<=n;++i)
				if(id[i]<=m)Ans[o][++tl]=id[i];
			for(int re i=n;i>=L[pr];--i)
				if(id[i]>m)Ans[o][++tl]=id[i];
		}
	}
}

void Main(){
	n=gi(),m=gi();
	for(int re i=1;i<=n;++i)w[i]=gl();
	for(int re i=m+1;i<=n;++i)bel[i]=gi();
	for(int re i=1;i<=n;++i)id[i]=i;
	std::sort(id+1,id+n+1,
		[](int i,int j){return w[i]<w[j];});
	work(0);
	for(int re i=1;i<=n;++i)w[i]=-w[i];
	std::reverse(id+1,id+n+1);work(1);
	bool o=ans[1]<ans[0];
	print(ans[o],'\n');
	for(int re i=1;i<=n;++i)
		print(Ans[o][i],' ');
}

inline void file(){
#ifdef zxyoi
	freopen("learn.in","r",stdin);
#endif
}signed main(){file();Main();return 0;}