이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
using namespace std;
const int nx=1e6+5;
int n, q, h[nx], dp[nx], t, l, r;
vector<vector<int>> d(nx);
struct segtree
{
int d[4*nx];
void build(int l, int r, int i)
{
if (l==r) return void(d[i]=h[l]);
int md=(l+r)/2;
build(l, md, 2*i);
build(md+1, r, 2*i+1);
d[i]=max(d[2*i], d[2*i+1]);
}
int queryleft(int l, int r, int i, int idx)
{
if (idx<=l) return -1;
if (l==r&&d[i]>h[idx]) return l;
if (l==r&&d[i]<=h[idx]) return -1;
int md=(l+r)/2, tmp=queryleft(md+1, r, 2*i+1, idx);
if (tmp!=-1) return tmp;
return queryleft(l, md, 2*i, idx);
}
int queryright(int l, int r, int i, int idx)
{
//cout<<l<<' '<<r<<' '<<i<<'\n';
if (idx>=r) return -1;
if (l==r&&d[i]>=h[idx]) return l;
if (l==r&&d[i]<h[idx])return -1;
int md=(l+r)/2, tmp=queryright(l, md, 2*i, idx);
if (tmp!=-1) return tmp;
return queryright(md+1, r, 2*i+1, idx);
}
} s;
int main()
{
cin.tie(NULL)->sync_with_stdio(false);
cin>>n>>q;
for (int i=1; i<=n; i++) cin>>h[i], d[h[i]].push_back(i);
s.build(1, n, 1);
for (int i=1; i<nx; i++)
{
dp[i]=dp[i-1];
sort(d[i].begin(), d[i].end());
for (auto x:d[i])
{
l=1; r=n;
int left=s.queryleft(1, n, 1, x), right=s.queryright(1, n, 1, x);
if (left!=-1) l=left+1;
if (right!=-1) r=right-1;
dp[i]+=(x-l+1)*(r-x+1);
//cout<<i<<' '<<l<<' '<<r<<' '<<x<<' '<<dp[i]<<' '<<left<<' '<<right<<'\n';
}
}
//cout<<s.queryright(1, n, 1, 1);
while (q--) cin>>t, cout<<dp[t]<<'\n';
}
| # | Verdict | Execution time | Memory | Grader output |
|---|
| Fetching results... |
| # | Verdict | Execution time | Memory | Grader output |
|---|
| Fetching results... |
| # | Verdict | Execution time | Memory | Grader output |
|---|
| Fetching results... |
| # | Verdict | Execution time | Memory | Grader output |
|---|
| Fetching results... |
| # | Verdict | Execution time | Memory | Grader output |
|---|
| Fetching results... |
| # | Verdict | Execution time | Memory | Grader output |
|---|
| Fetching results... |
| # | Verdict | Execution time | Memory | Grader output |
|---|
| Fetching results... |
| # | Verdict | Execution time | Memory | Grader output |
|---|
| Fetching results... |
| # | Verdict | Execution time | Memory | Grader output |
|---|
| Fetching results... |
| # | Verdict | Execution time | Memory | Grader output |
|---|
| Fetching results... |