AtCoder Beginner Contest 247 E – Max Min // 容斥原理

原题链接：E – Max Min (atcoder.jp)

题意：

给定一个数组，求满足最大值为X且最小值为Y的区间个数。

思路：容斥原理

因为必须要包含端点，直接求是不容易的。因此考虑去求不一定包含端点的区间数量，再做容斥。

代码参考：

//Jakon：容斥原理
#include <bits/stdc++.h>
#define int long long
using namespace std;

const int N = 200010;

int n, x, y, a[N], ans;

int cal(int l, int r)
{
    int res = 0, s = 1, t = 1;
    for(int i = 1; i <= n; i++) {
        if(a[i] >= l && a[i] <= r) s = t = i;
        else continue;
        while(a[t] >= l && a[t] <= r && t <= n) ++ t;
        res += (t - s) * (t - s + 1) / 2;
        i = t - 1;
    }
    return res;
}

signed main()
{
    cin >> n >> x >> y;
    for(int i = 1; i <= n; i++) cin >> a[i];
    ans = cal(y, x) - cal(y + 1, x) - cal(y, x - 1) + cal(y + 1, x - 1);
    cout << ans << endl;

    return 0;
}