結果

問題 No.2374 ASKT Subsequences
ユーザー mikammikam
提出日時 2023-07-07 22:14:35
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
AC  
実行時間 507 ms / 2,000 ms
コード長 8,701 bytes
コンパイル時間 3,729 ms
コンパイル使用メモリ 244,192 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-21 18:19:54
合計ジャッジ時間 7,113 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

// #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;
}
0