小y的序列

最新推荐文章于 2022-03-13 12:01:17 发布

whitewall_9

最新推荐文章于 2022-03-13 12:01:17 发布

阅读量317

点赞数

分类专栏：数学文章标签：线段树

本文链接：https://blog.csdn.net/qq_28409093/article/details/123130560

版权

数学专栏收录该内容

13 篇文章 0 订阅

订阅专栏

这篇博客探讨了一道编程竞赛题目，涉及序列的最大值与最小值差值问题。作者通过建立线段树数据结构，在O(N log N)的时间复杂度内解决了寻找满足特定条件子序列的问题。博客重点在于理解和应用线段树进行区间查询与修改，并提供了完整的代码实现。

摘要由CSDN通过智能技术生成

// Problem: 小y的序列
// Contest: NowCoder
// URL: https://ac.nowcoder.com/acm/contest/11186/C
// Memory Limit: 524288 MB
// Time Limit: 6000 ms
// 2022-02-25 11:33:22
// 
// Powered by CP Editor (https://cpeditor.org)

#include<bits/stdc++.h>
using namespace std;

#define rep(i,l,r) for(int i=(l);i<=(r);i++)
#define per(i,l,r) for(int i=(l);i>=(r);i--)
#define ll long long
#define pii pair<int, int>
#define mset(s,t) memset(s,t,sizeof(t))
#define mcpy(s,t) memcpy(s,t,sizeof(t))
#define fi first
#define se second
#define pb push_back
#define all(x) (x).begin(),(x).end()
#define SZ(x) ((int)(x).size())
#define mp make_pair

const ll mod = 1e9 + 7;

inline ll qmi (ll a, ll b) {
	ll ans = 1;
	while (b) {
		if (b & 1) ans = ans * a%mod;
		a = a * a %mod;
		b >>= 1;
	}
	return ans;
}
inline int read () {
	int x = 0, f = 0;
	char ch = getchar();
	while (!isdigit(ch)) f |= (ch=='-'),ch= getchar();
	while (isdigit(ch)) x = x * 10 + ch - '0', ch = getchar();
	return f?-x:x;
}
template<typename T> void print(T x) {
	if (x < 0) putchar('-'), x = -x;
	if (x >= 10) print(x/10);
	putchar(x % 10 + '0');
}
inline ll sub (ll a, ll b) {
	return ((a - b ) %mod + mod) %mod;
}
inline ll add (ll a, ll b) {
	return (a + b) %mod;
}
inline ll inv (ll a) {
	return qmi(a, mod - 2);
}
const int N = 4e6 + 10;

struct Node {
	int l, r;
	int minv, maxv;
	int dif;
}t[N];
int n, k;
int a[N];

inline void push_up (int p) {
	t[p].maxv = max (t[p << 1].maxv, t[(p << 1)+ 1].maxv);
	t[p].minv =  min (t[p << 1].minv, t[(p << 1) + 1].minv);
}
inline void build (int p, int l, int r) {
	t[p].l = l, t[p].r = r;
	if (l == r) {
		t[p].l = l, t[p].r = r;
		t[p].maxv = t[p].minv = a[l];
		return;
	}
	int mid = l + r>> 1;
	build(p * 2, l, mid);
	build(p * 2 + 1, mid + 1, r);
	push_up(p);
}
inline void modify (int p, int x, int v) {
	if (t[p].l <= x && t[p].r >= x) {
		t[p].maxv = t[p].minv = v;
		return;
	}
	int mid = (t[p].l + t[p].r) >> 1;
	if (x <= mid) modify (p << 1, x, v);
	else modify ((p << 1) + 1, x, v );
	push_up (p);
}
inline pii query (int p, int l, int r) {
	if (t[p].l >= l && t[p].r <= r) {
		return mp(t[p].minv, t[p].maxv);
	}
	int val1=  0x3f3f3f3f, val2 = -0x3f3f3f3f;
	int mid = (t[p].l + t[p].r) >> 1;
	if (l <= mid) {
		  pii temp = query(p << 1, l, r);
		    val1 = min (val1, temp.fi);
			val2 = max(val2, temp.se);
	}
	if (r > mid)
	{		 pii temp = query((p <<1) + 1, l, r);
		    val1 = min (val1, temp.fi);
			val2 = max(val2, temp.se);
	}
	return mp(val1, val2);
}
// bool check1 (int i, int mid) {
	// int maxv = query(1, i, mid, 0);
	// int minv = query (1, i, mid, 1);
	// return maxv - minv >= k;
// }
// bool check2 (int i, int mid) {
	// int maxv = query(1, i, mid, 0);
	// int minv = query (1, i, mid, 1);
	// return maxv - minv <= k;
// }
// bool check3(int i, int mid) {
    // int maxv = query(1, i, mid, 0);
	// int minv = query (1, i, mid, 1);
	// return maxv - minv == k;
// }
void solve() {
    scanf("%d%d", & n, & k);
	rep(i, 1, n) 
        scanf("%d", & a[i]);
	    build(1, 1, n);
	     ll res = 0;
        int l = 1, r1 = 1, r2 = 1;
        for ( ; l <= n; l ++ ) {
                pair<int, int> qry = query(1, l, r1);
                while ( r1 <= n && qry.second - qry.first < k ) r1 ++, qry = query(1, l, r1);
                qry = query(1, l, r2);
                while ( r2 <= n && qry.second - qry.first <= k ) r2 ++, qry = query(1, l, r2);
                r2 --;
                if ( r1 <= n ) res += (ll)r2 - (ll)r1 + 1;
                else break;
        }
        printf("%lld\n", res);
}
int main () {
    int t;
    t =1;
    //cin >> t;
    while (t --) solve();
    return 0;
}

这一题要看到一个关键性质。固定一个左端，最大值和最小值的插值是递增的用线段树O(Nlogn)