#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;
}
AcWing 1088 旅行问题 题解(动态规划—DP—单调队列优化DP)
于 2022-03-03 22:25:57 首次发布