「Atcoder247」 E - Max Min「二分」

E - Max Min

题目描述:

给定n个数a[i],问存在多少对(l, r)满足max{a[i]} == x && min{a[i]} == y

思路 :

我们枚举每个起点i,去找满足条件的区间的左端点l和右端点 r即可

确定左端点,则是找从i开始的第一个xy出现的位置的最大值

确定右端点,则是找从i开始的第一个>x || 的位置

用二分就行

#include 
using namespace std;

#define endl '\n'
#define inf 0x3f3f3f3f
#define mod7 1000000007
#define mod9 998244353
#define m_p(a,b) make_pair(a, b)
#define mem(a,b) memset((a),(b),sizeof(a))
#define io ios::sync_with_stdio(false); cin.tie(0); cout.tie(0)
#define debug(a) cout << "Debuging...|" << #a << ": " << a << "\n";
typedef long long ll;
typedef pair <int,int> pii;

#define MAX 300000 + 50
int n, m, k, x, y;
int tr[MAX];
vector<int>v, xr, yr;
ll l, r, lx, ly;
void work(){
    cin >> n >> x >> y;
    for(int i = 1; i <= n; ++i){
        cin >> tr[i];
        if(tr[i] == x)xr.push_back(i);
        if(tr[i] == y)yr.push_back(i);
        if(tr[i] > x || tr[i] < y)v.push_back(i);
    }
    v.push_back(n + 1);
    xr.push_back(inf);yr.push_back(inf);
    ll ans = 0;
    for(int i = 1; i <= n; ++i){
        r = *lower_bound(v.begin(), v.end(), i);
        lx = *lower_bound(xr.begin(), xr.end(), i);
        ly = *lower_bound(yr.begin(), yr.end(), i);
        l = max(lx, ly);
        ans += max(0ll, r - l);
    }
    cout << ans << endl;
}


int main(){
    io;
    work();
    return 0;
}

你可能感兴趣的:(Atcoder,二分)