insert 1, insert 2, insert 3...insert k

C++实现区间内合法1的数量

原创于 2023-11-20 01:07:59 发布 · 88 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#数据结构

2023牛客多校训练营专栏收录该内容

26 篇文章

订阅专栏

这篇文章介绍了一种使用C++编程语言解决的问题，计算给定数组中每个数字右侧存在至少一个1的子区间数量。使用vector代替栈来避免内存限制，并通过动态维护合法1的位置求解。

题目

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e6 + 5;
//stack<int> p[maxn], st;
//用stack会MLE，vector不会
vector<int> p[maxn], st;//p[i]表示i这个数字左边最近的合法的1的位置（即这个1和数字i之间有2,3,...,i-1)
                        //st维护当前合法的1的位置    
int main(){
    ios::sync_with_stdio(0);
    cin.tie(0);
    int n, i, j;
    long long res = 0;
    cin >> n;
    for(i = 1; i <= n; i++){//枚举区间右端点i，[1, i]有多少个合法的1，就有多少个合法的以i为            
                            //右端点的区间
        int x;
        cin >> x;
        if(x == 1){
            p[1].push_back(i);
            st.push_back(i);
            res += st.size();
        }
        else if(p[x - 1].empty()){
            while(!st.empty()) st.pop_back();
        }
        else{
            int pos = p[x - 1].back();
            p[x - 1].pop_back();
            p[x].push_back(pos);
            while(!st.empty() && st.back() > pos) st.pop_back();
            res += st.size();
        }
    }
    cout << res;
    return 0;
}