P1081 [NOIP2012 提高组] 开车旅行

思路:

首先令 \(nxt1_i\) 表示右侧最近的城市距离(\(id1_i\) 为编号),令 \(nxt2_i\) 表示右侧第二近的城市编号(\(id2_i\) 为编号);可以使用 set 找出离这个城市最近的 \(4\) 个城市(前面两个,后面两个)。

定义:

  • \(f_{i,j}\) 表示从 \(i\) 点出发走 \(2^j\) 轮最后到达的位置。

  • \(dp1_{i,j}\) 表示从 \(i\) 点出发走 \(2^j\) 轮最后 A 走过的距离。

  • \(dp2_{i,j}\) 表示从 \(i\) 点出发走 \(2^j\) 轮最后 B 走过的距离。

初始化:

\[f_{i,0} = id1_{id2_i} \]

\[dp1_{i,0} = nxt2_{i} \]

\[dp2_{i,0} = nxt1_{id2_i} \]

状态转移方程为:

\[f_{i,j} = f_{f_{i,j-1},j-1} \]

\[dp1_{i,j} = dp1_{i,j-1} + dp1_{f_{i,j-1},j-1} \]

\[dp2_{i,j} = dp2_{i,j-1} + dp2_{f_{i,j-1},j-1} \]

此时对于询问 \(1\) 和询问 \(2\):

  • 本质上是求出从每个城市出发后 \(A\) 走的距离与 \(B\) 走的距离。

  • 那么考虑从高位到低位贪心,即设当前跳到了 \(s\) 点,若 \(dp1_{s,i} + dp2_{s,i} \le x\),可以从 \(s\) 跳到 \(f_{s,i}\),需要令 \(x \gets x - (dp1_{s,i} + dp2_{s,i})\),然后继续遍历 \(i-1\) 位。

  • 因为是 A 先开车,所以 A 可能会在最后一轮结束后还能再开上一次,需要特判。

时间复杂度为 \(O((N+Q) \log N)\)。

完整代码:

cpp 复制代码
#include<bits/stdc++.h>
#define Add(x,y) (x+y>=mod)?(x+y-mod):(x+y)
#define lowbit(x) x&(-x)
#define pi pair<ll,ll>
#define pii pair<ll,pair<ll,ll>>
#define iip pair<pair<ll,ll>,ll>
#define ppii pair<pair<ll,ll>,pair<ll,ll>>
#define fi first
#define se second
#define full(l,r,x) for(auto it=l;it!=r;it++) (*it)=x
#define Full(a) memset(a,0,sizeof(a))
#define open(s1,s2) freopen(s1,"r",stdin),freopen(s2,"w",stdout);
using namespace std;
typedef double db;
typedef unsigned long long ull;
typedef long long ll;
bool Begin;
const ll N=1e5+10,M=18,INF=1e18;
inline ll read(){
    ll x=0,f=1;
    char c=getchar();
    while(c<'0'||c>'9'){
        if(c=='-')
          f=-1;
        c=getchar();
    }
    while(c>='0'&&c<='9'){
        x=(x<<1)+(x<<3)+(c^48);
        c=getchar();
    }
    return x*f;
}
inline void write(ll x){
	if(x<0){
		putchar('-');
		x=-x;
	}
	if(x>9)
	  write(x/10);
	putchar(x%10+'0');
}
struct Node{
	ll a,b;
	ll id;
	bool operator<(const Node&rhs)const{
		if(a!=rhs.a)
		  return a<rhs.a;
		return b<rhs.b;
	}
};
ll n,m,s,x,x0,s0,cnt,dis1,dis2,disa,disb;
ll a[N],nxt1[N],nxt2[N],id1[N],id2[N];
ll f[N][M],dp1[N][M],dp2[N][M];
Node h[N];
vector<Node> V;
multiset<Node> S;
void solve(ll s,ll x){
	dis1=dis2=0;
	for(int i=M-1;i>=0;i--){
		if(dp1[s][i]+dp2[s][i]<=x&&f[s][i]){
			dis1+=dp1[s][i],dis2+=dp2[s][i];
			x-=dp1[s][i]+dp2[s][i];
			s=f[s][i];
		}
	}
	if(nxt2[s]<=x)
	  dis1+=nxt2[s];
}
bool End;
int main(){
	n=read();
	for(int i=1;i<=n;i++){
		a[i]=read();
		h[i]={a[i],a[i],i};
	}
	S.insert({INF,INF,0});
	S.insert({INF-1,INF-1,0});
	S.insert({-INF,-INF,0});
	S.insert({-INF+1,-INF+1,0});
	for(int i=n;i>=1;i--){
		V.clear();
		V.push_back(*--S.lower_bound(h[i]));
		V.push_back(*--S.lower_bound(V[0]));
		V.push_back(*S.upper_bound(h[i]));
		V.push_back(*S.upper_bound(V[2]));
		for(auto &v:V)
		  v.a=abs(h[i].a-v.a);
		sort(V.begin(),V.end());
		nxt1[i]=V[0].a,nxt2[i]=V[1].a;
		id1[i]=V[0].id,id2[i]=V[1].id;
		cerr<<id1[i]<<' '<<id2[i]<<'\n';
		S.insert(h[i]);
	}
	for(int i=1;i<=n;i++){
		f[i][0]=id1[id2[i]];
		dp1[i][0]=nxt2[i];
		dp2[i][0]=nxt1[id2[i]];
	}
	for(int j=1;j<M;j++){
		for(int i=n;i>=1;i--){
			f[i][j]=f[f[i][j-1]][j-1];
			dp1[i][j]=dp1[i][j-1]+dp1[f[i][j-1]][j-1];
			dp2[i][j]=dp2[i][j-1]+dp2[f[i][j-1]][j-1];
		}
	}
	x0=read();
	for(int i=1;i<=n;i++){
		solve(i,x0);
		if(dis2&&(!s0||disa*dis2>disb*dis1)){
			s0=i;
			disa=dis1,disb=dis2;
		}
	}
	write(s0);
	putchar('\n');
	m=read();
	while(m--){
		s=read(),x=read();
		solve(s,x);
		write(dis1);
		putchar(' ');
		write(dis2);
		putchar('\n');
	}
	cerr<<'\n'<<abs(&Begin-&End)/1048576<<"MB";
	return 0;
}
相关推荐
飞滕人生TYF7 小时前
动态规划 详解
算法·动态规划
SSL_lwz11 小时前
P11290 【MX-S6-T2】「KDOI-11」飞船
c++·学习·算法·动态规划
ZZZ_O^O12 小时前
【动态规划-卡特兰数——96.不同的二叉搜索树】
c++·学习·算法·leetcode·动态规划
橘子遇见BUG1 天前
算法日记 31 day 动态规划(01背包)
算法·动态规划
我是哈哈hh1 天前
专题二十二_动态规划_子序列系列问题(数组中不连续的一段)_算法专题详细总结
数据结构·c++·算法·动态规划·子序列
橘子遇见BUG1 天前
算法日记 30 day 动态规划(背包问题)
算法·动态规划
是糖不是唐2 天前
代码随想录算法训练营第五十二天|Day52 图论
c语言·算法·深度优先·动态规划·图论
Lindsey_feiren2 天前
代码随想录day44算法随想录|动态规划07
算法·动态规划
姜西西_2 天前
动态规划 之 子序列系列 算法专题
算法·动态规划
丶Darling.2 天前
Day46 | 动态规划 :线性DP 最长递增子序列&&最长连续递增子序列
算法·动态规划