1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121
| #include <bits/stdc++.h>
#pragma optimization_level 3 #pragma GCC optimize("Ofast,no-stack-protector,unroll-loops,fast-math,O3") #pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx") #pragma GCC optimize("Ofast") #pragma GCC target("avx,avx2,fma") #pragma GCC optimization ("unroll-loops")
using namespace std;
struct PairHash {inline std::size_t operator()(const std::pair<int, int> &v) const { return v.first * 31 + v.second; }};
#define Code ios_base::sync_with_stdio(false); #define By ios::sync_with_stdio(0); #define Sumfi cout.tie(NULL);
using ll = long long; using ld = long double;
const ld PI = 3.14159265358979323846; const ll INF = 1e18; const ld EPS = 1e-9; const ll MAX_N = 505050; const ll mod = 1e9 + 7;
typedef pair<ll, ll> pll; typedef vector<pll> vpll; typedef array<ll,3> all3; typedef array<ll,5> all5; typedef vector<all3> vall3; typedef vector<all5> vall5; typedef vector<ld> vld; typedef vector<ll> vll; typedef vector<vll> vvll; typedef vector<int> vi; typedef vector<string> vs; typedef vector<vs> vvs; typedef unordered_set<ll> usll; typedef unordered_set<pll, PairHash> uspll; typedef unordered_map<ll, ll> umll; typedef unordered_map<pll, ll, PairHash> umpll;
#define rep(i,m,n) for(ll i=m;i<n;i++) #define rrep(i,m,n) for(ll i=n;i>=m;i--) #define all(a) begin(a), end(a) #define rall(a) rbegin(a), rend(a) #define ZERO(a) memset(a,0,sizeof(a)) #define MINUS(a) memset(a,0xff,sizeof(a)) #define sz(x) ll((x).size()) #define pyes cout<<"YES\n"; #define pno cout<<"NO\n"; #define pneg1 cout<<"-1\n"; #define CASE(x) cout<<"Case #"<<x<<": ";
template <typename T> void print(T &&t) { cout << t << "\n"; } template<typename T> void printv(vector<T>v){ll n=v.size();rep(i,0,n)cout<<v[i]<<" ";cout<<"\n";}
ll __gcd(ll x, ll y) { return !y ? x : __gcd(y, x % y); } ll __lcm(ll x, ll y) { return x * y / __gcd(x,y); } ll modpow(ll n, ll x, ll MOD = mod) { if(!x) return 1; ll res = modpow(n,x>>1,MOD); res = (res * res) % MOD; if(x&1) res = (res * n) % MOD; return res; }
ll solve(vvll& A) { ll n = sz(A), m = sz(A[0]); vll dp(m); auto sum = [](pair<ll, vll> p) {return p.first + accumulate(all(p.second),0ll);}; rep(i,0,n) { vll dpp(m); vector<pair<ll,vll>> top3(3); top3[0] = top3[1] = top3[2] = {0,{0,0,0}}; rep(j,0,m) { dpp[j] = dp[j] + A[i][j]; rrep(k,0,2) { top3[k].second[k] = max(top3[k].second[k], A[i][j]); if(k and sum(top3[k]) < sum(top3[k-1]) + A[i][j]) { top3[k] = top3[k-1]; top3[k].second[k] = A[i][j]; } sort(rall(top3[k].second));
dpp[j] = max(dpp[j], sum(top3[k])); }
if(sum(top3[0]) <= dp[j] + A[i][j]) top3[0] = {dp[j], {A[i][j], 0, 0}}; }
swap(dpp, dp); }
return dp[m-1]; }
int main() { Code By Sumfi
cout.precision(12);
ll tc = 1; for (ll i = 1; i <= tc; i++) { ll n,m,k,y,x,v; cin>>n>>m>>k; vvll A(n,vll(m)); rep(i,0,k) { cin>>y>>x>>v; A[y-1][x-1] = v; } print(solve(A)); } return 0; }
|