快速维护一个排列两边比他大的数字的位置

考虑从小到大枚举x,每次维护一个链表,链表里只有>=x的数,那么往左往右找只要暴力跳k次,删除也是O(1)的

hdu 6058 Kanade’s sum

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=5e5+5;
int a[N];
int pos[N];
int pre[N];
int np[N];
int l[85];
int r[85];
void erases(int p){
    pre[np[p]]=pre[p];
    np[pre[p]]=np[p];
}
int main(){
    int t;scanf("%d",&t);
    while(t--){
        int n,k;
        scanf("%d%d",&n,&k);
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
            pos[a[i]]=i;
        }
        for(int i=1;i<=n;i++){
            pre[i]=i-1;
            np[i]=i+1;
        }
        ll ans=0;
        for(int num=1;num<=n;num++){
            memset(l,0,sizeof(l));
            memset(r,0,sizeof(r));
            int p=pos[num];
            int cnt1=0,cnt2=0;
            for(int s=p;cnt1<=k&&s<=n;){
                r[cnt1++]=s;
                s=np[s];
            }
            r[cnt1]=n+1;
            for(int s=p;cnt2<=k&&s>=1;){
                l[cnt2++]=s;
                s=pre[s];
            }
            l[cnt2]=0;
           /* printf("\nr--");
            for(int i=0;i<=cnt1;i++){
                printf("%d ",r[i]);
            }
            printf("\nl--");
            for(int i=0;i<=cnt2;i++){
                printf("%d ",l[i]);
            }*/
            erases(p);
            for(int j=0;j<k;j++){
                if(j<=cnt2&&k-j<=cnt1)
                ans+=(1ll*num*(l[j]-l[j+1])*(r[k-j]-r[k-j-1]));
                //printf("ans--%I64d\n",ans);
            }
        }
        printf("%I64d\n",ans);
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值