Submission #837594

#	Time	Username	Problem	Language	Result	Execution time	Memory
837594		erdemkiraz	Aliens (IOI16_aliens)	C++17	4 / 100	10 ms	464 KiB

aliens

#include "aliens.h"
#include <bits/stdc++.h>
 
using namespace std;
 
typedef long long ll;
typedef pair<int, int> ii;
typedef long double ld;
 
const int M = 1e6 + 5;
 
int n, m, k;
int at[M];

vector<int> ats;
vector<ll> a, b;

class line{ public:
    ld m, n;
    int idx;
    line(ld m, ld n, int idx) {
        this -> m = m;
        this -> n = n;
        this -> idx = idx;
    }
    ld intersect(line oth) {
        return (ld) (n - oth.n) / (oth.m - m);
    }
};

class convex{ public:
    int ptr;
    vector < line > v;
    void init() {
        ptr = 0;
        v.clear();
    }
    void add(ld m, ld n, int idx) {
        while(ptr + 1 < v.size()) {
            line l1 = v[v.size() - 2];
            line l2 = v[v.size() - 1];
            line l3 = {m, n, idx};
            if(l1.intersect(l2) < l1.intersect(l3))
                break;
            v.pop_back();
        }
        v.push_back({m, n, idx});
    }
    pair<ld, int> query(ld x) {
        while(ptr + 1 < v.size()) {
            line l1 = v[ptr];
            line l2 = v[ptr + 1];
            if(l1.intersect(l2) > x)
                break;
            ptr++;
        }
        return {v[ptr].m * x + v[ptr].n, v[ptr].idx};
    }
};

vector<int> go;

tuple<ld, int, ll> get(ld discourage) {
  vector<ld> dp(m + 1, 1e30);

  dp[0] = 0;

  vector<ld> cost(m + 1);

  convex trick;
  trick.init();

  for(int i = 1; i <= m; i++) {
    ll len = max(0LL, b[i - 1] - a[i]);
    cost[i] = dp[i - 1] + a[i] * a[i] - len * len;

    trick.add(- 2 * a[i], cost[i], i);

    auto [res, idx] = trick.query(b[i]);

    dp[i] = res + b[i] * b[i] + discourage;
    go[i] = idx;
  }

  int j = m, cnt = 0;
  ll ans = 0;

  while(j > 0) {
    cnt++;

    int i = go[j];

    ll len = b[j] - a[i];
    ll prev = max(0LL, b[i - 1] - a[i]);
    // printf("i = %d j = %d\n", i, j);
    ans += len * len - prev * prev;

    j = i - 1;
  }

  // printf("discourage = %.3lf cnt = %d dp = %.3lf real = %lld\n", discourage, cnt, dp[m], ans);

  ans = dp[m] - cnt * discourage + 0.5;

  return {dp[m], cnt, ans};
}
 
ll take_photos(int nn, int mm, int kk, vector<int> rr, vector<int> cc) {
  n = nn;
  m = mm;
  k = kk;
 
  for(int i = 0; i < n; i++) {
    if(rr[i] > cc[i]) {
      swap(rr[i], cc[i]);
    }
    at[rr[i] + 1] = max(at[rr[i] + 1], cc[i] + 2);
  }

  a.push_back(0);
  b.push_back(0);

  for(int i = 1; i <= m + 1; i++) {
    if(at[i]) {
      if(!ats.empty() and ats.back() >= at[i]) {
        continue;
      }
      ats.push_back(at[i]);

      a.push_back(i);
      b.push_back(at[i]);
    }
  }

  m = (int) a.size() - 1;

  k = min(k, m);

  if(k == 1 or m == 1) {
    return (b.back() - a[1]) * (b.back() - a[1]);
  }

  go.resize(m + 1);

  ld l = 0, r = 1e13;

  ll ans = 1e18;

  for(int it = 0; it < 500; it++) {
    ld m = (l + r) / 2;

    auto [_, used, res] = get(m);

    if(used > k) {
      l = m;
    }
    else {
      ans = min(ans, res);
      r = m;
    }
  }

  return ans;
}

Compilation message (stderr)

aliens.cpp: In member function 'void convex::add(ld, ld, int)':
aliens.cpp:39:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<line>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   39 |         while(ptr + 1 < v.size()) {
      |               ~~~~~~~~^~~~~~~~~~
aliens.cpp: In member function 'std::pair<long double, int> convex::query(ld)':
aliens.cpp:50:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<line>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   50 |         while(ptr + 1 < v.size()) {
      |               ~~~~~~~~^~~~~~~~~~

• Subtask 1: 4 / 4
• Subtask 2: 0 / 12
• Subtask 3: 0 / 9
• Subtask 4: 0 / 16
• Subtask 5: 0 / 19
• Subtask 6: 0 / 40