题意:
给定一个数组,求满足最大值为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;
}