結果

問題 No.1031 いたずら好きなお姉ちゃん
ユーザー 沙耶花
提出日時 2020-04-17 23:26:01
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 357 ms / 3,500 ms
コード長 2,616 bytes
コンパイル時間 2,366 ms
コンパイル使用メモリ 202,416 KB
最終ジャッジ日時 2025-01-09 20:47:39
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 53
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define modulo 1000000007
#define mod(mod_x) ((((long long)mod_x)+modulo)%modulo)
#define Inf 100000000



template <typename T,typename F>
struct segtree{
	//元データx[i]はv[n+i]
	//v[i]の親はv[i/2],子はv[i*2]とv[i*2+1]
	F func;
	vector<T> v;
	int n;
	
	const T init_value = Inf;
	
	segtree(int sz,F f):func(f){
		n=1;
		while(true){
			if(n>=sz)break;
			n*=2;
		}
		v.resize(2*n,init_value);

		for(int i=n-1;i>=0;i--){
			v[i]=func(v[i<<1],v[(i<<1)+1]);
		}
	}
	
	segtree(vector<T> &x,F f):func(f){
		n=1;
		while(true){
			if(n>=x.size())break;
			n*=2;
		}
		v.resize(2*n,init_value);
		
		for(int i=0;i<x.size();i++){
			v[n+i]=x[i];
		}
		for(int i=n-1;i>=0;i--){
			v[i]=func(v[i<<1],v[(i<<1)+1]);
		}
	}

	void update(int x,T val){
		x+=n;
		v[x]=val;
		while(x>0){
			x>>=1;
			v[x]=func(v[x<<1],v[(x<<1)+1]);
		}
	}
	
	//区間[l,r)におけるクエリ処理
	T query(int l,int r){
		if(l>=r)return init_value;
		l+=n;
		r+=n;
		T res1 = init_value;
		T res2 = init_value;
		while(true){
			if(l&1){
				res1=func(res1,v[l++]);
			}
			if(r&1){
				res2=func(v[--r],res2);
			}
			if(l>=r)break;
			l>>=1;r>>=1;
		}
		return func(res1,res2);
	}
	
	void show(){
		int n = 1;
		for(int i=1;i<v.size();i++){
			for(int j=0;j<n;j++){
				if(j!=0)cout<<' ';
				cout<<v[i+j];
			}
			cout<<endl;
			i+=n-1;
			n*=2;
		}
	}
	
};

long long get(vector<int> P){
	int N = P.size();
	
	vector<int> V(N);
	for(int i=0;i<N;i++)V[P[i]] = i;
	
	auto f = [](int a,int b){
		return min(a,b);
	};
	
	segtree<int,decltype(f)> seg(N,f);
	
	vector<vector<int>> next(20,vector<int>(N,Inf));
	
	for(int i=N-1;i>=0;i--){
		int ind = V[i];
		next[0][V[i]] = seg.query(ind,N);
		seg.update(ind,ind);
	}
	
	for(int i=1;i<20;i++){
		for(int j=0;j<N;j++){
			if(next[i-1][j]!=Inf){
				next[i][j] = next[i-1][next[i-1][j]];
			}
		}
	}
	//cout<<next[0][1]<<endl;
	for(int i=0;i<N;i++)seg.update(V[i],i);
	//seg.show();
	long long ans = 0;
	for(int i=0;i<N;i++){
		int ok = V[i]+1;
		int ng = N+1;
		while(ng-ok>1){
			int mid = (ok+ng)/2;
			if(seg.query(V[i],mid)==i)ok = mid;
			else ng = mid;
			//cout<<ok<<','<<ng<<endl;
		}
		long long add = 0;
		int pos = V[i];
		for(int j=19;j>=0;j--){
			if(next[j][pos] < ok){
				pos = next[j][pos];
				add |= (1<<j);
			}
			else{
			}
		}
		//cout<<add<<endl;
		ans += add;
	}
	return ans;
}


int main() {
	
	int N;
	cin>>N;
	
	vector<int> P(N);
	for(int i=0;i<N;i++){
		cin>>P[i];
		P[i]--;
	}
	
	long long ans = get(P);
	//cout<<ans<<endl;
	reverse(P.begin(),P.end());
	ans += get(P);
	
	cout<<ans<<endl;
	
    return 0;
}
0