Submission #1343002

#	Time	Username	Problem	Language	Result	Execution time	Memory
1343002		po_rag526	Pilot (NOI19_pilot)	C++20	100 / 100	490 ms	112524 KiB

pilot

#include <bits/stdc++.h>
using namespace std;
#define ii pair <int, int>
#define X fisrt
#define Y second
const int maxn = 1e6 + 5;
int n, q, a[maxn];
int par[maxn], sz[maxn];
long long ans = 0, res[maxn];
int root (int u) {
    if (par[u] == 0) return u;
    return par[u] = root(par[u]);
}
void Union (int u, int v) {
    u = root(u), v = root(v);
    if (u == v) return;
    if (sz[u] < sz[v]) swap (u, v);
    ans += 1ll * sz[u] * sz[v];
    sz[u] += sz[v];
    par[v] = u;
}
vector <int> queries[maxn], pos[maxn];
signed main () {
    #define task "code"
    if (fopen (task".inp", "r")) {
        freopen (task".inp", "r", stdin);
        freopen (task".out", "w", stdout);
    }
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    cin >> n >> q;
    for (int i = 1, x; i <= n; i++) {
        cin >> a[i];
        pos[a[i]].push_back (i);
        sz[i] = 1;
    }
    for (int i = 1, x; i <= q; i++) {
        cin >> x;
        queries[x].push_back (i);
    }
    for (int i = 1; i <= 1e6; i++) {
        for (int p : pos[i]) {
            ans++;
            if (p > 1 && a[p - 1] <= i) {
                Union (p, p - 1);
//                cout << p << ' ' << p - 1 << '\n';
            }
            if (p < n && a[p + 1] <= i) {
                Union (p, p + 1);
//                cout << p << ' ' << p + 1 << '\n';
            }
        }
        for (int id : queries[i])
            res[id] = ans;
    }
    for (int i = 1; i <= q; i++)
        cout << res[i] << '\n';
}

Compilation message (stderr)

pilot.cpp: In function 'int main()':
pilot.cpp:26:17: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   26 |         freopen (task".inp", "r", stdin);
      |         ~~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~
pilot.cpp:27:17: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   27 |         freopen (task".out", "w", stdout);
      |         ~~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 3 / 3
• Subtask 2: 10 / 10
• Subtask 3: 12 / 12
• Subtask 4: 15 / 15
• Subtask 5: 5 / 5
• Subtask 6: 9 / 9
• Subtask 7: 14 / 14
• Subtask 8: 10 / 10
• Subtask 9: 11 / 11
• Subtask 10: 11 / 11