AcWing 1088 旅行问题 题解（动态规划—DP—单调队列优化DP）

原题传送门

#include<bits/stdc++.h>

using namespace std;

const int N = 2e6 + 10;

typedef long long ll;

int o[N], d[N];
ll q[N], qq[N];
bool vis[N];
int n;

int main()
{
	scanf("%d", &n);
	
	for(int i = 1; i <= n; i ++ ){
		scanf("%d%d", &o[i], &d[i]);//记录油量和里程数 
	}
	
	for(int i = 1; i <= n; ++ i){
		q[i] = q[i + n] = o[i] - d[i];//q在此时记录的是第i段路剩余的油量 
	}
	
	for(int i = 1; i <= 2 * n; ++ i){
		q[i] += q[i - 1];//计算剩余油量的前缀和，因为每一段路都应该有剩余油量
		//所以前缀和应该是个逐渐增大的数，所以视频中所讲任意一段s[j]-s[i]>=0(j在i后面) 
	}
	
	//第一遍是顺时针进行遍历，因为前缀和是从前向后递增的，所以从后向前进行，保证后面的每个s[i]都大于前面的 
	int hh = 0, tt = -1;//单调队列 
	for(int i = 2 * n; i >= 1; -- i){ 
		if(hh <= tt && qq[hh] > i + n - 1) hh ++ ;
		//因为滑动窗口此时相当于从后向前移动，所以如果队首元素大于窗口覆盖的最后一个位置，这个队首元素就应该出队 
		//因为单调队列内的元素是正常排列，所以出队操作还是hh++ 
		while(hh <= tt && q[qq[tt]] >= q[i]) tt -- ;
		//要保持队列内元素单调递增，所以如果队尾元素大于新遍历到的元素，就将此时队尾元素出队，直到队尾元素小于新加入的元素 
		qq[ ++ tt] = i;//新元素入队 
		if(i <= n && q[i - 1] <= q[qq[hh]]) vis[i] = 1;
		//因为滑动窗口是从后向前滑动的，所以当i<=n即代表已经将后n个元素遍历过，已经找到所有q[n~2n]中最小的值
		//如果最小值都大于q[i]，则代表从q[i]出发旅行一圈的剩余油量是递增的，即从q[i]出发可以旅行一圈，所以标记vis[i]=1 
	} 
	
	//逆时针计算
	d[0] = d[n];//要用到d[0]，所以要进行赋值 
	hh = 0, tt = -1;//清空单调队列 
	//因为是逆时针计算前缀和，所以应该是这一点的油储量减去上一段的耗油量 
	for(int i = 1; i <= n; ++ i){
		q[i] = q[i + n] = o[i] - d[i - 1];
	}
	//逆时针计算前缀和 
	for(int i = 2 * n; i >= 0; -- i){
		q[i] += q[i + 1];
	}
	
	//第二遍是逆时针进行遍历，因为前缀和是从后向前递增的，所以从前向后进行，保证后面的每个s[i]都大于后面的 
	for(int i = 1; i <= 2 * n; ++ i){
		if(hh <= tt && qq[hh] < i - n + 1) hh ++ ;
		//因为滑动窗口此时相当于从前向后移动，所以如果队首元素小于窗口覆盖的最后一个位置，这个队首元素就应该出队 
		//因为单调队列内的元素是正常排列，所以出队操作还是hh++ 
		while(hh <= tt && q[qq[tt]] >= q[i]) tt -- ;
		qq[ ++ tt] = i;
		if(i > n && q[qq[hh]] >= q[i + 1]) vis[i - n] = 1;
		//因为滑动窗口是从前向后滑动的，所以当i>n即代表已经将前n个元素遍历过，已经找到所有q[1~n]中最小的值
		//如果最小值都大于q[i]，则代表从q[i]出发旅行一圈的剩余油量是递增的，即从q[i]出发可以旅行一圈，所以标记vis[i]=1 		
		//因为是以圈替链，此时i>n，对应的出发点是i-n 
	}
	
	for(int i = 1; i <= n; ++ i ){
		if(vis[i]) printf("TAK\n");
		else printf("NIE\n");
	}
	return 0;
}