CodeForces - 1208 D.Restore Permutation (线段树)

Restore Permutation

An array of integers p1,p2,…,pn is called a permutation if it contains each number from 1 to n exactly once. For example, the following arrays are permutations: [3,1,2],[1],[1,2,3,4,5] and [4,3,1,2]. The following arrays are not permutations: [2],[1,1],[2,3,4].

There is a hidden permutation of length n.

For each index i, you are given si, which equals to the sum of all pj such that j<i and pj<pi. In other words, si is the sum of elements before the i-th element that are smaller than the i-th element.

Your task is to restore the permutation.

Input

The first line contains a single integer n (1≤n≤2⋅105) — the size of the permutation.

The second line contains n integers s1,s2,…,sn (0≤si≤n(n−1)2).

It is guaranteed that the array s corresponds to a valid permutation of length n.

Output

Print n integers p1,p2,…,pn — the elements of the restored permutation. We can show that the answer is always unique.

Examples
Input

3
0 0 0

Output

3 2 1

Input

2
0 1

Output

1 2

Input

5
0 1 1 1 10

Output

1 4 3 2 5

题意:

给数组s,s(i)表示p数组中在p(i)前面的所有小于p(i)的数的和
要求恢复p数组

分析:

容易发现1肯定是在s数组中最后一个0的位置,
假设该位置为x,我们把x后面的所有数都减去1
然后2肯定是s数组中最后一个0的位置,
假设该位置为x,我们把x后面的所有数都减去2
然后3肯定是s数组中最后一个0的位置,
假设该位置为x,我们把x后面的所有数都减去3

操作n次之后就得到所有数的位置啦!

另外为了防止用过的0影响之后的查找,每次找到x之后,把x改为inf

code:
#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
#define int long long
//typedef long long ll;
using namespace std;
const int maxm=2e5+5;
int mi[maxm<<2];
int laz[maxm<<2];
int p[maxm];
int n;
void pushup(int node){
    mi[node]=min(mi[node*2],mi[node*2+1]);
}
void pushdown(int node){
    if(laz[node]){
        mi[node*2]+=laz[node];
        mi[node*2+1]+=laz[node];
        laz[node*2]+=laz[node];
        laz[node*2+1]+=laz[node];
        laz[node]=0;
    }
}
void build(int l,int r,int node){
    if(l==r){
        cin>>mi[node];
        return ;
    }
    int mid=(l+r)/2;
    build(l,mid,node*2);
    build(mid+1,r,node*2+1);
    pushup(node);
}
int ask(int x,int l,int r,int node){
    if(l==r){
        mi[node]=(1LL<<60);//改成inf
        return l;
    }
    int mid=(l+r)/2;
    pushdown(node);
    int ans=1;
    if(mi[node*2+1]<=x)ans=ask(x,mid+1,r,node*2+1);
    else ans=ask(x,l,mid,node*2);
    pushup(node);
    return ans;
}
void update(int st,int ed,int val,int l,int r,int node){
    if(st<=l&&ed>=r){
        mi[node]+=val;
        laz[node]+=val;
        return ;
    }
    pushdown(node);
    int mid=(l+r)/2;
    if(st<=mid)update(st,ed,val,l,mid,node*2);
    if(ed>=mid)update(st,ed,val,mid+1,r,node*2+1);
    pushup(node);
}
signed main(){
    ios::sync_with_stdio(0);
    cin>>n;
    build(1,n,1);
    for(int i=1;i<=n;i++){
        int x=ask(0,1,n,1);
        p[x]=i;
        if(x!=n){//必须加这个,否则x+1可能大于n,然后就炸了
            update(x+1,n,-i,1,n,1);
        }
    }
    for(int i=1;i<=n;i++){
        cout<<p[i]<<' ';
    }
    cout<<endl;
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
CodeForces - 616D是一个关于找到一个序列中最长的第k好子段的起始位置和结束位置的问题。给定一个长度为n的序列和一个整数k,需要找到一个子段,该子段中不超过k个不同的数字。题目要求输出这个序列最长的第k好子段的起始位置和终止位置。 解决这个问题的方法有两种。第一种方法是使用尺取算法,通过维护一个滑动窗口来记录\[l,r\]中不同数的个数。每次如果这个数小于k,就将r向右移动一位;如果已经大于k,则将l向右移动一位,直到个数不大于k。每次更新完r之后,判断r-l+1是否比已有答案更优来更新答案。这种方法的时间复杂度为O(n)。 第二种方法是使用枚举r和双指针的方法。通过维护一个最小的l,满足\[l,r\]最多只有k种数。使用一个map来判断数的种类。遍历序列,如果当前数字在map中不存在,则将种类数sum加一;如果sum大于k,则将l向右移动一位,直到sum不大于k。每次更新完r之后,判断i-l+1是否大于等于y-x+1来更新答案。这种方法的时间复杂度为O(n)。 以上是两种解决CodeForces - 616D问题的方法。具体的代码实现可以参考引用\[1\]和引用\[2\]中的代码。 #### 引用[.reference_title] - *1* [CodeForces 616 D. Longest k-Good Segment(尺取)](https://blog.csdn.net/V5ZSQ/article/details/50750827)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^koosearch_v1,239^v3^insert_chatgpt"}} ] [.reference_item] - *2* [Codeforces616 D. Longest k-Good Segment(双指针+map)](https://blog.csdn.net/weixin_44178736/article/details/114328999)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^koosearch_v1,239^v3^insert_chatgpt"}} ] [.reference_item] [ .reference_list ]
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值