結果
問題 | No.1496 不思議な数え上げ |
ユーザー | penguinman |
提出日時 | 2021-03-30 06:49:44 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 738 ms / 3,000 ms |
コード長 | 1,484 bytes |
コンパイル時間 | 2,185 ms |
コンパイル使用メモリ | 202,028 KB |
最終ジャッジ日時 | 2025-01-20 00:42:27 |
ジャッジサーバーID (参考情報) |
judge5 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 39 |
ソースコード
#include<bits/stdc++.h> using std::cin; using std::cout; using std::endl; using std::vector; using ll = long long; #define rep(i,j,k) for(int i=int(j); i<int(k); i++) #define REP(i,j,k) for(int i=int(j); i<=int(k); i++) int main(){ ll N; cin >> N; vector<ll> P(N); rep(i,0,N) cin >> P[i]; vector<ll> sum(N+1), rev(N); rep(i,0,N){ sum[i+1] = sum[i]+P[i]; rev[--P[i]] = i; } std::set<ll> set; set.insert(-1); set.insert(N); rep(i,0,N){ ll A; cin >> A; auto itr = set.lower_bound(rev[i]); ll right = *itr; itr--; ll left = *itr; ll ans = 0; if(right-rev[i] < rev[i]-left){ rep(r,rev[i],right){ //右端固定 ll l = std::lower_bound(sum.begin(),sum.end(),sum[r+1]-A)-sum.begin(); //left < l ≤ rev[i] を満たす必要がある l = std::max(l, left+1); l = std::min(l, rev[i]+1); ans += rev[i]-l+1; } } else{ REP(l,left+1,rev[i]){ //左端固定 ll r = std::upper_bound(sum.begin(),sum.end(),sum[l]+A)-sum.begin()-2; //rev[i] ≤ r < right を満たす必要がある r = std::min(r, right-1); r = std::max(r, rev[i]-1); ans += r-rev[i]+1; } } cout << ans << "\n"; set.insert(rev[i]); } }