結果

問題 No.1975 Zigzag Sequence
ユーザー miscalcmiscalc
提出日時 2022-06-10 22:14:19
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 95 ms / 2,000 ms
コード長 2,550 bytes
コンパイル時間 4,304 ms
コンパイル使用メモリ 277,656 KB
実行使用メモリ 10,220 KB
最終ジャッジ日時 2023-10-21 05:19:38
合計ジャッジ時間 7,719 ms
ジャッジサーバーID
(参考情報)
judge13 / judge10
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,348 KB
testcase_01 AC 2 ms
4,348 KB
testcase_02 AC 2 ms
4,348 KB
testcase_03 AC 11 ms
4,348 KB
testcase_04 AC 5 ms
4,348 KB
testcase_05 AC 72 ms
6,868 KB
testcase_06 AC 51 ms
6,016 KB
testcase_07 AC 12 ms
4,348 KB
testcase_08 AC 50 ms
5,564 KB
testcase_09 AC 74 ms
7,384 KB
testcase_10 AC 52 ms
5,820 KB
testcase_11 AC 55 ms
6,164 KB
testcase_12 AC 14 ms
4,348 KB
testcase_13 AC 2 ms
4,348 KB
testcase_14 AC 2 ms
4,348 KB
testcase_15 AC 1 ms
4,348 KB
testcase_16 AC 2 ms
4,348 KB
testcase_17 AC 2 ms
4,348 KB
testcase_18 AC 68 ms
6,472 KB
testcase_19 AC 70 ms
6,472 KB
testcase_20 AC 76 ms
6,868 KB
testcase_21 AC 75 ms
6,872 KB
testcase_22 AC 95 ms
10,220 KB
testcase_23 AC 88 ms
10,220 KB
testcase_24 AC 92 ms
10,220 KB
testcase_25 AC 94 ms
10,220 KB
testcase_26 AC 78 ms
6,816 KB
testcase_27 AC 87 ms
8,184 KB
testcase_28 AC 93 ms
10,220 KB
testcase_29 AC 95 ms
10,220 KB
testcase_30 AC 95 ms
10,220 KB
testcase_31 AC 94 ms
10,220 KB
testcase_32 AC 67 ms
6,472 KB
testcase_33 AC 68 ms
6,472 KB
testcase_34 AC 72 ms
7,096 KB
testcase_35 AC 70 ms
7,096 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
//*
#include <atcoder/all>
using namespace atcoder;
//using mint = modint998244353;
using mint = modint1000000007;
//*/
using ll = long long;
using ld = long double;
using pll = pair<ll, ll>;
using tlll = tuple<ll, ll, ll>;
constexpr ll INF = 1LL << 60;
template<class T> bool chmin(T& a, T b) {if (a > b) {a = b; return true;} return false;}
template<class T> bool chmax(T& a, T b) {if (a < b) {a = b; return true;} return false;}
ll safemod(ll A, ll M) {ll res = A % M; if (res < 0) res += M; return res;}
ll divfloor(ll A, ll B) {if (B < 0) {return divfloor(-A, -B);} return (A - safemod(A, B)) / B;}
ll divceil(ll A, ll B) {if (B < 0) {return divceil(-A, -B);} return divfloor(A + B - 1, B);}
ll mypowl(ll A, ll B) {ll res = 1; for (int i = 0; i < B; i++) {res *= A;} return res;}
template<class T> void unique(vector<T> &V) {V.erase(unique(V.begin(), V.end()), V.end());}
template<class T> void sortunique(vector<T> &V) {sort(V.begin(), V.end()); V.erase(unique(V.begin(), V.end()), V.end());}
#define FINALANS(A) do {cout << (A) << '\n'; exit(0);} while (false)
template<class T> void printvec(const vector<T> &V) {int _n = V.size(); for (int i = 0; i < _n; i++) cout << V[i] << (i == _n - 1 ? "" : " ");cout << '\n';}
template<class T> void printvect(const vector<T> &V) {for (auto v : V) cout << v << '\n';}
template<class T> void printvec2(const vector<vector<T>> &V) {for (auto &v : V) printvec(v);}

template <class T> class zarts
{
public:
  int N;
  vector<T> vals;
  vector<T> vec;
  // map<T, int> mp;
  unordered_map<T, int> mp;
  zarts(const vector<T> &V, int cnt = 0)
  {
    N = V.size();

    vals = vector<T>(V);
    sortunique(vals);
    for (auto v : vals)
    {
      mp[v] = cnt;
      cnt++;
    }

    vec.resize(N);
    for (int i = 0; i < N; i++)
      vec[i] = mp[V[i]];
  }
};

int main()
{
  ll N;
  cin >> N;
  vector<ll> A(N);
  for (ll i = 0; i < N; i++)
  {
    cin >> A.at(i);
  }
  A = zarts(A).vec;

  vector<mint> B(N), C(N), D(N), E(N);
  fenwick_tree<mint> fw1(N), fw2(N);
  for (ll i = 0; i < N; i++)
  {
    B.at(i) = fw1.sum(0, A.at(i));
    D.at(i) = fw1.sum(A.at(i) + 1, N);
    fw1.add(A.at(i), mint(2).pow(i));
  }
  for (ll i = N - 1; i >= 0; i--)
  {
    C.at(i) = fw2.sum(0, A.at(i));
    E.at(i) = fw2.sum(A.at(i) + 1, N);
    fw2.add(A.at(i), mint(2).pow(N - 1 - i));
  }

  mint ans = 0;
  for (ll i = 0; i < N; i++)
  {
    mint tmp1 = B.at(i) * C.at(i);
    mint tmp2 = D.at(i) * E.at(i);
    ans += tmp1 + tmp2;
  }
  cout << ans.val() << endl;
}
0