前提・実現したいこと
AtCoder Beginner Contest 212のD問題(https://atcoder.jp/contests/abc212/tasks/abc212_d)でpriority_queueを用いて以下のようなコードを書いたところ、TLEとなりました。(https://atcoder.jp/contests/abc212/submissions/24678360)
計算量はO(QlogQ)だと思ったのですが、どこで計算量が大きくなっているのでしょうか。
該当のソースコード
c++
1#define _GLIBCXX_DEBUG 2#include <bits/stdc++.h> 3using namespace std; 4using LL = long long; 5using VLL = vector<LL>; 6#define REP(i,m,n) for(LL i=(m);i<(n);i++) 7#define all(v) v.begin(), v.end() 8 9int main(){ 10 LL q; 11 cin >> q; 12 VLL p(q),x(q),sum(q,0); 13 REP(i,0,q){ 14 cin >> p[i]; 15 if(p[i]!=3){cin >> x[i];} 16 17 if(i>0){sum[i] = sum[i-1];} 18 if(p[i]==2){sum[i] += x[i];} 19 if(p[i]==1){x[i] -= sum[i];} 20 } 21 22 priority_queue< 23 LL, 24 std::vector<LL>, 25 std::greater<LL>> pq; 26 REP(i,0,q){ 27 if(p[i]==1){ 28 pq.push(x[i]); 29 } 30 if(p[i]==3){ 31 LL ans = pq.top() + sum[i]; 32 pq.pop(); 33 cout << ans << endl; 34 } 35 } 36 37 return 0; 38}
試したこと
公式の解説(https://atcoder.jp/contests/abc212/editorial/2356)を読んで自分のコードと見比べました。
入力部分を分けて、累積和を配列でとったことが違うのは分かりましたが、それによって計算量が大幅に変わるとは思いませんでした。
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。
2021/08/01 22:24