[Tyvj P4875]排列

原題鏈接

一開始想的是n²做法
預處理+n²枚舉
然後看了題解

利用單調棧
固定左端點
右端點跳到最大最小值發生改變的地方
時間複雜度nlogn

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<queue>
#include<vector>
#include<climits>
#include<string>
#include<cstdlib>
#include<ctime>
#define MOD 1000000007
#define LL long long
using namespace std;

LL t,n,a[100005],rmax[100005],rmin[100005],top,q[100005],maxn,minn,j,j1,j2,d,ans[100005];

void pre()
{
    LL i;

    q[0]=n+1;
    top=0;

    for(i=n;i>=1;i--)
    {
        while(top&&a[q[top]]<=a[i]) top--;

        rmax[i]=q[top];
        top++;
        q[top]=i;   
    }

    q[0]=n+1;   
    top=0;

    for(i=n;i>=1;i--)
    {
        while(top&&a[q[top]]>=a[i]) top--;

        rmin[i]=q[top];
        top++;
        q[top]=i;   
    }
}

int main()
{
    LL i;

    scanf("%lld",&t);

    while(t--)
    {
        memset(ans,0,sizeof(ans));

        scanf("%lld",&n);

        for(i=1;i<=n;i++)
            scanf("%lld",&a[i]);

        pre();

        for(i=1;i<=n;i++)
        {
            j=i;
            j1=i;
            j2=i;

            while(j<=n)
            {
                if(rmax[j1]<rmin[j2])
                {
                    d=rmax[j1]-j;
                    ans[a[j1]-a[j2]]+=d;

                    j=rmax[j1];
                    j1=rmax[j1];
                }           

                else
                {
                    d=rmin[j2]-j;
                    ans[a[j1]-a[j2]]+=d;

                    j=rmin[j2];
                    j2=rmin[j2];            
                }               
            }
        }

        for(i=1;i<n;i++) ans[i]+=ans[i-1];
        for(i=0;i<n;i++) printf("%lld\n",ans[i]);
    }

    return 0;
}