Problem p02580 (C++) — buggy submission, verdict: Time Limit Exceeded

#pragma target("avx")
#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
typedef long double ld;
typedef pair<ll, ll> P;
typedef vector<ll> V;
typedef unordered_map<ll, ll> U_MAP;
typedef priority_queue<ll> pq;
typedef priority_queue<ll, vector<ll>, greater<ll>> rpq;
constexpr ll INF = 1e9, MOD = 1e9 + 7, ohara = 1e6 + 10;
constexpr ll LINF = 1e18;

#define rep(i, n) for (ll(i) = 0; (i) < (int)(n); (i)++)
#define rrep(i, a, b) for (ll i = (a); i < (b); i++)
#define rrrep(i, a, b) for (ll i = (a); i >= (b); i--)
#define all(v) (v).begin(), (v).end()
#define Size(n) (n).size()
#define Cout(x) cout << (x) << endl
#define doublecout(a) cout << fixed << setprecision(15) << a << endl;
#define fi first
#define se second
#define m_p make_pair
#define p_b push_back
string to_string(string s) { return '"' + s + '"'; }
string to_string(const char *s) { return to_string((string)s); }
string to_string(bool b) { return (b ? "true" : "false"); }
template <typename A, typename B> string to_string(pair<A, B> p) {
  return "(" + to_string(p.first) + ", " + to_string(p.second) + ")";
}
template <typename A> string to_string(A v) {
  bool first = true;
  string res = "{";
  for (const auto &x : v) {
    if (!first) {
      res += ", ";
    }
    first = false;
    res += to_string(x);
  }
  res += "}";
  return res;
}
void debug_out() { cerr << endl; }
template <typename Head, typename... Tail> void debug_out(Head H, Tail... T) {
  cerr << " " << to_string(H);
  debug_out(T...);
}
#define debug(...) cerr << "[" << #__VA_ARGS__ << "]:", debug_out(__VA_ARGS__)

//------ Believe yourself as a genius!!!!!! ------

struct SegmentTree {
  int siz;
  vector<ll> node;
  SegmentTree(int n, ll init) {
    siz = 1;
    while (siz < n)
      siz <<= 1;
    node.assign(2 * siz, init);
  }

  void update(int k, ll x) {
    k += siz - 1;
    node[k] = x;
    while (k > 0) {
      k = (k - 1) / 2;
      node[k] = max(node[2 * k + 1], node[2 * k + 2]);
    }
  }

  ll query(int a, int b, int k = 0, int l = 0, int r = -1) {
    if (r < 0)
      r = siz;
    if (r <= a || b <= l)
      return -LINF;
    if (a <= l && r <= b)
      return node[k];
    ll vl = query(a, b, 2 * k + 1, l, (l + r) / 2);
    ll vr = query(a, b, 2 * k + 2, (l + r) / 2, r);
    return max(vl, vr);
  }
};

int dy[] = {1, 0, -1, 0};
int dx[] = {0, 1, 0, -1};
// int dy[]={-1,0,1,-1,1,-1,0,1};int dx[]={-1,-1,-1,0,0,1,1,1};
string alph("abcdefghijklmnopqrstuvwxyz"), s;
ll n, cnt, ans, a, b, c, d, tmp, m, h, w, x[ohara], y[ohara], sum, k, q;
ll coux[ohara], couy[ohara];
ll nowx[ohara], nowy[ohara];
priority_queue<P> que;
V dat[ohara];

int main(void) {
  cin.tie(0);
  cout.tie(0);
  ios::sync_with_stdio(false);

  cin >> h >> w >> m;
  rep(i, m) cin >> y[i] >> x[i], y[i]--, x[i]--;
  rep(i, m) coux[x[i]]++, couy[y[i]]++;

  rep(i, m) {
    ans = max(ans, coux[x[i]] + couy[y[i]] - 1);
    dat[x[i]].p_b(y[i]);
  }
  SegmentTree tree(h + 2, -LINF);
  rep(i, m) tree.update(y[i], couy[y[i]]);

  rep(kx, w) {
    cnt = coux[kx];
    U_MAP out;
    rep(j, Size(dat[kx])) { tree.update(dat[kx][j], 0); }
    ans = max(ans, cnt + tree.query(0, h));
    rep(j, Size(dat[kx])) { tree.update(dat[kx][j], couy[dat[kx][j]]); }
  }
  Cout(ans);
  return 0;
}

Error: TLE

Fixed submission:

#pragma target("avx")
#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
typedef long double ld;
typedef pair<ll, ll> P;
typedef vector<ll> V;
typedef unordered_map<ll, ll> U_MAP;
typedef priority_queue<ll> pq;
typedef priority_queue<ll, vector<ll>, greater<ll>> rpq;
constexpr ll INF = 1e9, MOD = 1e9 + 7, ohara = 1e6 + 10;
constexpr ll LINF = 1e18;

#define rep(i, n) for (ll(i) = 0; (i) < (int)(n); (i)++)
#define rrep(i, a, b) for (ll i = (a); i < (b); i++)
#define rrrep(i, a, b) for (ll i = (a); i >= (b); i--)
#define all(v) (v).begin(), (v).end()
#define Size(n) (n).size()
#define Cout(x) cout << (x) << endl
#define doublecout(a) cout << fixed << setprecision(15) << a << endl;
#define fi first
#define se second
#define m_p make_pair
#define p_b push_back
string to_string(string s) { return '"' + s + '"'; }
string to_string(const char *s) { return to_string((string)s); }
string to_string(bool b) { return (b ? "true" : "false"); }
template <typename A, typename B> string to_string(pair<A, B> p) {
  return "(" + to_string(p.first) + ", " + to_string(p.second) + ")";
}
template <typename A> string to_string(A v) {
  bool first = true;
  string res = "{";
  for (const auto &x : v) {
    if (!first) {
      res += ", ";
    }
    first = false;
    res += to_string(x);
  }
  res += "}";
  return res;
}
void debug_out() { cerr << endl; }
template <typename Head, typename... Tail> void debug_out(Head H, Tail... T) {
  cerr << " " << to_string(H);
  debug_out(T...);
}
#define debug(...) cerr << "[" << #__VA_ARGS__ << "]:", debug_out(__VA_ARGS__)

//------ Believe yourself as a genius!!!!!! ------

int dy[] = {1, 0, -1, 0};
int dx[] = {0, 1, 0, -1};
// int dy[]={-1,0,1,-1,1,-1,0,1};int dx[]={-1,-1,-1,0,0,1,1,1};
string alph("abcdefghijklmnopqrstuvwxyz"), s;
ll n, cnt, ans, a, b, c, d, tmp, m, h, w, x[ohara], y[ohara], sum, k, q;
ll coux[ohara], couy[ohara];
ll nowx[ohara], nowy[ohara];
priority_queue<P> que;
V dat[ohara];

int main(void) {
  cin.tie(0);
  cout.tie(0);
  ios::sync_with_stdio(false);

  cin >> h >> w >> m;
  rep(i, m) cin >> y[i] >> x[i], y[i]--, x[i]--;
  rep(i, m) coux[x[i]]++, couy[y[i]]++;

  rep(i, m) {
    ans = max(ans, coux[x[i]] + couy[y[i]] - 1);
    dat[x[i]].p_b(y[i]);
  }
  rep(i, m) que.push({couy[y[i]], y[i]});

  rep(kx, w) {
    cnt = coux[kx];
    U_MAP out;
    rep(j, Size(dat[kx])) { out[dat[kx][j]] = 1; }
    priority_queue<P> que2;
    while (!que.empty()) {
      P to = que.top();
      // Cout(to.fi);
      if (out[to.se]) {
        que.pop();
        que2.push(to);
      } else {
        cnt = max(cnt, cnt + to.fi);
        break;
      }
    }
    while (!que2.empty()) {
      P to = que2.top();
      que.push(to);
      que2.pop();
    }
    ans = max(ans, cnt);
  }
  Cout(ans);
  return 0;
}