Submission #542705

#	Time	Username	Problem	Language	Result	Execution time	Memory
542705		cig32	Distributing Candies (IOI21_candies)	C++17	Compilation error	0 ms	0 KiB

candies

#include "bits/stdc++.h"
using namespace std;
#ifdef ONLINE_JUDGE
#include "candies.h"
#endif
#include <vector>
#ifndef ONLINE_JUDGE
#include <vector>

std::vector<int> distribute_candies(std::vector<int> c, std::vector<int> l,
                                    std::vector<int> r, std::vector<int> v);
#endif

const int MAXN = 2e5 + 10;
const int MOD = 1e9 + 7;
#define ll long long
mt19937_64 rng((int)std::chrono::steady_clock::now().time_since_epoch().count());
int rnd(int x, int y) {
  int u = uniform_int_distribution<int>(x, y)(rng); return u;
}

ll result(ll cap, vector<int> v) {
  int q = v.size();
  ll cur = 0;
  for(int i=0; i<q; i++) {
    cur += v[i];
    cur = min(cur, cap);
    cur = max(cur, 0ll);
  }
  return cur;
}
std::vector<int> distribute_candies(std::vector<int> c, std::vector<int> l, std::vector<int> r, std::vector<int> v) {
  int n = c.size();
  int q = v.size();
  std::vector<int> s(n);
  ll lb = 0, rb = 2e14;
  while(lb < rb) {
    ll mid = (lb + rb + 1) >> 1;
    ll res = result(mid, v);
    if(res == 0) lb = mid;
    else rb = mid - 1;
  }
  ll lz = lb; // last zero
  lb = lz + 1, rb = 2e14;
  while(lb < rb) {
    ll mid = (lb + rb) >> 1;
    ll dis = result(mid, v);
    ll prv = result(mid - 1, v);
    if(dis == prv) rb = mid;
    else lb = mid + 1;
  }
  ll one = result(lb, v);
  ll two = result(lb - 1, v);
  ll se; // start equal
  if(one == two) se = lb;
  else se = 2e14;
  for(int i=0; i<n; i++) {
    if(c[i] <= lz) s[i] = 0;
    else {
      if(c[i] >= se) c[i] = se - 1;
      s[i] = c[i] - lz;
    }
  }
  return s;
}

#ifndef ONLINE_JUDGE
int main() {
    int n;
    assert(1 == scanf("%d", &n));
    std::vector<int> c(n);
    for(int i = 0; i < n; ++i) {
        assert(scanf("%d", &c[i]) == 1);
    }

    int q;
    assert(1 == scanf("%d", &q));
    std::vector<int> l(q), r(q), v(q);
    for(int i = 0; i < q; ++i) {
        assert(scanf("%d %d %d", &l[i], &r[i], &v[i]) == 3);
    }

    std::vector<int> ans = distribute_candies(c, l, r, v);

    for(int i = 0; i < n; ++i) {
        if (i > 0) {
            printf(" ");
        }
        printf("%d", ans[i]);
    }
    printf("\n");
    fclose(stdout);
    return 0;
}
#endif
/*
3
10 15 13
2
0 2 20
0 2 -11
*/

Compilation message (stderr)

candies.cpp: In function 'std::vector<int> distribute_candies(std::vector<int>, std::vector<int>, std::vector<int>, std::vector<int>)':
candies.cpp:34:7: warning: unused variable 'q' [-Wunused-variable]
   34 |   int q = v.size();
      |       ^
/usr/bin/ld: /tmp/ccVhPyqc.o: in function `main':
grader.cpp:(.text.startup+0x0): multiple definition of `main'; /tmp/ccXUtR3b.o:candies.cpp:(.text.startup+0x0): first defined here
collect2: error: ld returned 1 exit status