結果
問題 | No.121 傾向と対策:門松列(その2) |
ユーザー | kei |
提出日時 | 2018-08-11 20:42:22 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 680 ms / 5,000 ms |
コード長 | 2,880 bytes |
コンパイル時間 | 1,840 ms |
コンパイル使用メモリ | 178,744 KB |
実行使用メモリ | 42,344 KB |
最終ジャッジ日時 | 2024-09-23 06:26:50 |
合計ジャッジ時間 | 6,414 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 33 ms
34,688 KB |
testcase_01 | AC | 43 ms
34,860 KB |
testcase_02 | AC | 16 ms
34,564 KB |
testcase_03 | AC | 340 ms
42,344 KB |
testcase_04 | AC | 680 ms
42,320 KB |
testcase_05 | AC | 325 ms
42,188 KB |
testcase_06 | AC | 401 ms
42,156 KB |
testcase_07 | AC | 404 ms
42,260 KB |
testcase_08 | AC | 368 ms
42,232 KB |
ソースコード
#include "bits/stdc++.h" using namespace std; typedef long long ll; typedef pair<int, int> pii; typedef pair<ll, ll> pll; const int INF = 1e9; const ll LINF = 1e18; template<class S,class T> ostream& operator << (ostream& out,const pair<S,T>& o){ out << "(" << o.first << "," << o.second << ")"; return out; } template<class T> ostream& operator << (ostream& out,const vector<T> V){ for(int i = 0; i < V.size(); i++){ out << V[i]; if(i!=V.size()-1) out << " ";} return out; } template<class T> ostream& operator << (ostream& out,const vector<vector<T> > Mat){ for(int i = 0; i < Mat.size(); i++) { if(i != 0) out << endl; out << Mat[i];} return out; } template<class S,class T> ostream& operator << (ostream& out,const map<S,T> mp){ out << "{ "; for(auto it = mp.begin(); it != mp.end(); it++){ out << it->first << ":" << it->second; if(mp.size()-1 != distance(mp.begin(),it)) out << ", "; } out << " }"; return out; } /* <url:https://yukicoder.me/problems/no/121> 問題文============================================================ ================================================================= 解説============================================================= ================================================================ */ struct BIT { ll N; vector<ll> bit; BIT(ll N):N(N) { /* BITは[1..N]で扱う */ bit.resize(N + 1, 0); } void add(int x, ll val){ while (x <= N) { bit[x] += val; x += x & -x; } } ll sum(int x){ ll ret = 0; while (x) { ret += bit[x]; x &= (x - 1); } return (ret); } ll sum(int l,int r){ return sum(r) - sum(l-1); } }; const int MAXN = 1e6+10; ll solve(){ ll res = 0; int N; cin >> N; vector<int> A(N); for(auto& in:A) cin >> in; // compress vector<int> compA(A); sort(A.begin(),A.end()); A.erase(unique(A.begin(),A.end()),A.end()); for(int i = 0; i < N;i++){ compA[i] = (int)((lower_bound(A.begin(),A.end(),compA[i]) - A.begin()) + 1); } // main calc BIT lbit(MAXN),rbit(MAXN); res = 0; for(int i = 0; i < N;i++) rbit.add(compA[i],1); for(int i = 0; i < N;i++){ int V = compA[i]; rbit.add(V,-1); res += lbit.sum(V-1)*rbit.sum(V-1); res += lbit.sum(V+1,MAXN)*rbit.sum(V+1,MAXN); lbit.add(V,1); } vector<ll> Lcnt(MAXN),Rcnt(MAXN); for(int i = 0; i < N;i++) Rcnt[compA[i]]++; ll sub = 0; for(int i = 0; i < N;i++){ sub -= Lcnt[compA[i]]*Rcnt[compA[i]]; res -= sub; Lcnt[compA[i]]++; Rcnt[compA[i]]--; sub += Lcnt[compA[i]]*Rcnt[compA[i]]; } res -= sub; return res; } int main(void) { cin.tie(0); ios_base::sync_with_stdio(false); cout << solve() << endl; return 0; }