結果

問題 No.1371 交換門松列・松
ユーザー leaf_1415leaf_1415
提出日時 2021-01-29 22:40:42
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 158 ms / 4,000 ms
コード長 2,899 bytes
コンパイル時間 798 ms
コンパイル使用メモリ 88,836 KB
実行使用メモリ 25,088 KB
最終ジャッジ日時 2024-06-27 09:01:40
合計ジャッジ時間 3,811 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 7 ms
14,048 KB
testcase_01 AC 8 ms
14,208 KB
testcase_02 AC 7 ms
14,048 KB
testcase_03 AC 8 ms
14,080 KB
testcase_04 AC 7 ms
14,208 KB
testcase_05 AC 7 ms
14,052 KB
testcase_06 AC 7 ms
14,048 KB
testcase_07 AC 7 ms
14,044 KB
testcase_08 AC 8 ms
14,080 KB
testcase_09 AC 8 ms
14,208 KB
testcase_10 AC 7 ms
14,052 KB
testcase_11 AC 8 ms
14,208 KB
testcase_12 AC 7 ms
14,044 KB
testcase_13 AC 7 ms
14,080 KB
testcase_14 AC 7 ms
14,080 KB
testcase_15 AC 7 ms
14,080 KB
testcase_16 AC 7 ms
14,080 KB
testcase_17 AC 7 ms
14,080 KB
testcase_18 AC 66 ms
25,048 KB
testcase_19 AC 65 ms
25,020 KB
testcase_20 AC 65 ms
24,920 KB
testcase_21 AC 67 ms
24,932 KB
testcase_22 AC 66 ms
25,088 KB
testcase_23 AC 147 ms
25,088 KB
testcase_24 AC 157 ms
24,960 KB
testcase_25 AC 157 ms
25,064 KB
testcase_26 AC 149 ms
25,088 KB
testcase_27 AC 145 ms
24,960 KB
testcase_28 AC 151 ms
24,932 KB
testcase_29 AC 151 ms
24,936 KB
testcase_30 AC 153 ms
24,960 KB
testcase_31 AC 158 ms
24,948 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <cstdio>
#include <cmath>
#include <ctime>
#include <cstdlib>
#include <cassert>
#include <vector>
#include <list>
#include <stack>
#include <queue>
#include <deque>
#include <map>
#include <set>
#include <bitset>
#include <string>
#include <algorithm>
#include <utility>
#include <complex>
#define rep(x, s, t) for(llint (x) = (s); (x) <= (t); (x)++)
#define chmin(x, y) (x) = min((x), (y))
#define chmax(x, y) (x) = max((x), (y))
#define all(x) (x).begin(),(x).end()
#define inf 1e18

using namespace std;

typedef long long llint;
typedef long long ll;
typedef pair<llint, llint> P;

struct BIT{
	int size;
	vector<llint> bit;
	BIT(){size = 0;}
	BIT(int s){
		size = s;
		bit.resize(size+1);
		init();
	}
	void init(){
		for(int i = 1; i <= size; i++) bit[i] = 0;
	}
	llint query(int i){
		llint ret = 0;
		while(i > 0){
			ret += bit[i];
			i -= i&(-i);
		}
		return ret;
	}
	void add(int i, llint x){
		while(i <= size){
			bit[i] += x;
			i += i&(-i);
		}
	}
};


ll n;
ll a[200005];
vector<ll> avec[200005], qvec[200005];
BIT bit(200005);

bool check(ll a, ll b, ll c)
{
	if(a == b || b == c || c == a) return false;
	if(a > b && b < c) return true;
	if(a < b && b > c) return true;
	return false;
}
bool calc(ll x)
{
	if(x <= 1 || x >= n) return true;
	return check(a[x-1], a[x], a[x+1]);
}

int main(void)
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	
	cin >> n;
	rep(i, 1, n) cin >> a[i];
	
	ll ans = 0;
	rep(i, 2, n-1){
		ll l, h;
		if(a[i-1] < a[i]){
			l = max(a[i-1], a[i+1]), h = a[i];
			avec[h].push_back(l);
			qvec[l].push_back(h);
		}
	}
	for(int i = n; i >= 1; i--){
		for(auto q : qvec[i]) ans += bit.query(q-1);//, cout << i << " " << q << endl;
		for(auto q : avec[i]) bit.add(q, 1);
		
		//rep(j, 0, n) cout << bit.query(j) << " ";  cout << endl;
	}
	
	//cout << ans << endl;
	
	bit.init();
	rep(i, 1, n) avec[i].clear(), qvec[i].clear();
	rep(i, 2, n-1){
		ll l, h;
		if(a[i-1] > a[i]){
			h = min(a[i-1], a[i+1]), l = a[i];
			avec[h].push_back(l);
			qvec[l].push_back(h);
		}
	}
	for(int i = n; i >= 1; i--){
		for(auto q : qvec[i]) ans += bit.query(q-1);
		for(auto q : avec[i]) bit.add(q, 1);
	}
	
	ans -= n-2;
	ans /= 2;
	
	
	bit.init();
	rep(i, 1, n) avec[i].clear(), qvec[i].clear();
	rep(i, 2, n-1){
		ll l, h;
		if(a[i-1] < a[i]){
			l = max(a[i-1], a[i+1]), h = a[i];
			qvec[h].push_back(l);
		}else{
			h = min(a[i-1], a[i+1]), l = a[i];
			avec[h].push_back(l);
		}
	}
	for(int i = n; i >= 1; i--){
		for(auto q : qvec[i]) ans += bit.query(n) - bit.query(q);
		for(auto q : avec[i]) bit.add(q, 1);
	}
	
	
	rep(i, 2, n){
		swap(a[1], a[i]);
		bool flag = true;
		rep(j, -1, 1) flag &= calc(1+j) && calc(i+j);
		if(flag) ans++;
		swap(a[1], a[i]);
	}
	rep(i, 2, n-1){
		swap(a[n], a[i]);
		bool flag = true;
		rep(j, -1, 1) flag &= calc(n+j) && calc(i+j);
		if(flag) ans++;
		swap(a[n], a[i]);
	}
	
	cout << ans << endl;
	
	return 0;
}
0