Submission #330771

#	Time	Username	Problem	Language	Result	Execution time	Memory
330771		ThaiBaHung	Pilot (NOI19_pilot)	C++14	89 / 100	1090 ms	135592 KiB

pilot

#include <bits/stdc++.h>
using namespace std;
const int N = 1e6 + 3;
int n, q;
int a[N];
int qu[N];
long long L[N], R[N];
stack < int > st;
vector < int > num[N];
vector < int > pos[N];
long long ans[N];
vector < int > cur;

int ma = 0;

int main()
{
    ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    //freopen("pilot.in","r",stdin); freopen("pilot.out","w",stdout);

    cin >> n >> q;

    for (int i = 1; i <= n; i++)
    {
        cin >> a[i];
        num[a[i]].push_back(i);
    }

    for (int i = 1; i <= q; i++)
        cin >> qu[i], ma = max(ma, qu[i]), pos[qu[i]].push_back(i);

    st.push(0);
    a[0] = 1e9 + 7;

    for (int i = 1; i <= n; i++)
    {
        while (st.size() && a[i] >= a[st.top()])
            st.pop();
        L[i] = st.top() + 1;
        st.push(i);
    }

    while (st.size()) st.pop();
    a[n + 1] = 1e9 + 7;
    st.push(n + 1);

    for (int i = n; i >= 1; i--)
    {
        while (st.size() && a[i] > a[st.top()])
            st.pop();
        R[i] = st.top() - 1;
        st.push(i);
    }

    long long res = 0;

    for (int i = 1; i <= ma; i++)
    {
        for (int u : num[i])
            cur.push_back(u);

        if (pos[i].size())
        {
            for (int u : cur)
            {
                long long lo = u - L[u] + 1;
                long long hi = R[u] - u + 1;
                res += hi * lo;
            }

            for (int vt : pos[i])
                ans[vt] = res;

            cur.clear();
        }
    }

    for (int i = 1; i <= q; i++)
        cout << ans[i] << '\n';
}

• 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: 0 / 11