#include <bits/stdc++.h>
#define int long long
using namespace std;
struct star {
int i, j, c;
};
struct SegTree {
struct node {
long long maxx, lazy;
int leftSon, rightSon;
};
int root, lst, rst;
vector<node> segTree;
int newNode() {
int v = segTree.size();
segTree.push_back( { 0, 0, -1, -1 } );
return v;
}
void init( int l, int r ) {
lst = l;
rst = r;
root = newNode();
}
void propag( int v, int l, int r ) {
segTree[v].maxx += segTree[v].lazy;
if ( l != r ) {
if ( segTree[v].leftSon == -1 )
segTree[v].leftSon = newNode();
segTree[segTree[v].leftSon].lazy += segTree[v].lazy;
if ( segTree[v].rightSon == -1 )
segTree[v].rightSon = newNode();
segTree[segTree[v].rightSon].lazy += segTree[v].lazy;
}
segTree[v].lazy = 0;
}
void updatePoint( int v, int l, int r, int p, long long x ) {
propag( v, l, r );
if ( l > p || r < p )
return;
segTree[v].maxx = max( segTree[v].maxx, x );
if ( l == r )
return;
int mid = (l + r) / 2;
if ( segTree[v].leftSon == -1 )
segTree[v].leftSon = newNode();
updatePoint( segTree[v].leftSon, l, mid, p, x );
if ( segTree[v].leftSon == -1 )
segTree[v].leftSon = newNode();
updatePoint( segTree[v].rightSon, mid + 1, r, p, x );
}
void updatePoint( int p, long long x ) {
updatePoint( root, lst, rst, p, x );
}
void updateInterval( int v, int l, int r, int lu, int ru, long long x ) {
propag( v, l, r );
if ( l > ru || r < lu )
return;
segTree[v].maxx = max( segTree[v].maxx, x );
if ( lu <= l && r <= ru ) {
segTree[v].lazy = x;
propag( v, l, r );
return;
}
int mid = (l + r) / 2;
if ( segTree[v].leftSon == -1 )
segTree[v].leftSon = newNode();
updateInterval( segTree[v].leftSon, l, mid, lu, ru, x );
if ( segTree[v].leftSon == -1 )
segTree[v].leftSon = newNode();
updateInterval( segTree[v].rightSon, mid + 1, r, lu, ru, x );
segTree[v].maxx = max( segTree[segTree[v].leftSon].maxx, segTree[segTree[v].rightSon].maxx );
}
void updateInterval( int l, int r, long long x ) {
updateInterval( root, lst, rst, l, r, x );
}
long long query( int v, int l, int r, int lq, int rq ) {
propag( v, l, r );
if ( l > rq || r < lq )
return 0;
if ( lq <= l && r <= rq )
return segTree[v].maxx;
int mid = (l + r) / 2;
long long a = (segTree[v].leftSon == -1 ? 0 : query( segTree[v].leftSon, l, mid, lq, rq ) );
long long b = (segTree[v].rightSon == -1 ? 0 : query( segTree[v].rightSon, mid + 1, r, lq, rq ) );
return max( a, b );
}
long long query( int r ) {
return query( root, lst, rst, lst, r );
}
};
const int MAX_N = 2e5;
const int MAX_M = 2e5;
int n;
int h[MAX_N + 2], parent[MAX_N + 2], leftSon[MAX_N + 2], rightSon[MAX_N + 2];
long long dp[2002][2002];
star stars[MAX_M];
vector<int> starsByColumn[MAX_N + 2];
set<int> heights[MAX_N + 2];
SegTree ds[MAX_N + 2];
void calcCost( int v ) {
ds[v].init( 0, n + 1 );
if ( v == 0 )
return;
int l = leftSon[v], r = rightSon[v];
calcCost( leftSon[v] );
calcCost( rightSon[v] );
for ( int x = 0; x <= n; x++ ) {
if ( x <= h[v] )
dp[v][x] = dp[leftSon[v]][x] + dp[rightSon[v]][x];
else
dp[v][x] = max( dp[leftSon[v]][x] + dp[rightSon[v]][h[v]], dp[leftSon[v]][h[v]] + dp[rightSon[v]][x] );
}
for ( int i: starsByColumn[v] )
dp[v][stars[i].j] = max( dp[v][stars[i].j], dp[v][h[v]] + stars[i].c );
for ( int x = 1; x <= n; x++ )
dp[v][x] = max( dp[v][x], dp[v][x - 1] );
}
signed main() {
int m;
long long total = 0;
cin >> n;
for ( int i = 1; i <= n; i++ )
cin >> h[i];
cin >> m;
for ( int i = 0; i < m; i++ ) {
cin >> stars[i].i >> stars[i].j >> stars[i].c;
total += stars[i].c;
}
h[n + 1] = n + 1;
vector<int> stack;
for ( int i = 1; i <= n + 1; i++ ) {
vector<int> path;
while ( !stack.empty() && h[i] >= h[stack.back()] ) {
path.push_back( stack.back() );
stack.pop_back();
}
path.push_back( i );
stack.push_back( i );
for ( int j = 0; j < path.size() - 1; j++ )
parent[path[j]] = path[j + 1];
}
for ( int v = 1; v <= n; v++ ) {
if ( v < parent[v] )
leftSon[parent[v]] = v;
else
rightSon[parent[v]] = v;
}
for ( int i = 0; i < m; i++ )
starsByColumn[stars[i].i].push_back( i );
calcCost( n + 1 );
cout << total - dp[n + 1][n + 1];//ds[n + 1].query( n );
return 0;
}
Compilation message
constellation3.cpp: In function 'void calcCost(long long int)':
constellation3.cpp:132:9: warning: unused variable 'l' [-Wunused-variable]
132 | int l = leftSon[v], r = rightSon[v];
| ^
constellation3.cpp:132:25: warning: unused variable 'r' [-Wunused-variable]
132 | int l = leftSon[v], r = rightSon[v];
| ^
constellation3.cpp: In function 'int main()':
constellation3.cpp:172:28: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
172 | for ( int j = 0; j < path.size() - 1; j++ )
| ~~^~~~~~~~~~~~~~~~~
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
7 ms |
35676 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
7 ms |
35676 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
7 ms |
35676 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
