Submission #2161132

---

Source Code Expand

#include <iostream>
#include <cstdio>
#include <vector>
#define _USE_MATH_DEFINES
#include <math.h>
#include <cstring>
#include <numeric>
#include <algorithm>
#include <stdlib.h>
#include <functional>
#include <string>
#include <list>
#include <fstream>
#include <iomanip>
#include <array>
#include <map>
#include <queue>
#include <limits.h>
#include <set>
#include <stack>
#include <random>
#include <complex>
#include <unordered_map>
#include <unordered_set>
#include <chrono>
#define rep(i,s,n) for(int i = (s); (n) > i; i++)
#define REP(i,n) rep(i,0,n)
#define RANGE(x,a,b) ((a) <= (x) && (x) <= (b))
#define DUPLE(a,b,c,d) (RANGE(a,c,d) || RANGE(b,c,d) || RANGE(c,a,b) || RANGE(d,a,b))
#define INCLU(a,b,c,d) (RANGE(a,c,d) && (b,c,d))
#define PW(x) ((x)*(x))
#define ALL(x) (x).begin(), (x).end()
#define MODU 1000000007
#define bitcheck(a,b)   ((a >> b) & 1)
#define bitset(a,b)      ( a |= (1 << b))
#define bitunset(a,b)    (a &= ~(1 << b))
#define MP(a,b) make_pair((a),(b))
#define Manh(a,b) (abs((a).first-(b).first) + abs((a).second - ((b).second))
#define pritnf printf
#define scnaf scanf
#define itn int

#include <nmmintrin.h>
#ifdef _MSC_VER

#define __builtin_popcount _mm_popcnt_u32
#define __builtin_popcountll _mm_popcnt_u64
#endif
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
template<typename A, size_t N, typename T>
void Fill(A(&array)[N], const T &val) {
	std::fill((T*)array, (T*)(array + N), val);
}

struct Edge {
	int from, to;
};

template<class T, class OP, class PROP>
class Lazy_Segment_Tree {//0-indexed 半開
public:
	vector<T> data, lazy;
	vector<bool> isupd;
	int n;
	Lazy_Segment_Tree(int a) :n(1) {
		while (n < a) n *= 2;
		data.resize(n * 2 - 1);
		lazy.resize(n * 2 - 1);
		isupd.resize(n * 2 - 1);
	}

	void prop(int num, int s) {
		if (isupd[num]) {
			data[num] = PROP()(data[num], lazy[num], 1, s);
			if (num < n - 1) {
				lazy[num * 2 + 1] = PROP()(lazy[num * 2 + 1], lazy[num]);
				lazy[num * 2 + 2] = PROP()(lazy[num * 2 + 2], lazy[num]);
				isupd[num * 2 + 1] = 1;
				isupd[num * 2 + 2] = 1;
			}
			lazy[num] = T();
			isupd[num] = 0;
		}
	}
	void op(int a, int b, T v, int num = 0, int base = 1) {
		if (a >= b) return;
		int l = (num + 1 - base) * (n / base), r = l + n / base;
		prop(num, r - l);
		if (a == l && b == r) {
			isupd[num] = 1;
			lazy[num] = PROP()(lazy[num], v);
			prop(num, r - l);
		}
		else {
			int nr = (l + r) / 2;
			if (nr > a) op(a, min(b, nr), v, num * 2 + 1, base * 2);
			if (nr < b) op(max(a, nr), b, v, num * 2 + 2, base * 2);
			prop(num * 2 + 1, (r - l) / 2), prop(num * 2 + 2, (r - l) / 2);
			data[num] = OP()(data[num * 2 + 1], data[num * 2 + 2]);
		}
	}

	T query(int a, int b, int num = 0, int base = 1) {
		if (a >= b) return 0;
		int l = (num + 1 - base) * (n / base), r = l + n / base;

		prop(num, r - l);

		if (a == l && b == r)
			return data[num];
		else {
			int nr = (l + r) / 2;
			T ret = T();
			if (nr > a) ret = OP()(ret, query(a, min(b, nr), num * 2 + 1, base * 2));
			if (nr < b) ret = OP()(ret, query(max(a, nr), b, num * 2 + 2, base * 2));
			return ret;
		}
	}
};


template<class T>
class ADD {
public:
	T operator ()(const T a, const T b, const int ra = 1, int rb = 1) {
		return a*ra + b*rb;
	}
};

template<class T>
class SUM {
public:
	T operator ()(const T a, const T b) {
		return a + b;
	}
};

struct N {
public:
	ll num;
	N(ll num) : num(num%MODU) {}
	N() :num(0) {}
	N operator + (const N obj) const {
		return N(obj.num + num);
	}
	N operator * (const N obj) const {
		return N(obj.num * num);
	}
	bool operator < (const N obj)const {
		return obj.num > num;
	}
};


int main() {

	int n;
	ll a, b;

	cin >> n >> a >> b;
	if (a < b) swap(a, b);
	vector<ll> num(n);
	REP(i, n) {
		scanf("%lld", &num[i]);
	}

	vector<int> cb(n, 1),ca(n,1);
	REP(i, n - 1) {
		if (num[i] + b <= num[i + 1]) {
			cb[i + 1] = cb[i] + 1;
		}
	}

	for (int i = n - 1; 0 < i; i--) {
		if (num[i-1] + a <= num[i]) {
			ca[i - 1] = ca[i] + 1;
		}
	}

	Lazy_Segment_Tree<N, SUM<N>, ADD<N>> seg(n + 1);

	
	seg.op(0, ca[0] + 1,1);

	rep(i, 1, n-1) {
		if (num[i + 1] - num[i - 1] < b) {
			continue;
		}
		int l = i-cb[i-1],r = upper_bound(num.begin(),num.begin() + i, num[i]-a) - num.begin();
		ll sum = seg.query(l,r + 1).num;

		seg.op(i + 1, i + 1 + ca[i], sum);
	}

	int l = n-1 - cb[n - 2], r = upper_bound(num.begin(), num.begin() + n-1, num[n-1]-a) - num.begin();
	ll sum = seg.query(l, r + 1).num;
	sum += seg.query(n-cb[n - 1],n).num;
	sum += seg.query(n,n+1).num;
	sum %= MODU;
	cout << sum << endl;
	return 0;
}

Submission Info

Compile Error

./Main.cpp: In function ‘int main()’:
./Main.cpp:167:25: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
   scanf("%lld", &num[i]);
                         ^

Judge Result