Submission #1059325

#	Time	Username	Problem	Language	Result	Execution time	Memory
1059325		SN0WM4N	Financial Report (JOI21_financial)	C++14	5 / 100	411 ms	55632 KiB

Main

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#define sort undefined_function // To use stable_sort instead sort 
#define bpc __builtin_popcount
#define ull unsigned long long
#define ld double
#define ll long long
#define mp make_pair
#define F first
#define S second
 
#pragma GCC optimize("O3")
 
using namespace __gnu_pbds;
using namespace std;
 
typedef tree<long long, null_type, less_equal<long long>,
    rb_tree_tag, tree_order_statistics_node_update> Tree;
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
 
const ll INF = 9223372036854775807LL;
const ll inf = 2147483647;
const ll MOD = 998244353; //[998244353, 1e9 + 7, 1e9 + 13]
const ld PI = acos(-1);
const ll NROOT = 300;
 
ll binpow(ll a, ll b, ll _MOD = -1) {
        if (_MOD == -1)
                _MOD = MOD;
        ll res = 1;
        for (; b; b /= 2, a *= a, a %= _MOD)
                if (b & 1) res *= a, res %= _MOD;
        return res;
}
 
void set_IO(string s) {
#ifndef LOCAL
        string in  = s +  ".in";
        string out = s + ".out";
        freopen(in.c_str(),  "r",  stdin);
        freopen(out.c_str(), "w", stdout);
#endif
}
 
bool dataOverflow(ll a, ll b) {return (log10(a) + log10(b) >= 18);}
ll gcd(ll a, ll b) {return b ? gcd(b, a % b) : a;}
ll lcm(ll a, ll b) {return a * b / gcd(a, b);}
ll ceil(ll a, ll b) {return (a + b - 1) / b;}
 
class ST{
private:
        int n;
        vector<ll> st;
 
        void update(int nod, int l, int r, int pos, ll val) {
                if (l == r) {
                        st[nod] += val;
                        return;
                }
 
                int mid = (l + r) >> 1;
                if (pos <= mid) 
                        update(nod << 1, l, mid, pos, val);
                else 
                        update((nod << 1) | 1, mid + 1, r, pos, val);
                st[nod] = max(st[nod << 1], st[(nod << 1) | 1]);
        }
 
        ll query(int nod, int l, int r, int L, int R) {
                if (r < L || R < l)
                        return -100000000;
                if (L <= l && r <= R) 
                        return st[nod];
 
                int mid = (l + r) >> 1;
                ll q1 = query(nod << 1, l, mid, L, R);
                ll q2 = query((nod << 1) | 1, mid + 1, r, L, R);
 
                return max(q1, q2);
        }
 
public:
        ST(int _n) {
                n = _n;
                st.resize(4 * n + 10);
        }
 
        void update(int pos, ll val) {
                update(1, 1, n, pos, val);
        }
 
        ll query(int L, int R) {
                if (R < L) return 0;
                return query(1, 1, n, L, R);
        }

        void clear() {
                for (auto &x : st)
                        x = 0;
        }
};
 
int n, d;
vector<int> v;
 
int32_t main() {
        ios_base::sync_with_stdio(0);
        cin.tie(0);

        cin >> n >> d;
    
        v.resize(n + 1);
        vector<pair<int, int>> aux;

        set<int> s;

        for (int i = 1; i <= n; i ++) {
                cin >> v[i];
                s.insert(v[i]);
        }

        int index = 0;
        map<int, int> ID;

        for (auto &x : s) 
            ID[x] = ++index;

        vector<vector<int>> f(n + 1);

        for (int i = 1; i <= n; i ++) 
            f[ID[v[i]]].push_back(i);


        ll res = 0;
        ST ans(n + 1);

        for (int val = 1; val <= n; val ++) {
            for (auto &i : f[val]) {
                ll tmp = ans.query(max(1, i - d), i) + 1;
                ans.update(i, tmp - 1);
                res = max(res, tmp);
            }

            for (auto &i : f[val])
                ans.update(i, 1);
        }

        cout << res << "\n";

        return 0;
}

Compilation message (stderr)

Main.cpp: In function 'void set_IO(std::string)':
Main.cpp:41:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   41 |         freopen(in.c_str(),  "r",  stdin);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~
Main.cpp:42:16: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   42 |         freopen(out.c_str(), "w", stdout);
      |         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 0 / 14
• Subtask 2: 0 / 14
• Subtask 3: 0 / 20
• Subtask 4: 0 / 12
• Subtask 5: 5 / 5
• Subtask 6: 0 / 35