結果

問題 No.1371 交換門松列・松
ユーザー 夕叢霧香(ゆうむらきりか)
提出日時 2020-11-03 02:25:47
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,867 bytes
コンパイル時間 1,047 ms
コンパイル使用メモリ 86,972 KB
最終ジャッジ日時 2025-01-15 19:17:01
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 15 TLE * 14
権限があれば一括ダウンロードができます

ソースコード

diff # 

#pragma GCC optimize ("O3")
#include<algorithm>
#include<cassert>
#include<iostream>
#include<vector>
using namespace std;
typedef long long lint;
typedef vector<int>vi;
typedef pair<int,int>pii;
#define rep(i,n)for(int i=0;i<(int)(n);++i)

bool kado(lint a,lint b,lint c){
  return (a-b)*(c-b)>0;
}

const int N=300000;
pii cand[N];

int main(){
  int n;
  cin>>n;
  vi a(n);
  rep(i,n)cin>>a[i],a[i]--;
  lint ans=0;
  //近い要素
  rep(i,n){
    for(int j=i+1;j<=min(n-1,i+2);j++){
      swap(a[i],a[j]);
      int ok=1;
      for(int k=max(0,i-2);k<=min(n-3,j);k++){
        ok&=kado(a[k],a[k+1],a[k+2]);
      }
      if(ok)ans++;
      swap(a[i],a[j]);
    }
  }
  //遠い要素
  //cand[i] を計算する
  rep(i,n){
    int ma=n-1,mi=0;
    int x=a[i];
    if(i<n-2){
      if(a[i+1]<a[i+2])mi=max(mi,a[i+1]+1);
      else ma=min(ma,a[i+1]-1);
    }
    if(i>=2){
      if(a[i-1]<a[i-2])mi=max(mi,a[i-1]+1);
      else ma=min(ma,a[i-1]-1);
    }
    vector<pii>c;
    if(i>0&&i<n-1){
      int l=min(a[i-1],a[i+1])-1,r=max(a[i-1],a[i+1])+1;
      if(mi<=l)c.push_back(pii(mi,l));
      if(r<=ma)c.push_back(pii(r,ma));
    }else{
      c.push_back(pii(mi,ma));
    }
    //実は、区間はちょうど 1 個。
    assert(c.size()==1);
    cand[x]=c[0];
  }
  //TLE: BIT を使わずに O(N^2) で計算する。
  rep(i,n){
    int l=cand[i].first,r=cand[i].second;
    r=min(r,i-1);
    for(int j=l;j<=r;j++){
      int lj=cand[j].first,rj=cand[j].second;
      if(lj<=i&&i<=rj)ans++;
    }
  }
  //近い要素の影響を消す。
  rep(i,n){
    int l=cand[a[i]].first,r=cand[a[i]].second;
    for(int j=max(0,i-2);j<=min(n-1,i+2);j++){
      if(a[i]<=a[j])continue;
      if(l<=a[j]&&a[j]<=r){
        int lj=cand[a[j]].first,rj=cand[a[j]].second;
        if(lj<=a[i]&&a[i]<=rj){
          ans--;
        }
      }
    }
  }
  cout<<ans<<endl;
}
0