結果

問題 No.1975 Zigzag Sequence
ユーザー umezo
提出日時 2022-06-10 23:59:27
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 70 ms / 2,000 ms
コード長 1,935 bytes
コンパイル時間 2,360 ms
コンパイル使用メモリ 205,572 KB
最終ジャッジ日時 2025-01-29 20:28:06
ジャッジサーバーID
（参考情報） 		judge1 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff # 

#define rep(i,n) for(int i=0;i<(int)(n);i++)
#define ALL(v) v.begin(),v.end()
typedef long long ll;

#include <bits/stdc++.h>
using namespace std;

const int MOD=1e9+7;

//1-index
template<typename T>
struct BIT{
private:
  vector<T> A;
  const int n;
  
public:
  BIT(int _n) : A(_n+1,0), n(_n){}
  
  //i番目までの累積和
  T sum(int i){
    T s=0;
    while(i>0){
      s=(s+A[i])%MOD;
      i-=i&-i;
    }
    return s%MOD;
  }
  
  //[i,j]の要素の総和
  T sum(int i,int j){
    return (sum(j)-sum(i-1)+MOD)%MOD;
  }
  
  //i番目にxを加算
  void add(int i,T x){
    while(i<=n){
      A[i]=(A[i]+x)%MOD;
      i+=i&-i;
    }
  }
};

template <typename T>
vector<T> compress(vector<T> &X){
  // ソートした結果を vals に
  vector<T> vals=X;
  sort(vals.begin(),vals.end());
  // 隣り合う重複を削除(unique), 末端のゴミを削除(erase)
  vals.erase(unique(vals.begin(), vals.end()),vals.end());
  // 各要素ごとに二分探索で位置を求める
  for(int i=0;i<(int)X.size(); i++){
    X[i]=lower_bound(vals.begin(),vals.end(),X[i])-vals.begin();
  }
  return vals;
}

int main(){
  ios::sync_with_stdio(false);
  std::cin.tie(nullptr);

  int n;
  cin>>n;
  vector<int> A(n);
  rep(i,n) cin>>A[i];
  compress(A);
  rep(i,n) A[i]++;
  
  BIT<ll> bit1(n+1),bit2(n+1);
  vector<ll> B(n);
  ll t=1;
  for(int i=0;i<n;i++){
    B[i]=bit1.sum(1,A[i]-1);
    bit1.add(A[i],t);
    t=t*2%MOD;
  }
  t=1;
  for(int i=n-1;i>=0;i--){
    B[i]=B[i]*bit2.sum(1,A[i]-1)%MOD;
    bit2.add(A[i],t);
    t=t*2%MOD;
  }
  ll ans=0;
  rep(i,n) ans=(ans+B[i])%MOD;
  
  BIT<ll> bit3(n+1),bit4(n+1);
  vector<ll> C(n);
  t=1;
  for(int i=0;i<n;i++){
    C[i]=bit3.sum(A[i]+1,n+1);
    bit3.add(A[i],t);
    t=t*2%MOD;
  }
  t=1;
  for(int i=n-1;i>=0;i--){
    C[i]=C[i]*bit4.sum(A[i]+1,n+1)%MOD;
    bit4.add(A[i],t);
    t=t*2%MOD;
  }
  rep(i,n) ans=(ans+C[i])%MOD;
  cout<<ans<<endl;
  
  
  return 0;
}
0