1905D - Cyclic MEX - CodeForces Solution


binary search constructive algorithms data structures implementation math two pointers

Please click on ads to support us..

C++ Code:

#include <bits/stdc++.h>
using namespace std;
#define gc getchar_unlocked
#define rep(i,a,b) for (int i = a; i < b; i++)
#define each(i, k, n) for (int i = k;i<=n;i++)
#define ll long long
#define pb push_back
#define mp make_pair
#define endl '\n'
#define f first
// #define s second
ll min(ll a,int b) { if (a<b) return a; return b; }
ll min(int a,ll b) { if (a<b) return a; return b; }
ll max(ll a,int b) { if (a>b) return a; return b; }
ll max(int a,ll b) { if (a>b) return a; return b; }
ll gcd(ll a,ll b) { if (b==0) return a; return gcd(b, a%b); }
ll lcm(ll a,ll b) { return a/gcd(a,b)*b; }
string to_upper(string a) { for (int i=0;i<(int)a.size();++i) if (a[i]>='a' && a[i]<='z') a[i]-='a'-'A'; return a; }
string to_lower(string a) { for (int i=0;i<(int)a.size();++i) if (a[i]>='A' && a[i]<='Z') a[i]+='a'-'A'; return a; }
bool prime(ll a) { if (a==1) return 0; for (int i=2;i<=round(sqrt(a));++i) if (a%i==0) return 0; return 1; }
typedef long int int32;
typedef unsigned long int uint32;
typedef long long int int64;
typedef unsigned long long int  uint64;
#define all(x) x.begin(), x.end()
#define sortall(x) sort(all(x))
#define tr(it, a) for (auto it = a.begin(); it != a.end(); it++)
#define PI 3.1415926535897932384626
#define mod 1000000007
typedef pair<int, int> pii;
typedef pair<ll, ll> pl;
typedef vector<int> vi;
typedef vector<ll> vl;
typedef vector<pii> vpii;
typedef vector<pl> vpl;
typedef vector<vi> vvi;
typedef vector<vl> vvl;
#define N 1e5
ll mex(vl &a,ll i){
    set <ll> s(a.begin(),a.begin()+i+1);
    ll ans = 0;
    while(s.count(ans)){
        ans++;
    }
    return ans;
}
void solve(){
    ll n;
    cin >> n;
    vl v(n);
    rep(i,0,n){
        cin >> v[i];
    }
    vl a;
    ll i = 0;
    while(i<n && v[i]!=0){
        i++;
    }
    ll j = i;
    rep(k,0,n){
        a.pb(v[j%n]);
        j++;
    }
    // rep(i,0,n){
    //     cout << a[i] << " ";
    // }
    // cout << endl;
    vl less(n);
    stack <ll> s;
    vl dp(n,n);
    s.push(0);
    rep(i,1,n){
        while(s.size() && a[i] < a[s.top()]){
            s.pop();
        }
        less[i] = s.top();
        s.push(i);
    }
    // rep(i,1,n){
    //     cout << less[i] << " ";
    // }
    // cout << endl;
    // rep(i,0,s.size()){
    //     cout<<s.top() << " ";
    //     s.pop();
    // }
    rep(i,1,n){
        dp[i] = dp[less[i]]  + (i-less[i])*a[i];
    }
    ll ans = n;
    each(i,1,n-1){
        ans = max(ans,dp[i]);
    }
    cout << ans;
}
int main()
{
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    int t;
    cin >> t;
    while (t--)
    {
        solve();
        cout<<endl;
    }
    return 0;
}


Comments

Submit
0 Comments
More Questions

1302. Deepest Leaves Sum
1209. Remove All Adjacent Duplicates in String II
994. Rotting Oranges
983. Minimum Cost For Tickets
973. K Closest Points to Origin
969. Pancake Sorting
967. Numbers With Same Consecutive Differences
957. Prison Cells After N Days
946. Validate Stack Sequences
921. Minimum Add to Make Parentheses Valid
881. Boats to Save People
497. Random Point in Non-overlapping Rectangles
528. Random Pick with Weight
470. Implement Rand10() Using Rand7()
866. Prime Palindrome
1516A - Tit for Tat
622. Design Circular Queue
814. Binary Tree Pruning
791. Custom Sort String
787. Cheapest Flights Within K Stops
779. K-th Symbol in Grammar
701. Insert into a Binary Search Tree
429. N-ary Tree Level Order Traversal
739. Daily Temperatures
647. Palindromic Substrings
583. Delete Operation for Two Strings
518. Coin Change 2
516. Longest Palindromic Subsequence
468. Validate IP Address
450. Delete Node in a BST