Submission #23091

#	Time	Username	Problem	Language	Result	Execution time	Memory
23091		model_code	Poklon (COCI17_poklon)	C++11	140 / 140	4179 ms	263476 KiB

poklon

#include <cstdio>
#include <string>
#include <vector>
#include <map>
#include <cstdlib>
#include <algorithm>
#include <cstring>

using namespace std;

typedef long long llint;
typedef pair <int, int> pii;

const int MAXN = 1 << 20;

int n, q;

vector <pii> T[MAXN];
int a[MAXN];
int ls[MAXN], rs[MAXN];
int left[MAXN];
int right[MAXN];
int ans[MAXN];

void update (int x, int l, int r, int L, int v) {
  if (L > r || L < l) return;
  T[x].push_back({rs[v], v});
  int mid = (l + r) / 2;
  if (l == r) return;
  update(x * 2, l, mid, L, v);
  update(x * 2 + 1, mid+1, r, L, v);
}

void query_update (int x, int l, int r, int L, int R, int p1, int p2) {
  if (L > r || R < l) return;
  if (l >= L && r <= R) {
    T[x].push_back({p1, -1});
    T[x].push_back({p2, -2});
    return;
  }
  int mid = (l + r) / 2;
  query_update(x * 2, l, mid, L, R, p1, p2);
  query_update(x * 2 + 1, mid+1, r, L, R, p1, p2);
}

void solve (int x, int l, int r) {
  sort(T[x].begin(), T[x].end());
  int cnt = 0;
  for (auto v: T[x]) {
    if (v.second == -1) ++cnt;
    else
    if (v.second == -2) --cnt;
    else ans[v.second] += cnt;
  }
  if (l == r) return;
  int mid = (l+r) / 2;
  solve(x*2, l, mid);
  solve(x*2+1, mid+1, r);
}

int main (void){
  scanf("%d%d", &n, &q);
  for (int i = 0; i < n; ++i) scanf("%d", &a[i]);
  for (int i = 0; i < q; ++i) {
    scanf("%d%d", &ls[i], &rs[i]);
    --ls[i]; --rs[i];
    update(1, 0, n-1, ls[i], i);
  }
  map <int, int> M;
  for (int i = 0; i < n; ++i) {
    int pos = -1;
    if (M.count(a[i])) pos = M[a[i]];
    left[i] = pos;
    M[a[i]] = i;
  }
  M.clear();
  for (int i = n-1; i >= 0; --i) {
    int pos = n;
    if (M.count(a[i])) pos = M[a[i]];
    right[i] = pos;
    M[a[i]] = i;
  }
  right[n] = n;

  for (int i = 0; i < n; ++i) {
    int rpos = right[i];
    if (rpos == n) continue;
    query_update(1, 0, n-1, left[i] + 1, i, rpos, right[rpos]);
  }

  solve(1, 0, n-1);
  for (int i = 0; i < q; ++i)
    printf("%d\n", ans[i]);

  return 0;
}

Compilation message (stderr)

poklon.cpp: In function 'int main()':
poklon.cpp:62:24: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   scanf("%d%d", &n, &q);
                        ^
poklon.cpp:63:49: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   for (int i = 0; i < n; ++i) scanf("%d", &a[i]);
                                                 ^
poklon.cpp:65:34: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%d%d", &ls[i], &rs[i]);
                                  ^

• Subtask 1: 140 / 140