[Acwing] 1088. 旅行问题 单调队列问题

前言

传送门 :

思路

我们考虑 $a[i]=oil[i]-dist[i]$也就是如果从当前点顺时针走,需要消耗多少油,在考虑邮箱是空的情况下

我们对$a[i]$求一个前缀和,因此问题合法化就转化为$sum[i]-sum[j]>0$

我们通过移动$sum[i]>sum[j]$,因此我们考虑对于经过的车站中$sum[j{]}_{min}是否小于sum[i]$,因此就变成了一个单调队列问题

MYcode

void solve()
{
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>oil[i]>>dist[i];
		s[i] = s[i+n] = oil[i] - dist[i];
	}
	
	for(int i=1;i<=n*2;i++) s[i]+=s[i-1];
	
	int  hh =0 , tt = -1;
	
	for(int i = n * 2;i >= 1;i --)
        {
            //起点为i的情况下，最长路径的为n,所以超过i + n就是不合法的
            if(hh <= tt && q[hh] > i + n - 1) hh ++;

            while(hh <= tt && s[q[tt]] >= s[i]) tt --;
            q[++ tt] = i;

            //计算顺时针(因为s[i]固定，所以要求最小的s[q[hh]]（s[j]），因为如果最小的s[j] - s[i]都符合要求，那么剩下的的也一定符合要求)
            if(i <= n && s[q[hh]] - s[i - 1] >= 0) ans[i] = true;
            //因为j的范围是i <= j <= i - n - 1，所以q[hh]有可能要用到当前i的值，所以要在while之后计算
        }
    
    dist[0] = dist[n];
    for(int i=n;i>=1;i -- ) s[i] = s[i+n] = oil[i] - dist[i - 1];
      for(int i = n * 2;i >= 1;i --) s[i] += s[i + 1];
        hh = 0,tt = -1;
        for(int i = 1;i <= n * 2;i ++)
        {
            //逆时针的情况下,起点为i的情况下，q[hh]要在前半段[i - n + 1,i]中找
            if(hh <= tt && q[hh] < i - n + 1) hh ++;


            while(hh <= tt && s[q[tt]] >= s[i]) tt --;
            q[++ tt] = i;

            //计算逆时针(因为s[i]固定，所以要求最小的s[q[hh]]（s[j]），因为如果最小的s[j] - s[i]都符合要求，那么剩下的的也一定符合要求)
            if(i >= n + 1 && s[q[hh]] - s[i + 1] >= 0) ans[i - n] = true;
            //因为j的范围是:i - n + 1 <= j <= i,所以q[hh]有可能要用到当前i的值，所以要在while之后计算
        }
        
      for (int i = 1; i <= n; i ++ )
        if (ans[i]) puts("TAK");
        else puts("NIE");

        
}