結果

問題 No.2779 Don't make Pair
コンテスト
ユーザー FplusFplusF
提出日時 2024-06-07 21:43:15
言語 C++23
(gcc 13.3.0 + boost 1.89.0)
結果
AC  
実行時間 72 ms / 2,000 ms
コード長 2,376 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 3,907 ms
コンパイル使用メモリ 255,264 KB
実行使用メモリ 8,688 KB
最終ジャッジ日時 2024-12-26 07:32:22
合計ジャッジ時間 5,972 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>
using namespace std;
using ll=long long;
using ull=unsigned long long;
using pll=pair<ll,ll>;
using tll=tuple<ll,ll,ll>;
using ld=long double;
const ll INF=(1ll<<60);
#define rep(i,n) for (ll i=0;i<(ll)(n);i++)
#define replr(i,l,r) for (ll i=(ll)(l);i<(ll)(r);i++)
#define all(v) v.begin(),v.end()
#define len(v) ((ll)v.size())
template<class T> inline bool chmin(T &a,T b){
    if(a>b){
        a=b;
        return true;
    }
    return false;
}
template<class T> inline bool chmax(T &a,T b){
    if(a<b){
        a=b;
        return true;
    }
    return false;
}
template <class T,auto op,T e> struct segment_tree{
    int n;
    vector<T> v;
    segment_tree(int n_):n(n_),v(n*2,e){}
    segment_tree(vector<T> a):n((int)a.size()),v(n*2,e){
        for(int i=0;i<n;i++) v[i+n]=a[i];
        for(int i=n-1;1<=i;i--) v[i]=op(v[i<<1],v[i<<1|1]);
    }
    void update(int i,T x){
        assert(0<=i&&i<=n);
        i+=n;
        v[i]=x;
        while(1<i){
            i>>=1;
            v[i]=op(v[i<<1],v[i<<1|1]);
        }
    }
    T get(int i){
        i+=n;
        return v[i];
    }
    T fold(int l,int r){  //[l,r)
        assert(0<=l&&l<=n);
        assert(0<=r&&r<=n);
        assert(l<=r);
        T ret_l=e,ret_r=e;
        l+=n;
        r+=n;
        for(;l<r;l>>=1,r>>=1){
            if(l&1) ret_l=op(ret_l,v[l++]);
            if(r&1) ret_r=op(v[--r],ret_r);
        }
        return op(ret_l,ret_r);
    }
};
template<class T> vector<T> compress(vector<T> &x){
    vector<T> unique_v=x;
    int n=x.size();
    sort(all(unique_v));
    unique_v.erase(unique(all(unique_v)),unique_v.end());
    for(int i=0;i<n;i++){
        x[i]=lower_bound(all(unique_v),x[i])-unique_v.begin();
    }
    return unique_v;
}
int main(){
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    ll n;
    cin >> n;
    vector<ll> a(n);
    rep(i,n) cin >> a[i];
    compress(a);
    segment_tree<ll,[&](ll i,ll j){return max(i,j);},0> seg(n),seg2(n);
    rep(i,n){
        ll x=seg.get(a[i]);
        seg.update(a[i],x+1);
    }
    vector<ll> ans;
    rep(i,n-1){
        ll x=seg.get(a[i]);
        seg.update(a[i],x-1);
        ll y=seg2.get(a[i]);
        seg2.update(a[i],y+1);
        if(seg.fold(0,n)<=1&&seg2.fold(0,n)<=1) ans.push_back(i);
    }
    cout << len(ans) << '\n';
    for(auto i:ans) cout << i+1 << ' ';
    cout << '\n';
}
0