結果
| 問題 |
No.2374 ASKT Subsequences
|
| コンテスト | |
| ユーザー |
 mikam
|
| 提出日時 | 2023-07-07 22:14:35 |
| 言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
| 結果 |
AC
|
| 実行時間 | 509 ms / 2,000 ms |
| コード長 | 8,701 bytes |
| コンパイル時間 | 3,343 ms |
| コンパイル使用メモリ | 238,100 KB |
| 最終ジャッジ日時 | 2025-02-15 07:36:55 |
|
ジャッジサーバーID （参考情報） |
judge4 / judge5 |
（要ログイン）
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 3 |
| other | AC * 28 |
ソースコード
// #include "atcoder/convolution"
#include "atcoder/dsu"
#include "atcoder/fenwicktree"
#include "atcoder/lazysegtree"
#include "atcoder/math"
#include "atcoder/maxflow"
#include "atcoder/mincostflow"
#include "atcoder/modint"
#include "atcoder/scc"
#include "atcoder/segtree"
#include "atcoder/string"
#include "atcoder/twosat"
using namespace atcoder;
#include <bits/stdc++.h>
using namespace std;
// #include <boost/multiprecision/cpp_int.hpp>
#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#define rep(i, n) for (int i = 0; i < (int)(n); i++)
#define rep2(i,a,b) for (int i = (int)(a); i < (int)(b); i++)
#define all(v) v.begin(),v.end()
#define inc(x,l,r) ((l)<=(x)&&(x)<(r))
#define Unique(x) sort(all(x)), x.erase(unique(all(x)), x.end())
#define pcnt __builtin_popcountll
typedef long long ll;
#define int ll
using ld = long double;
using vi = vector<int>;
using vs = vector<string>;
using P = pair<int,int>;
using vp = vector<P>;
// using Bint = boost::multiprecision::cpp_int;
template<typename T1,typename T2> bool chmax(T1 &a, const T2 b) {if (a < b) {a = b; return true;} else return false; }
template<typename T1,typename T2> bool chmin(T1 &a, const T2 b) {if (a > b) {a = b; return true;} else return false; }
template<typename T> using priority_queue_greater = priority_queue<T, vector<T>, greater<T>>;
template<typename T> istream &operator>>(istream& is,vector<T> &v){for(T &in:v)is>>in;return is;}
template<typename T1,typename T2> ostream &operator<< (ostream &os, const pair<T1,T2> &p){os << p.first <<" "<<p.second;return os;}
ostream &operator<< (ostream &os, const modint1000000007 &m){os << m.val();return os;}
istream &operator>> (istream &is, modint1000000007 &m){ll in;is>>in;m=in;return is;}
ostream &operator<< (ostream &os, const modint998244353 &m){os << m.val();return os;}
istream &operator>> (istream &is, modint998244353 &m){ll in;is>>in;m=in;return is;}
template<class... T> void input(T&... a){(cin>> ... >> a);}
#ifdef LOCAL
template<typename T> ostream &operator<<(ostream &os,const vector<T> &v){os<<"\x1b[32m";rep(i,v.size())os<<v[i]<<(i+1!=v.size()?" ":"");os<<"\x1b[0m";return os;}
template<class T> void print(T& a){cout << "\x1b[32m"<< a<< '\n' << "\x1b[0m";}
template<class T,class... Ts> void print(const T&a, const Ts&... b){cout << "\x1b[32m" << a;(cout<<...<<(cout<<' ',b));cout<<'\n' << "\x1b[0m";}
#else
template<typename T> ostream &operator<<(ostream &os,const vector<T> &v){rep(i,v.size())os<<v[i]<<(i+1!=v.size()?" ":"");return os;}
template<class T> void print(T& a){cout << a<< '\n';}
template<class T,class... Ts> void print(const T&a, const Ts&... b){cout << a;(cout<<...<<(cout<<' ',b));cout<<'\n';}
#endif
#define VI(v,n) vi v(n); input(v)
#define INT(...) int __VA_ARGS__; input(__VA_ARGS__)
#define STR(...) string __VA_ARGS__; input(__VA_ARGS__)
#define CHAR(...) char __VA_ARGS__; input(__VA_ARGS__)
int sign(ll x){return x>0?1:x<0?-1:0;}
ll ceil(ll x,ll y){assert(y!=0);if(sign(x)==sign(y))return (x+y-1)/y;return -((-x/y));}
ll floor(ll x,ll y){assert(y!=0);if(sign(x)==sign(y))return x/y;if(y<0)x*=-1,y*=-1;return x/y-(x%y<0);}
ll abs(ll x,ll y){return abs(x-y);}
ll bit(int n){return 1ll<<n;}
bool ins(string s,string t){return s.find(t)!=string::npos;}
P operator+ (const P &p, const P &q){ return P{p.first+q.first,p.second+q.second};}
P operator- (const P &p, const P &q){ return P{p.first-q.first,p.second-q.second};}
int yesno(bool ok,string y="Yes",string n="No"){ cout<<(ok?y:n)<<endl;return 0;}
int YESNO(bool ok,string y="YES",string n="NO"){ cout<<(ok?y:n)<<endl;return 0;}
int di[]={-1,0,1,0,-1,-1,1,1};
int dj[]={0,1,0,-1,-1,1,-1,1};
const ll INF = 8e18;
//using mint = modint1000000007;
//using mint = modint998244353;
//mint stom(const string &s,int b=10){mint res = 0;for(auto c:s)res *= b,res += c-'0';return res;}
struct BitVector {
using uint = unsigned;
constexpr static int w = 32;
#define rem(k) ((k) & (w - 1))
int len, blocks;
vector<uint> bit, sum;
BitVector(){}
BitVector(int len) : len(len), blocks((len+w-1)/w){
bit.assign(blocks, 0);
sum.assign(blocks, 0);
}
void set(const int k, const int b = 1){
if(b) bit[k / w] |= 1U << rem(k);
else bit[k / w] &= ~(1U << rem(k));
}
void build(){
sum[0] = 0;
for(int i = 1; i < blocks; i++){
sum[i] = sum[i - 1] + __builtin_popcount(bit[i - 1]);
}
}
int operator[](const int k) const{
return bit[k / w] >> rem(k) & 1;
}
inline int rank(const int k) const{
return sum[k / w] + __builtin_popcount(bit[k / w] & ((1U << rem(k)) - 1));
}
inline int rank(const int val, const int k) const{
return val ? rank(k) : k - rank(k);
}
#undef rem
};
template <class T, int maxlog>
struct WaveletMatrix {
int len;
BitVector matrix[maxlog];
int mid[maxlog];
WaveletMatrix(){}
WaveletMatrix(vector<T> v) : len(v.size()){
vector<T> l(len), r(len);
fill(matrix, matrix + maxlog, BitVector(len + 1));
for(int level = maxlog - 1; level >= 0; level--){
int left = 0, right = 0;
for(int i = 0; i < len; i++){
if(v[i] >> level & 1){
matrix[level].set(i);
r[right++] = v[i];
}else{
l[left++] = v[i];
}
}
mid[level] = left;
matrix[level].build();
v.swap(l);
copy(r.begin(), r.begin() + right, v.begin() + left);
}
}
inline int succ(const int f, const int l, const int level) const{
return matrix[level].rank(f, l) + mid[level] * f;
}
//find v[k]
T access(int k) const{
assert(0 <= k && k < len);
T res = 0;
for(int level = maxlog - 1; level >= 0; level--){
const bool f = matrix[level][k];
if(f) res |= T(1) << level;
k = matrix[level].rank(f, k) + mid[level]*f;
}
return res;
}
T operator[](const int k) const{ return access(k); }
// count i (0 <= i < r) && v[i] == x
int rank(const T x, int r) const{
assert(0 <= r && r <= len);
int l = 0;
for(int level = maxlog - 1; level >= 0; level--){
l = succ(x >> level & 1, l, level);
r = succ(x >> level & 1, r, level);
}
return r - l;
}
//k-th value v[l,r)
T kth_smallest(int l, int r, int k) const{
assert(0 <= l && l <= r && r <= len);
assert(0 <= k && k < r - l);
T res = 0;
for(int level = maxlog - 1; level >= 0; level--){
int cnt = matrix[level].rank(0, r) - matrix[level].rank(0, l);
bool f = cnt <= k;
if(f){
res |= T(1) << level;
k -= cnt;
}
l = succ(f, l, level);
r = succ(f, r, level);
}
return res;
}
T kth_largest(int l, int r, int k) const{
return kth_smallest(l, r, r-l-k-1);
}
// count i (l <= i < r), (v[i] < upper)
int range_freq(int l, int r, const T upper) const{
int res = 0;
for(int level = maxlog - 1; level >= 0; level--){
bool f = upper >> level & 1;
if(f) res += matrix[level].rank(0, r) - matrix[level].rank(0, l);
l = succ(f, l, level);
r = succ(f, r, level);
}
return res;
}
// count i (l <= i < r), (lower <= v[i] < upper)
int range_freq(int l, int r, const T lower, const T upper) const{
assert(0 <= l && l <= r && r <= len);
return range_freq(l, r, upper) - range_freq(l, r, lower);
}
// max v[i] (l <= i < r), (v[i] < upper)
T prev_value(int l, int r, const T upper) const{
assert(0 <= l && l <= r && r <= len);
int cnt = range_freq(l, r, upper);
return cnt == 0 ? -1 : kth_smallest(l, r, cnt - 1);
}
// min v[i] (l <= i < r), (lower <= v[i])
T next_value(int l, int r, const T lower) const{
assert(0 <= l && l <= r && r <= len);
int cnt = range_freq(l, r, lower);
return cnt == r - l ? -1 : kth_smallest(l, r, cnt);
}
};
signed main() {
cin.tie(0);
ios_base::sync_with_stdio(false);
cout << fixed << setprecision(20);
INT(n);
vi a(n);cin>>a;
WaveletMatrix<int,20> m(a);
int ans = 0;
for(int j=1;j+2<n;j++)for(int k=j+1;k+1<n;k++){
int ai = a[k]-10;
if(ai<1||ai>2000)continue;
int d = a[j]-a[k];
if(d<=0)continue;
ans += m.range_freq(0,j,ai,ai+1)*m.range_freq(k,n,ai+d+11,ai+d+12);
}
print(ans);
return 0;
}