//#pragma GCC optimize("O3")
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#include <ext/rope>
#define mem(a,v) memset((a), (v), sizeof (a))
#define enl printf("\n")
#define case(t) printf("Case #%d: ", (t))
#define ni(n) scanf("%d", &(n))
#define nl(n) scanf("%I64d", &(n))
#define nai(a, n) for (int i = 0; i < (n); i++) ni(a[i])
#define nal(a, n) for (int i = 0; i < (n); i++) nl(a[i])
#define pri(n) printf("%d\n", (n))
#define prl(n) printf("%I64d\n", (n))
#define pii pair<int, int>
#define pil pair<int, long long>
#define pll pair<long long, long long>
#define vii vector<pii>
#define vil vector<pil>
#define vll vector<pll>
#define vi vector<int>
#define vl vector<long long>
#define pb push_back
#define mp make_pair
#define fi first
#define se second
using namespace std;
using namespace __gnu_pbds;
typedef long long ll;
typedef cc_hash_table<int,int,hash<int>> ht;
typedef tree<int,null_type,less<int>,rb_tree_tag,tree_order_statistics_node_update> oset;
const double pi = acos(-1);
const int MOD = 1e9 + 7;
const int INF = 2e9 + 7;
const int MAXN = 5e5 + 5;
const double eps = 1e-9;
int dp[MAXN], seg1[MAXN*4], seg2[MAXN*4];
int BIT1[MAXN], BIT2[MAXN];
void upd1(int ind, int v) {
ind = MAXN - ind - 1;
while (ind < MAXN) {
BIT1[ind] = max(BIT1[ind], v);
ind += ind & -ind;
}
}
void upd2(int ind, int v) {
ind++;
while (ind < MAXN) {
BIT2[ind] = max(BIT2[ind], v);
ind += ind & -ind;
}
}
int qry1(int ind) {
ind = MAXN - ind - 1;
int ret = -INF;
while (ind > 0) {
ret = max(ret, BIT1[ind]);
ind -= ind & -ind;
}
return ret;
}
int qry2(int ind) {
ind++;
int ret = -INF;
while (ind > 0) {
ret = max(ret, BIT2[ind]);
ind -= ind & -ind;
}
return ret;
}
int ind[MAXN], rev[MAXN];
pair<int,pii> a[MAXN];
int main() {
int n, u, d, s;
scanf("%d %d %d %d", &n, &u, &d, &s);
a[0] = mp(-1, mp(s, 0));
for (int i = 1; i <= n; i++) {
scanf("%d %d %d", &a[i].fi, &a[i].se.fi, &a[i].se.se);
ind[i] = i;
}
sort(ind, ind + n+1, [](int l, int r) {
return a[l].se.fi < a[r].se.fi;
});
for (int i = 0; i <= n; i++) {
dp[i] = -INF;
rev[ind[i]] = i;
}
sort(ind, ind + n+1, [](int l, int r) {
if (a[l].fi == a[r].fi)
return a[l].se.fi < a[r].se.fi;
return a[l].fi < a[r].fi;
});
dp[0] = 0;
dp[n+1] = 0;
for (int i = 0; i < MAXN; i++)
BIT1[i] = BIT2[i] = -INF;
upd1(rev[0], dp[0] - u * s);
upd2(rev[0], dp[0] + d * s);
for (int i = 1; i <= n; i++) {
int j = i;
for (; j <= n && a[ind[j]].fi == a[ind[i]].fi; j++) {
int cur = ind[j];
int v1 = qry1(rev[cur]); // upwards
if (v1 != -INF) dp[j] = max(dp[j], a[cur].se.se + v1 + u * a[cur].se.fi);
int v2 = qry2(rev[cur]); // downwards
if (v2 != -INF) dp[j] = max(dp[j], a[cur].se.se + v2 - d * a[cur].se.fi);
}
int hi = j - 1;
vi pre, suf;
// upwards
int tmp = -INF;
int sm = 0;
for (j = i; j <= hi; j++) {
int cur = ind[j];
sm += a[cur].se.se;
if (tmp == -INF)
pre.pb(-INF);
else
pre.pb(tmp + sm - d * a[cur].se.fi);
if (dp[j] != -INF)
tmp = max(tmp, dp[j] + d * a[cur].se.fi - sm);
}
tmp = -INF;
sm = 0;
// downwards
for (j = hi; j >= i; j--) {
int cur = ind[j];
sm += a[cur].se.se;
if (tmp == -INF)
suf.pb(-INF);
else
suf.pb(tmp + sm + u * a[cur].se.fi);
if (dp[j] != -INF)
tmp = max(tmp, dp[j] - u * a[cur].se.fi - sm);
}
reverse(suf.begin(), suf.end());
for (j = i; j <= hi; j++) {
int cur = ind[j];
//cerr << dp[j] << " " << pre[j-i] << " " << suf[j-i] << "\n";
dp[j] = max(dp[j], max(pre[j-i], suf[j-i]));
if (dp[j] != -INF) {
upd1(rev[cur], dp[j] - u * a[cur].se.fi);
upd2(rev[cur], dp[j] + d * a[cur].se.fi);
}
}
i = hi;
}
int v1 = qry1(rev[0]); // upwards
if (v1 != -INF)
dp[n+1] = max(dp[n+1], v1 + u * s);
int v2 = qry2(rev[0]); // downwards
if (v2 != -INF)
dp[n+1] = max(dp[n+1], v2 - d * s);
printf("%d\n", dp[n+1]);
return 0;
}
Compilation message
salesman.cpp: In function 'int main()':
salesman.cpp:82:7: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%d %d %d %d", &n, &u, &d, &s);
~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
salesman.cpp:85:8: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%d %d %d", &a[i].fi, &a[i].se.fi, &a[i].se.se);
~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
5 ms |
4352 KB |
Output is correct |
| 2 |
Correct |
5 ms |
4224 KB |
Output is correct |
| 3 |
Correct |
5 ms |
4224 KB |
Output is correct |
| 4 |
Correct |
6 ms |
4352 KB |
Output is correct |
| 5 |
Correct |
9 ms |
4352 KB |
Output is correct |
| 6 |
Correct |
19 ms |
4736 KB |
Output is correct |
| 7 |
Correct |
44 ms |
5500 KB |
Output is correct |
| 8 |
Correct |
89 ms |
6624 KB |
Output is correct |
| 9 |
Correct |
130 ms |
7800 KB |
Output is correct |
| 10 |
Correct |
290 ms |
11480 KB |
Output is correct |
| 11 |
Correct |
313 ms |
11256 KB |
Output is correct |
| 12 |
Correct |
468 ms |
13688 KB |
Output is correct |
| 13 |
Correct |
436 ms |
13752 KB |
Output is correct |
| 14 |
Correct |
564 ms |
16120 KB |
Output is correct |
| 15 |
Correct |
539 ms |
16096 KB |
Output is correct |
| 16 |
Correct |
578 ms |
16120 KB |
Output is correct |
| 17 |
Correct |
6 ms |
4224 KB |
Output is correct |
| 18 |
Correct |
5 ms |
4352 KB |
Output is correct |
| 19 |
Correct |
6 ms |
4352 KB |
Output is correct |
| 20 |
Correct |
7 ms |
4352 KB |
Output is correct |
| 21 |
Correct |
6 ms |
4352 KB |
Output is correct |
| 22 |
Correct |
8 ms |
4352 KB |
Output is correct |
| 23 |
Correct |
8 ms |
4352 KB |
Output is correct |
| 24 |
Correct |
8 ms |
4480 KB |
Output is correct |
| 25 |
Correct |
73 ms |
6696 KB |
Output is correct |
| 26 |
Correct |
159 ms |
9332 KB |
Output is correct |
| 27 |
Correct |
292 ms |
13628 KB |
Output is correct |
| 28 |
Correct |
314 ms |
13076 KB |
Output is correct |
| 29 |
Correct |
446 ms |
16192 KB |
Output is correct |
| 30 |
Correct |
471 ms |
17164 KB |
Output is correct |
