Cod sursa(job #2475040)

Utilizator ApostolIlieDanielApostol Daniel ApostolIlieDaniel Data 16 octombrie 2019 09:09:22
Problema Cuburi2 Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.61 kb
#include <bits/stdc++.h>

using namespace std;
//#define fin stdin
//#define fout stdout
FILE *fin = fopen ("cuburi2.in", "r"), *fout = fopen ("cuburi2.out", "w");

const int MAXN = 320000;
const long long INF = 1e18;

int n;

long long st[MAXN + 1], dr[MAXN + 1], sl[MAXN + 1];

int h[MAXN + 1];
int good;
void preprocesare () {
  int i;
  for (i = 1; i <= n; i++)
    sl[i] = sl[i - 1] + h[i];
  for (i = 1; i <= n; i++)
    st[i] = st[i - 1] + sl[i - 1];
  for (i = n; i > 0; i--)
    dr[i] = dr[i + 1] + (sl[n] - sl[i]);
}

inline pair <long long, long long> sol (int copst, int l, int r) {
  if (copst < l || copst > r)
    return {INF, INF};
  long long valst = st[copst] - st[l] - (copst - l) * (sl[l - 1]);
  long long valdr = dr[copst] - dr[r] - (r - copst) * (sl[n] - sl[r]);
  return {valst, valdr};
}

inline long long solve (int l, int r) {
  long long mn;
  int st = l;
  int dr = r;
  mn = sol (l, l, r).first + sol (l, l, r).second;
  good = l;
  while (st <= dr) {
    int mid = (st + dr) / 2;
    pair <long, long> rez = sol (mid, l, r);
    if (mn > rez.first + rez.second) {
        mn = rez.first + rez.second;
        good = mid;
    }
    if (rez.first < rez.second)
        st = mid + 1;
    else
        dr = mid - 1;
  }
  return mn;
}

int main() {
  int q, l, r;
  fscanf (fin, "%d%d", &n, &q);
  for (int i = 1; i <= n; i++)
    fscanf (fin, "%d", &h[i]);
  preprocesare ();
  while (q--) {
    fscanf (fin, "%d%d", &l, &r);
    long long sol = solve (l, r);
    fprintf (fout, "%d %lld\n", good, sol);
  }
  fclose (fin);
  fclose (fout);
  return 0;
}