Submission #1086543

#	Time	Username	Problem	Language	Result	Execution time	Memory
1086543		mingga	Infinite Race (EGOI24_infiniterace2)	C++17	100 / 100	62 ms	12516 KiB

Main

#include "bits/stdc++.h"

using namespace std;

#define ln "\n"
#define dbg(x) cout << #x << " = " << x << ln
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define inf 2e18
#define fast_cin()                  \
  ios_base::sync_with_stdio(false); \
  cin.tie(NULL)
#define out(file) freopen(file, "w", stdout)
#define in(file) freopen(file, "r", stdin)
#define all(x) (x).begin(), (x).end()
#define sz(x) ((int)(x).size())
#define int long long
int MOD = 1e9 + 7;
const int N = 2e5 + 7;
int n, q;
int x[N];

void sub1() {
  int cnt = 0;
  int cur = 0;
  while(q--) {
    int u; cin >> u;
    if(u < 0) {
      cur = 0;
    } else {
      if(cur == 1) cnt++;
      else cur = 1;
    }
  }
  cout << cnt;
}

void sub2() {
  vector<int> state(n+1, 0);
  int ans = 0;
  int cur = 0;
  set<int> s;
  for(int i = 1; i <= q; i++) {
    // cout << x[i] << ' ' << *s.lower_bound(x[i])
    if(*s.lower_bound(x[i]) == x[i]) {
      ans++;
      s.clear();
    }
      s.insert(x[i]);
    // cout << ans << ' ';
  }
  cout << ans;
}

void full() {
  set<int> s;
  int ans = 0;
  for(int i = 1; i <= q; i++) {
    cin >> x[i];
    if(x[i] > 0) {
      if(*s.lower_bound(x[i]) == x[i]) {
        ans++;
        s.clear();
      }
      s.insert(x[i]);
    } else {
      x[i] = -x[i];
      auto cur = s.lower_bound(x[i]);
      if(*cur == x[i]) {
        s.erase(*cur);
      }
    }
    // cout << ans << ' ';
  }
  cout << ans;
}

signed main() {
    fast_cin();
    cin >> n >> q;
    // if(n == 2) sub1();
    // else {
    //   bool ok = 1;
    //   for(int i = 1; i <= q; i++) {
    //     cin >> x[i];
    //     if(x[i] < 0) ok = 0;
    //   }
    //   if(ok) sub2();
    // }
    full();
    cerr << "\nTime: " << clock() * 1000 / CLOCKS_PER_SEC;
}

Compilation message (stderr)

Main.cpp: In function 'void sub2()':
Main.cpp:43:7: warning: unused variable 'cur' [-Wunused-variable]
   43 |   int cur = 0;
      |       ^~~

• Subtask 1: 29 / 29
• Subtask 2: 34 / 34
• Subtask 3: 22 / 22
• Subtask 4: 15 / 15