백준 소스코드 [C++] 14897 서로 다른 수와 쿼리 1

www.acmicpc.net/problem/14897

14897번: 서로 다른 수와 쿼리 1

백준 소스코드 [C++] 14897 서로 다른 수와 쿼리 1

#include <iostream>
#include <algorithm>
#include <queue>
#include <string.h>
#include <limits.h>
#include <vector>
#include <math.h>
#include <stack>
#include <bitset>
#include <string>
#define all(v) v.begin(), v.end()
#define pii pair<int,int>
#define make_unique(v) v.erase(unique(v.begin(), v.end()), v.end())
typedef long long ll;
using namespace std;
int n, m,sqrtn;

int cnt[1000001];
int sum=0;
void add(int a) {
	if (cnt[a] == 0) sum++;
	cnt[a]++;
}

void sub(int a) {
	cnt[a]--;
	if (cnt[a] == 0) sum--;
}

struct query {
	int idx, i, j;
};
bool compare(query& x,query& y) {
	if (x.i / sqrtn != y.i / sqrtn) return x.i / sqrtn < y.i / sqrtn;
	return x.j < y.j;
}

int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	cin >> n;
	sqrtn = (int)sqrt(n);
	vector<int> arr(n);
	vector<int> brr(n);
	for (int i = 0;i < n;i++)cin >> arr[i];
	brr = arr;
	sort(all(brr));
	make_unique(brr);
	for (int i = 0;i < n;i++)arr[i]=lower_bound(all(brr),arr[i])-brr.begin();

	int x, y;
	cin >> m;
	vector<query> q(m);
	vector<int>res(m);
	for (int a = 0;a < m;a++) {
		cin >> x >> y;
		q[a] = { a,x - 1,y - 1 };
	}

	sort(q.begin(), q.end(), compare);

	int s = q[0].i, e = q[0].j;
	for (int a = s;a <= e;a++) add(arr[a]);
	res[q[0].idx] = sum;

	for (int a = 1;a < m;a++) {
		while (q[a].i < s)add(arr[--s]);
		while (q[a].j > e)add(arr[++e]);
		while (q[a].i > s)sub(arr[s++]);
		while (q[a].j < e)sub(arr[e--]);
		res[q[a].idx] = sum;
	}
	for (int i = 0;i < m;i++)  cout << res[i] << '\n';

	return 0;
}