結果
問題 |
No.1975 Zigzag Sequence
|
ユーザー |
![]() |
提出日時 | 2023-05-14 09:03:17 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 193 ms / 2,000 ms |
コード長 | 2,914 bytes |
コンパイル時間 | 1,207 ms |
コンパイル使用メモリ | 119,708 KB |
最終ジャッジ日時 | 2025-02-13 00:26:52 |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 33 |
ソースコード
#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; }