牛客練習賽66 E、騷區間

思路:
明顯枚舉每個端點作爲騷區間的左端點,假設我們枚舉ai作爲左端點,然後在他右邊找到第一個小於ai的數的位置x,然後再在x的右邊找第一個小於ai的數的位置y,明顯以ai爲左端點的騷區間只能在[x,y)這個區間中選擇右端點,同理也可以枚舉每個數作爲右端點,然後找他左邊第一個第二個大於ai的數x,y,那麼以這個數爲右端點的騷區間的左端點只能在(x,y]中選擇,這樣的話問題就化成了枚舉每個點作爲左端點,然後得到一個區間,在log的時間複雜度的下,求這個區間中的每個數對應的區間中包含左端點的區間個數
最後令點 i 作爲右端點,找到左端點的可行區間 [ L , R ] ,在線段樹上統計區間 [ L , R ] 內有多少個數,就說明有多少個數可以在點 i 作爲右端點時,作爲左端點與其匹配。

//p[i][j]表示p[i][j]爲左端點可行右端點的區間的端點,如果爲正,表示剛加入進來,爲負就消除。
for(auto it:p[i]){
	if(it>0)update(it,1);
	else update(-it,-1);
}

對於找右邊第一個小於ai的數的位置,可以用set維護或者線段樹維護,用set就把數字從小到大依次加入,然後對pos進行查找。

線段樹寫法

#pragma GCC optimize(2)
#include<bits/stdc++.h>
using namespace std;
const int man = 1e6+10;
#define IOS ios::sync_with_stdio(0)
#define ull unsigned ll
#define uint unsigned
#define pai pair<int,int>
#define pal pair<ll,ll>
#define IT iterator
#define pb push_back
#define fi first
#define se second
#define For(i,j,k) for (int i=(int)(j);i<=(int)(k);++i)
#define Rep(i,j,k) for (int i=(int)(j);i>=(int)(k);--i)
#define endl '\n'
#define ll long long
const ll mod = 1e9+7;
int a[man];
int c[man],n;
#define lowbit(x) x&(-x)
void update(int pos,int va){
    for(int i = pos;i <= n;i += lowbit(i)){
        c[i] += va;
    }
}
 
int query(int pos){
    int ans = 1;
    for(int i = pos;i >= 1;i -= lowbit(i)){
        ans += c[i];
    }
    return ans;
}
 
int minn[man<<2],maxx[man<<2];
 
void build(int l,int r,int rt){
    minn[rt] = 1e9;
    maxx[rt] = 0;
    if(l==r){
        minn[rt] = maxx[rt] = a[l];
        return;
    }
    int m = l +r >>1;
    build(l,m,rt<<1);
    build(m+1,r,rt<<1|1);
    maxx[rt] = max(maxx[rt<<1],maxx[rt<<1|1]);
    minn[rt] = min(minn[rt<<1],minn[rt<<1|1]);
}
 
void query_min(int l,int r,int L,int R,int rt,int va,int &ans){
    if(r<L||l>R||r<l)return;
    if(minn[rt]>=va)return;
    if(l==r){
        ans = l;
        return;
    }
    int m = l + r >>1;
    if(minn[rt<<1]<va)query_min(l,m,L,R,rt<<1,va,ans);
    if(ans==n+1)query_min(m+1,r,L,R,rt<<1|1,va,ans);
}
 
void query_max(int l,int r,int L,int R,int rt,int va,int &ans){
    if(r<L||l>R||r<l)return;
    if(maxx[rt]<=va)return;
    if(l==r){
        ans = l;
        return;
    }
    int m = l + r >>1;
    if(maxx[rt<<1|1]>va)query_max(m+1,r,L,R,rt<<1|1,va,ans);
    if(ans==0)query_max(l,m,L,R,rt<<1,va,ans);
}
int vis[man];
vector<int>p[man];
 
signed main() {
    #ifndef ONLINE_JUDGE
        //freopen("in.txt", "r", stdin);
        //freopen("out.txt","w",stdout);
    #endif
    scanf("%d",&n);
    for(int i = 1;i <= n;i++){
        scanf("%d",a+i);
    }
    build(1,n,1);
    ll ans = 0;
    for(int i = 1;i <= n;i++){
        //if(i>1)continue;
        int l = n+1,r = n+1;
        query_min(1,n,i+1,n,1,a[i],l);
        query_min(1,n,l+1,n,1,a[i],r);
        if(l!=n+1)p[l].emplace_back(i);
        if(r!=n+1)p[r].emplace_back(-i);
        // cout << l <<" " <<r <<endl;
        for(auto it:p[i]){
            if(it>0)update(it,1);
            else{
                it *= -1;
                update(it,-1);
            }
        }
        l = r = 0;
        query_max(1,n,1,i-1,1,a[i],r);
        query_max(1,n,1,r-1,1,a[i],l);
        ans += query(r) - query(l);
    }
    printf("%lld\n",ans);
    return 0;
}

set寫法

#pragma GCC optimize(2)
#include<bits/stdc++.h>
using namespace std;
const int man = 1e6+10;
#define IOS ios::sync_with_stdio(0)
#define ull unsigned ll
#define uint unsigned
#define pai pair<int,int>
#define pal pair<ll,ll>
#define IT iterator
#define pb push_back
#define fi first
#define se second
#define For(i,j,k) for (int i=(int)(j);i<=(int)(k);++i)
#define Rep(i,j,k) for (int i=(int)(j);i>=(int)(k);--i)
#define endl '\n'
#define ll long long
const ll mod = 1e9+7;
int a[man];
int c[man],n;
#define lowbit(x) x&(-x)
void update(int pos,int va){
    for(int i = pos;i <= n;i += lowbit(i)){
        c[i] += va;
    }
}
 
int query(int pos){
    int ans = 1;
    for(int i = pos;i >= 1;i -= lowbit(i)){
        ans += c[i];
    }
    return ans;
}
int pos[man],l[man],r[man];
vector<int>p[man];
 
signed main() {
    #ifndef ONLINE_JUDGE
        //freopen("in.txt", "r", stdin);
        //freopen("out.txt","w",stdout);
    #endif
    scanf("%d",&n);
    for(int i = 1;i <= n;i++){
        scanf("%d",a+i);
        pos[a[i]] = i;
    }
    ll ans = 0;
    set<int>s;
    s.clear();
    s.insert(n+1);
    s.insert(n+2);
    for(int i = 1;i <= n;i++){
        auto it = s.upper_bound(pos[i]);
        p[*it].emplace_back(pos[i]);
        it++;
        p[*it].emplace_back(-pos[i]);
        s.insert(pos[i]);
    }
    s.clear();s.insert(0),s.insert(-1),s.insert(n+1);
    for(int i = n;i >= 1;i--){
        auto it = s.upper_bound(pos[i]);
        it--;
        r[pos[i]] = *it;
        it--;
        l[pos[i]] = *it;
        s.insert(pos[i]);
    }
    for(int i = 1;i <= n;i++){
        for(auto it:p[i]){
            if(it>0)update(it,1);
            else update(-it,-1);
        }
        //cout << l[i] << " " <<r[i] << endl;
        ans += query(r[i]) - query(l[i]);
    }
    printf("%lld\n",ans);
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章