結果
問題 | No.1975 Zigzag Sequence |
ユーザー | srjywrdnprkt |
提出日時 | 2023-05-14 09:03:17 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 181 ms / 2,000 ms |
コード長 | 2,914 bytes |
コンパイル時間 | 1,476 ms |
コンパイル使用メモリ | 124,252 KB |
実行使用メモリ | 8,448 KB |
最終ジャッジ日時 | 2024-05-07 08:06:20 |
合計ジャッジ時間 | 7,146 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
6,816 KB |
testcase_01 | AC | 2 ms
6,944 KB |
testcase_02 | AC | 2 ms
6,944 KB |
testcase_03 | AC | 18 ms
6,944 KB |
testcase_04 | AC | 9 ms
6,940 KB |
testcase_05 | AC | 145 ms
6,940 KB |
testcase_06 | AC | 100 ms
6,940 KB |
testcase_07 | AC | 19 ms
6,944 KB |
testcase_08 | AC | 96 ms
6,940 KB |
testcase_09 | AC | 146 ms
6,940 KB |
testcase_10 | AC | 99 ms
6,944 KB |
testcase_11 | AC | 110 ms
6,944 KB |
testcase_12 | AC | 27 ms
6,940 KB |
testcase_13 | AC | 1 ms
6,940 KB |
testcase_14 | AC | 1 ms
6,940 KB |
testcase_15 | AC | 1 ms
6,944 KB |
testcase_16 | AC | 1 ms
6,940 KB |
testcase_17 | AC | 2 ms
6,940 KB |
testcase_18 | AC | 131 ms
6,944 KB |
testcase_19 | AC | 130 ms
6,944 KB |
testcase_20 | AC | 133 ms
6,944 KB |
testcase_21 | AC | 137 ms
6,940 KB |
testcase_22 | AC | 162 ms
8,320 KB |
testcase_23 | AC | 159 ms
8,320 KB |
testcase_24 | AC | 161 ms
8,320 KB |
testcase_25 | AC | 162 ms
8,320 KB |
testcase_26 | AC | 152 ms
6,940 KB |
testcase_27 | AC | 181 ms
7,936 KB |
testcase_28 | AC | 167 ms
8,320 KB |
testcase_29 | AC | 167 ms
8,448 KB |
testcase_30 | AC | 169 ms
8,320 KB |
testcase_31 | AC | 165 ms
8,320 KB |
testcase_32 | AC | 132 ms
6,940 KB |
testcase_33 | AC | 133 ms
6,940 KB |
testcase_34 | AC | 151 ms
6,940 KB |
testcase_35 | AC | 153 ms
6,940 KB |
ソースコード
#include <iostream> #include <vector> #include <cmath> #include <map> #include <set> #include <iomanip> #include <queue> #include <algorithm> #include <numeric> #include <deque> #include <complex> using namespace std; using ll = long long; const ll modc = 1e9+7; class mint { ll x; public: mint(ll x=0) : x((x%modc+modc)%modc) {} mint operator-() const { return mint(-x); } mint& operator+=(const mint& a) { if ((x += a.x) >= modc) x -= modc; return *this; } mint& operator-=(const mint& a) { if ((x += modc-a.x) >= modc) x -= modc; return *this; } mint& operator*=(const mint& a) { (x *= a.x) %= modc; return *this; } mint operator+(const mint& a) const { mint res(*this); return res+=a; } mint operator-(const mint& a) const { mint res(*this); return res-=a; } mint operator*(const mint& a) const { mint res(*this); return res*=a; } mint pow(ll t) const { if (!t) return 1; mint a = pow(t>>1); a *= a; if (t&1) a *= *this; return a; } mint inv() const { return pow(modc-2); } mint& operator/=(const mint& a) { return (*this) *= a.inv(); } mint operator/(const mint& a) const { mint res(*this); return res/=a; } friend ostream& operator<<(ostream& os, const mint& m){ os << m.x; return os; } friend istream& operator>>(istream& ip, mint&m) { ip >> m.x; return ip; } }; template<class T> struct BIT { vector<T> bit; int N; BIT (int n, T p = 0){ N = n; bit.resize(N); } void add(int loc, T val){ loc++; while(loc <= N){ bit[loc-1] += val; loc += loc & -loc; } } T sum(int l, int r){ T res = _sum(r) - _sum(l-1); return res; } T _sum(int r){ r++; T res = 0; while(r > 0){ res += bit[r-1]; r -= r & -r; } return res; } }; template <typename T> void compress(vector<T> &A){ map<T, T> comp; int N = A.size(), i=0; for (int i=0; i<N; i++) comp[A[i]]; for (auto &e : comp){ e.second = i; i++; } for (int i=0; i<N; i++) A[i] = comp[A[i]]; } int main(){ int N; cin >> N; mint ans = 0, b=2; vector<int> A(N); for (int i=0; i<N; i++) cin >> A[i]; compress(A); BIT<mint> tree(N), tree2(N); for (int i=0; i<N; i++){ tree2.add(A[i], b.pow(N-1-i)); } for (int i=0; i<N; i++){ tree2.add(A[i], -b.pow(N-1-i)); ans += tree.sum(0, A[i]-1) * tree2.sum(0, A[i]-1); ans += tree.sum(A[i]+1, N-1) * tree2.sum(A[i]+1, N-1); tree.add(A[i], b.pow(i)); } cout << ans << endl; return 0; }