[CSP-S模拟测试]:Travel(贪心+构造)

题目描述

  给定一个长度为$n$的格子序列$x_1,x_2,...,x_n$。每一次$Lyra$可以选择向左跳到任意一个还没到过的位置,也可以向右跳到任意一个还没到过的位置。如果现在$Lyra$在格子$i$,她下一步跳向格子$j$,那么这次跳跃的花费为$|x_i-x_j|$。注意,跳意味着格子$i$和格子$j$中间其他的格子都不会被这次跳跃影响。并且,$Lyra$不应该跳出边界。
  $Lyra$的初始位置在格子$s$。$Lyra$将会在到访过所有格子恰好一次之后,在某个位置停下来,这样就完成了任务。
  $Lyra$想知道如果她一共向左跳了$L$次,那么她要完成任务的最小总花费是多少,并希望你输出任意一种花费最小的方案。显然如果$Lyra$向左走了$L$次,那一定会向右走$n-L-l$次。
  特殊的,如果$Lyra$没有办法完成任务,请输出一行$-1$。


输入格式

第一行,三个整数$n,L,s$,分别表示序列的大小,向左走的次数,和初始位置。
第二行,$n$个数字,表示序列$x_i$。


输出格式

第一行,一个数字,表示答案。
如果能完成任务,则第二行,输出$n-1$个数字,表示方案。注意,$Lyra$初始的位置已经确定了,所以不要输出。


样例

样例输入:

3 1 2
1 2 3

样例输出:

3
1 3


数据范围与提示

样例解释:

$Lyra$一开始在$2$的位置,$2\rightarrow 1\rightarrow 3$的路径中,$Lyra$一共向左走了$1$次,花费为$|2-1|+|1-3|=3$。

数据范围:

测试点$1\sim 2,n\leqslant 8,0\leqslant x_i \leqslant 10^9$。
测试点$3\sim 8,n\leqslant 20,0\leqslant x_i \leqslant 10^9$。
测试点$9\sim 10,n\leqslant 2\times 10^5,x_i =i$。
测试点$11\sim 20,n\leqslant 2\times 10^5,0\leqslant x_i \leqslant 10^9$。
对于所有数据,都满足$x_1<x_2<...<x_{n-1}<x_n,1\leqslant s\leqslant n,0\leqslant L\leqslant n-1$。


题解

先来考虑$-1$的情况,分为两种情况。

  $\alpha.L=0,s\neq 0$,显然我们无论如何都不能走到$s$左边的点。

  $\beta.L=n-1,s\neq n$,则我们向右走的步数为$0$,其它类比上面。

再来考虑一般情况。

$x_1<x_2<...<x_{n-1}<x_n$是一个非常好的性质,也就是说,尽可能少的覆盖区间一定是最优的;这里所说的区间是指假如我从$l$走到了$r$,那么就将区间$[l,r]$覆盖了一遍,也就是付出了$x_r-x_l$的代价(假设$l$在左,$r$在右)。

那么考虑如何选择这条路径。

这时候我们分为一开始向右走和一开始向左走,显然这两种情况无非就是将区间$reverse$一下,那么我们现在只考虑向左走的情况。

起点已经固定为$s$,不妨设终点为$t$,那么路径一定类似下图$\downarrow$

设在$s$左侧向左走了$l_1$步,在$t$右侧向右走了$l_2$步,那么可以通过调整在$s$左侧向左走的步数来满足$t$的右侧。

如果都无法满足,则意味着我们需要在$s\sim t$这段区间再向左走,如下图$\downarrow$

那么这样一定不是最优的,因为存在方案$s\sim t'$一定比$s\sim t$更优。

于是可以枚举每一个$t$,去最小即可。

路径直接构造就好了。

一开始向右走的情况同理,两种方案取$\min$即可。

时间复杂度:$\Theta(n)$。

期望得分:$100$分。

实际得分:$100$分。


代码时刻

#include<bits/stdc++.h>
using namespace std;
int n,l,s;
int xl[200001],xr[200001],top;
bool vis[200001];
int ansl,ansr,quel[200001],quer[200001],pos[200001];
pair<int,int> sta[200001];
void solvel()
{
	if(l<s)
	{
		for(int i=s-1;i>=s-l+1;i--)quel[++quel[0]]=i;
		for(int i=1;i<=s-l;i++)quel[++quel[0]]=i;
		for(int i=s+1;i<=n;i++)quel[++quel[0]]=i;
		ansl=xl[s]+xl[n]-(xl[1]<<1);
		return;
	}
	if(l+1==n-1)
	{
		for(int i=s-1;i;i--)quel[++quel[0]]=i;
		for(int i=n;i>s;i--)quel[++quel[0]]=i;
		ansl=((xl[n]-xl[1])<<1)+xl[s]-xl[s+1];
		return;
	}
	for(int i=s+2;i<n;i++)sta[++top]=make_pair(xl[i]-xl[i-1],i);
	sort(sta+1,sta+top+1);
	for(int i=1;i<=top;i++)pos[sta[i].second]=i;
	int res=0;
	for(int i=1;i<=l-s+1;i++)res+=sta[i].first;
	int mx=res<<1,t=n,j=l-s+1;
	for(int i=n-1,flag=l-s+1;i>=n-l+s-1;i--)
	{
		if(pos[i]<=flag)res-=sta[pos[i]].first;
		else res-=sta[flag--].first;
		while(flag&&i<=sta[flag].second)flag--;
		if(mx>(res<<1)+xl[n]-xl[i]){mx=(res<<1)+xl[n]-xl[i];t=i;j=flag;}
	}
	for(int i=s-1;i;i--)quel[++quel[0]]=i;
	for(int i=s+2;i<t;i++)if(pos[i]<=j)vis[i]=1;
	for(int i=s+1;i<t;i++)
	{
		if(!vis[i+1])quel[++quel[0]]=i;
		else
		{
			int flag=i+1;
			while(vis[flag])flag++;
			for(int j=flag-1;j>=i;j--)quel[++quel[0]]=j;
			i=flag-1;
		}
	}
	for(int i=n;i>=t;i--)quel[++quel[0]]=i;
	ansl=xl[s]+xl[n]-2*xl[1]+mx;
}
void solver()
{
	top=0;
	memset(vis,0,sizeof(vis));
	if(l+2>s)
	{
		for(int i=n-s;i>=l-s+3;i--)quer[++quer[0]]=i;
		for(int i=1;i<=l-s+2;i++)quer[++quer[0]]=i;
		for(int i=n-s+2;i<=n;i++)quer[++quer[0]]=i;
		ansr=xr[n-s+1]+xr[n]-(xr[1]<<1);
		return;
	}
	if(l==1)
	{
		for(int i=n-s;i;i--)quer[++quer[0]]=i;
		for(int i=n;i>n-s+1;i--)quer[++quer[0]]=i;
		ansr=((xr[n]-xr[1])<<1)+xr[n-s+1]-xr[n-s+2];
		return;
	}
	for(int i=n-s+3;i<n;i++)sta[++top]=make_pair(xr[i]-xr[i-1],i);
	sort(sta+1,sta+top+1);
	for(int i=1;i<=top;i++)pos[sta[i].second]=i;
	int res=0;
	for(int i=1;i<=s-l-1;i++)res+=sta[i].first;
	int mx=res<<1,t=n,j=s-l-1;
	for(int i=n-1,flag=s-l-1;i>=l-s+n+1;i--)
	{
		if(pos[i]<=flag)res-=sta[pos[i]].first;
		else res-=sta[flag--].first;
		while(flag&&i<=sta[flag].second)flag--;
		if(mx>(res<<1)+xr[n]-xr[i]){mx=(res<<1)+xr[n]-xr[i];t=i;j=flag;}
	}
	for(int i=n-s;i;i--)quer[++quer[0]]=i;
	for(int i=n-s+3;i<t;i++)if(pos[i]<=j)vis[i]=1;
	for(int i=n-s+2;i<t;i++)
	{
		if(!vis[i+1])quer[++quer[0]]=i;
		else
		{
			int flag=i+1;
			while(vis[flag])flag++;
			for(int j=flag-1;j>=i;j--)quer[++quer[0]]=j;
			i=flag-1;
		}
	}
	for(int i=n;i>=t;i--)quer[++quer[0]]=i;
	ansr=xr[n-s+1]+xr[n]-2*xr[1]+mx;
}
int main()
{
	scanf("%d%d%d",&n,&l,&s);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&xl[i]);
		xr[n-i+1]=-xl[i];
	}
	if((!l&&s!=1)||(l==n-1&&s!=n)){puts("-1");return 0;}
	solvel();solver();
	if(ansl<ansr)
	{
		printf("%d\n",ansl);
		for(int i=1;i<n;i++)printf("%d ",quel[i]);
	}
	else
	{
		printf("%d\n",ansr);
		for(int i=1;i<n;i++)printf("%d ",n-quer[i]+1);
	}
	return 0;
}

rp++

posted @ 2019-10-10 17:03  HEOI-动动  阅读(221)  评论(0编辑  收藏  举报