2018.1.25 T3 数列问题

本文链接：https://blog.csdn.net/ggn_2015/article/details/79165455

在学校住了两天感觉挺好的，但因为Goseqh回家了，我也只好回家了。

我最近写程序可能是状态比较差，漏洞百出，各种诡异的错误挑不错来。调试了好长时间这道题才调过。

【题面】

题目描述

更正，当k=0时，满足条件的数不能为这个数自身，即 $i \neq j$ 。

输入描述

输出描述

样例输入

样例输出和数据范围

【思路】

离线 + 线段树。

我们可以出一个数组 $d$ ， $d_i=\min\{j:|a_i-a_j|=k,j>i\}$ ,若不存在满足条件的j，令 $d_i=+\infty$ 。可以从右到左扫描当前序列，用map记录每种数值最后一次出现（即下标最靠前一次出现）时的位置。对于每一个位置 $i$ ，有 $d_i=min(LastExist(a_i+k),LastExist(a_i-k))-i$ 。然后再更新： $LastExist(a_i)=i$ 。

考虑假如每次询问的x都等于1，那有什么巧妙的解决问题的办法。

我们不难发现所有的 $i< j,|a_i-a_j|=k$ （这里的 $j$ 实际上是 $d_i$ ,因为 $d_i$ 一定是i右面最近的那个j），只要i和j都在1~y的范围内， $j-i$ 就会称为可能的答案。我们不妨记录一个数组 $b$ ，令 $b_j=j-i$ （若不存在i, $b_j=+\infty$ ），这样的话我们只需要查询b数组在1~y上的最小值就可以解决问题。因为右边界比左边界更不容易进入到1~y的范围里，而只有这两个边界，同时进入范围内才可能对答案造成贡献。

显然，如果算出的最小值为 $+\infty$ 那么就说明没有满足条件的i和j，输出-1就可以了。

按照相似的思路，我们可以把所有询问读入，按照x值（即左端点值）从大到小排序。

用线段树维护一个支持单点修改，查询区间最小值的，长度为n的序列 $\{b_i\}$ 。记录一个L值，L值从右向左推移。L值每到达一个位置 $i$ ，就让 $b_{d_i}=\min(b_{d_i},d_i-i)$ 。按照预先排好的顺序考虑所有询问，每一次把L值向左推到当前询问的x的位置，查询 $\min\{b_x,b_{x+1},...,b_y\}$ 即为答案。（L就相当于上上一段中的“1”，是一个固定了的起始位置）

【代码】

其实还是蛮好写的，但是我状态不太好，写错了好多地方，好久才调试出来。

#include<cstdio>
#include<map>
#include<algorithm>
using namespace std;

#define maxn (100000 + 10)
#define inf  (0x7f7f7f7f)

int lch[maxn], rch[maxn], msg[maxn], ncnt;//min
int lst[maxn], rst[maxn];
void build(int root, int l, int r){
    lst[root] = l; rst[root] = r;
    if(l == r){
        lch[root] = rch[root] = -1; //no son
        msg[root] = inf;
    }else{
        int mid = (l+r)/2;
        int nl = ++ncnt, nr = ++ncnt;
        build(nl, l, mid);
        build(nr, mid+1, r);
        msg[root] = min(msg[nl], msg[nr]);
        lch[root] = nl; rch[root] = nr;
    }
}

int query(int root, int l, int r){
    if(lst[root]>r || rst[root]<l) return inf;
    if(l<=lst[root] && rst[root]<=r) return msg[root];
    int nl = lch[root], nr = rch[root];
    return min(query(nl, l, r),query(nr, l, r));
}

void change(int root, int pos, int val){
    if(lst[root] == rst[root]){
        msg[root] = min(msg[root], val);
    }else{
        int mid = (lst[root] + rst[root])/2;
        if(pos <= mid){
            change(lch[root], pos, val);
        }else change(rch[root], pos, val);
        msg[root] = min(msg[lch[root]], msg[rch[root]]);
    }
}

int a[maxn], d[maxn], n, k;
map<int, int> lastp;

#define POS(X) ((X) == 0 ? inf : (X))
#define ANS(X) ((X) >= inf ? -1 : (X))

struct que{
    int l, r, t;
}qs[maxn]; int ans[maxn];

inline bool cmp(que a, que b){
    return a.l > b.l;
}

int main(){
    freopen("sequence.in","r",stdin);
    freopen("sequence.out","w",stdout);
    scanf("%d%d", &n, &k);
    for(int i=1; i<=n; i++) scanf("%d", &a[i]);
    for(int i=n; i>=1; i--){
        d[i] = min(POS(lastp[a[i] + k]), POS(lastp[a[i] - k]));
        lastp[a[i]] = i;
    }
    int q; scanf("%d", &q);
    for(int i=1; i<=q; i++) scanf("%d%d", &qs[i].l, &qs[i].r), qs[i].t=i;
    int L = n+1, root = ++ncnt;
    build(root, 1, n);
    sort(qs+1, qs+q+1, cmp);
    for(int i=1; i<=q; i++){
        while(L > qs[i].l){
            L--;
            if(d[L] != inf) change(root, d[L], d[L]-L);
        }
        ans[qs[i].t] = ANS(query(root, qs[i].l, qs[i].r));
    }
    for(int i=1; i<=q; i++){
        printf("%d\n", ans[i]);
    }
    return 0;
}

/*
7 1
2 1 3 8 4 2 1
4
1 5
4 6
4 7
2 5
*/