結果

問題 No.1975 Zigzag Sequence
ユーザー umezoumezo
提出日時 2022-06-10 23:59:27
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 68 ms / 2,000 ms
コード長 1,935 bytes
コンパイル時間 2,542 ms
コンパイル使用メモリ 212,388 KB
実行使用メモリ 8,212 KB
最終ジャッジ日時 2023-10-21 05:59:08
合計ジャッジ時間 5,847 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge10
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,348 KB
testcase_01 AC 2 ms
4,348 KB
testcase_02 AC 2 ms
4,348 KB
testcase_03 AC 8 ms
4,348 KB
testcase_04 AC 4 ms
4,348 KB
testcase_05 AC 60 ms
7,920 KB
testcase_06 AC 39 ms
6,204 KB
testcase_07 AC 9 ms
4,348 KB
testcase_08 AC 40 ms
6,492 KB
testcase_09 AC 59 ms
7,352 KB
testcase_10 AC 42 ms
6,480 KB
testcase_11 AC 46 ms
6,768 KB
testcase_12 AC 11 ms
4,348 KB
testcase_13 AC 2 ms
4,348 KB
testcase_14 AC 2 ms
4,348 KB
testcase_15 AC 2 ms
4,348 KB
testcase_16 AC 2 ms
4,348 KB
testcase_17 AC 2 ms
4,348 KB
testcase_18 AC 49 ms
8,212 KB
testcase_19 AC 49 ms
8,212 KB
testcase_20 AC 48 ms
8,212 KB
testcase_21 AC 50 ms
8,212 KB
testcase_22 AC 54 ms
8,212 KB
testcase_23 AC 53 ms
8,212 KB
testcase_24 AC 58 ms
8,212 KB
testcase_25 AC 59 ms
8,212 KB
testcase_26 AC 63 ms
8,212 KB
testcase_27 AC 68 ms
8,212 KB
testcase_28 AC 57 ms
8,212 KB
testcase_29 AC 57 ms
8,212 KB
testcase_30 AC 57 ms
8,212 KB
testcase_31 AC 57 ms
8,212 KB
testcase_32 AC 48 ms
8,212 KB
testcase_33 AC 49 ms
8,212 KB
testcase_34 AC 63 ms
8,212 KB
testcase_35 AC 62 ms
8,212 KB
権限があれば一括ダウンロードができます

ソースコード

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