結果

問題 No.1031 いたずら好きなお姉ちゃん
コンテスト
ユーザー carrot46
提出日時 2020-04-18 12:32:05
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 232 ms / 3,500 ms
コード長 3,364 bytes
コンパイル時間 2,063 ms
コンパイル使用メモリ 182,588 KB
実行使用メモリ 18,048 KB
最終ジャッジ日時 2024-10-03 23:28:50
合計ジャッジ時間 11,355 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 53
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <iomanip>
using namespace std;
#define reps(i,s,n) for(int i = s; i < n; i++)
#define rep(i,n) reps(i,0,n)
#define Rreps(i,n,e) for(int i = n - 1; i >= e; --i)
#define Rrep(i,n) Rreps(i,n,0)
#define ALL(a) a.begin(), a.end()
#define fi first
#define se second
typedef long long ll;
typedef vector<ll> vec;
typedef vector<vec> mat;

ll N,M,H,W,K,Q,A,B;
string S;
const ll MOD = 998244353;
//const ll MOD = (1e+9) + 7;
const ll INF = 1LL<<60;
typedef pair<ll, ll> P;

int main() {
    class BIT{
            ll n;
      		vector<ll> bitree;
        public:
        	BIT(unsigned long _n) : bitree(_n + 1, 0){
              n = _n;
            }
        	void add(ll id, ll x){
              while(id <= n){
                bitree[id] += x;
                id += id & -id;
              }
              return;
            }
        	ll sum(ll id){
              ll temp = 0;
              while(id > 0){
                temp += bitree[id];
                id -= id & -id;
              }
              return temp;
            }
      };
    class SEGTREE{
                ll n, dflt;
          		vector<ll> dat;
            public:
            	SEGTREE(unsigned long _n, ll _a) : dat(_n * 4, _a){
            	    n = 1;
            	    dflt = _a;
            	    while(n < _n) n *= 2;
                }
                void update(ll k, ll a){
                    k += n;
                    dat[k] = a;
                    while(k > 0){
                        k /= 2;
                        dat[k] = max(dat[k * 2], dat[k * 2 + 1]);
                    }
                }
                void init(vector<ll> &v){
            	    for(int i = 0; i < v.size(); ++i) dat[i + n] = v[i];
            	    for(int i = n - 1; i >= 1; --i) {
            	        dat[i] = max(dat[i * 2], dat[i * 2 + 1]);
            	    }
            	}
                ll Q(ll a, ll b){
            	    return query(a, b, 1, 0, n);
            	}
                ll query(ll a, ll b, ll k, ll l, ll r){
                    if(r <= a || b <= l) return dflt;
                    if(a <= l && r <= b) return dat[k];
                    ll vl = query(a, b, k * 2, l, (l + r) / 2);
                    ll vr = query(a, b, k * 2 + 1, (l + r) / 2, r);
                    return max(vl, vr);
                }
                //単項にアクセスする[]だが、書き換え厳禁
                //一応後でinitすれば書き換えてもよさそうだが…
                ll & operator [](int k) {return dat[k + n];}
        };
    cin>>N;
    vec p(N + 2), p_inv(N + 1);
    p[0] = p[N + 1] = N;
    p_inv[N] = 0;
    reps(i,1,N+1) {
        cin>>p[i];
        --p[i];
        p_inv[p[i]] = i;
    }
    ll ans = 0;
    rep(rp,2){
        BIT bit(N + 2);
        SEGTREE seg(N + 2, 0);
        set<ll> st1, st2;
        st1.insert(N);
        st2.insert(-1);
        reps(i,1,N+1){
            while(*st2.rbegin() > p[i]){
                bit.add(p_inv[*st2.rbegin()], -1);
                st2.erase(*st2.rbegin());
            }
            ans += bit.sum(i - 1) - bit.sum(seg.Q(*st1.upper_bound(p[i]), N + 1));
            bit.add(i, 1);
            seg.update(p[i], i);
            st1.insert(p[i]);
            st2.insert(p[i]);
        }
        reverse(ALL(p));
        reps(i,1,N + 1) p_inv[p[i]] = i;
    }
    cout<<ans<<endl;
}
0