forked from Priyansh19077/CP-Templates
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathDistinct_in_range_queries_nlogn.cpp
123 lines (123 loc) · 4.17 KB
/
Distinct_in_range_queries_nlogn.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
/* Priyansh Agarwal*/
#include<bits/stdc++.h>
#include<ext/pb_ds/assoc_container.hpp>
#include<ext/pb_ds/tree_policy.hpp>
#include<algorithm>
#include<unordered_map>
#include<vector>
#include<unordered_set>
#include<set>
#include<map>
#include<queue>
#include<stack>
#include<map>
#include<chrono>
using namespace std;
using namespace __gnu_pbds;
using namespace chrono;
#define fastio() ios_base::sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL)
#define MOD 1000000007
#define MOD1 998244353
#define nline "\n"
#define pb push_back
#define mp make_pair
#define ff first
#define ss second
#define PI 3.141592653589793238462
#define debug(x) cout << #x << " " << x <<endl;
#define set_bits __builtin_popcount
typedef long long ll;
typedef unsigned long long ull;
typedef long double lld;
typedef tree<int, null_type, less<int>, rb_tree_tag, tree_order_statistics_node_update > pbds;
/*---------------------------------------------------------------------------------------------------------------------------*/
ll gcd(ll a, ll b) {if (b > a) {return gcd(b, a);} if (b == 0) {return a;} return gcd(b, a % b);}
ll expo(ll a, ll b, ll mod) {ll res = 1; while (b > 0) {if (b & 1)res = (res * a) % mod; a = (a * a) % mod; b = b >> 1;} return res;}
void extendgcd(ll a, ll b, ll*v) {if (b == 0) {v[0] = 1; v[1] = 0; v[2] = a; return ;} extendgcd(b, a % b, v); ll x = v[1]; v[1] = v[0] - v[1] * (a / b); v[0] = x; return;} //pass an arry of size 3
ll mminv(ll a, ll b) {ll arr[3]; extendgcd(a, b, arr); return arr[0];} //for non prime b
ll mminvprime(ll a, ll b) {return expo(a, b - 2, b);}
bool revsort(ll a, ll b) {return a > b;}
void swap(int &x, int &y) {int temp = x; x = y; y = temp;}
ll combination(ll n, ll r, ll m, ll* fact) {ll val1 = fact[n]; ll val2 = mminvprime(fact[r], m); ll val3 = mminvprime(fact[n - r], m); return ((val1 * val2) % m * val3) % m;}
void google(int t) {cout << "Case #" << t << ": ";}
vector<int> sieve(int n) {int*arr = new int[n + 1](); vector<int> vect; for (int i = 2; i <= n; i++)if (arr[i] == 0) {vect.push_back(i); for (int j = 2 * i; j <= n; j += i)arr[j] = 1;} return vect;}
/*--------------------------------------------------------------------------------------------------------------------------*/
bool compare(pair<pair<int, int>, int> p1, pair<pair<int, int>, int> p2)
{
return p1.ff.ss < p2.ff.ss;
}
void update(int start, int *bit, int n, int value)
{
for (; start <= n; start += start & (-start))
bit[start] += value;
}
int query(int start, int *bit)
{
int sum = 0;
for (; start > 0; start -= start & (-start))
sum += bit[start];
return sum;
}
int main()
{
fastio();
// #ifndef ONLINE_JUDGE
// freopen("Input.txt", "r", stdin);
// freopen("Output.txt", "w", stdout);
// freopen("Error.txt", "w", stderr);
// #endif
auto start1 = high_resolution_clock::now();
int n, q;
cin >> n >> q;
int arr[n];
int *bit = new int[n + 5]();
int *freq = new int[n + 5];
for (int i = 0; i <= n; i++)
freq[i] = -1;
int number = 1;
map<int, int> coordinate_compression;
for (int i = 0; i < n; i++)
{
cin >> arr[i];
if (coordinate_compression.find(arr[i]) == coordinate_compression.end())
{
coordinate_compression[arr[i]] = number;
arr[i] = number;
number++;
}
else
arr[i] = coordinate_compression[arr[i]];
}
vector<pair<pair<int, int>, int>> ans(q);
for (int i = 0; i < q; i ++)
{
int a, b;
cin >> a >> b;
ans[i] = {{a, b}, i};
}
sort(ans.begin(), ans.end(), compare);
int num = 0;
int *ans1 = new int[q];
for (int i = 0; i < n; i++)
{
if (freq[arr[i]] != -1)
update(freq[arr[i]] + 1, bit, n + 4, -1);
freq[arr[i]] = i;
update(i + 1, bit, n + 4, 1);
while (num < q && ans[num].ff.ss == i + 1)
{
int index = ans[num].ss;
int value = query(ans[num].ff.ss, bit) - (ans[num].ff.ff > 1 ? query(ans[num].ff.ff - 1, bit) : 0);
ans1[ans[num].ss] = value;
num++;
}
}
for (int i = 0; i < q; i++)
cout << ans1[i] << "\n";
auto stop1 = high_resolution_clock::now();
auto duration = duration_cast<microseconds>(stop1 - start1);
// #ifndef ONLINE_JUDGE
// cerr << "Time: " << duration.count() / 1000.0 << endl;
// #endif
return 0;
}