From 035ae0db488809f8c6a2d0956614c344ed8e4f32 Mon Sep 17 00:00:00 2001 From: omagdy7 Date: Tue, 7 Mar 2023 16:21:15 +0200 Subject: Solved a dp problem on codeforces --- codeforces/SerejaAndSuffixes/main | Bin 0 -> 38312 bytes codeforces/SerejaAndSuffixes/main.cpp | 107 ++++++++++++++++++++++++++ codeforces/SerejaAndSuffixes/main_input0.txt | 12 +++ codeforces/SerejaAndSuffixes/main_output0.txt | 10 +++ 4 files changed, 129 insertions(+) create mode 100755 codeforces/SerejaAndSuffixes/main create mode 100755 codeforces/SerejaAndSuffixes/main.cpp create mode 100644 codeforces/SerejaAndSuffixes/main_input0.txt create mode 100644 codeforces/SerejaAndSuffixes/main_output0.txt (limited to 'codeforces') diff --git a/codeforces/SerejaAndSuffixes/main b/codeforces/SerejaAndSuffixes/main new file mode 100755 index 0000000..e8e8408 Binary files /dev/null and b/codeforces/SerejaAndSuffixes/main differ diff --git a/codeforces/SerejaAndSuffixes/main.cpp b/codeforces/SerejaAndSuffixes/main.cpp new file mode 100755 index 0000000..c14fd91 --- /dev/null +++ b/codeforces/SerejaAndSuffixes/main.cpp @@ -0,0 +1,107 @@ +#include +using namespace std; + +using ll = long long; +using pi = pair; +using vpi = vector; +using vi = vector; +using vll = vector; +using mpii = map; +using mpll = map; +using db = long double; + +#define pb push_back +#define all(x) (x).begin(), (x).end() +#define rall(x) (x).rbegin(), (x).rend() +#define lb lower_bound +#define ub upper_bound + +const int MOD = (int)1e9 + 7; +const db PI = acos((db)-1); +const int dx[4]{1, 0, -1, 0}; +const int dy[4]{0, 1, 0, -1}; + +//pretty printing +template +void printm(const map &mp) { + cerr << "{" << endl; + for (auto p : mp) { + cerr << " { " << p.first << " : " << p.second << " }\n"; + } + cerr << "}" << endl; +} +template +void printv(const vector &v) { + cerr << "["; + for (int i = 0; i < v.size(); i++) { + if (i == v.size() - 1) { + cerr << v[i]; + } else { + cerr << v[i] << ", "; + } + } + cerr << "]\n"; +} + +template +void printvv(const vector> &v) { + cerr << "[\n"; + for (auto &vec : v) { + cout << " "; + printv(vec); + } + cerr << "]\n"; +} +void print() { + cerr << "\n"; +} + +template +void print(T val, TS... vals) { + cerr << val << " "; + print(vals...); +} + + +/* stuff you should look for: + --------------------------- + * special cases (n=1?) + * int overflow, array bounds + * do smth instead of nothing and stay organized + * WRITE STUFF DOWN + * DON'T GET STUCK ON ONE APPROACH + */ + +void solve() { + int n, m; + cin >> n >> m; + vi v(n); + vi q(m); + set st; + for (auto &x : v) { + cin >> x; + st.insert(x); + } + for (auto &x : q) cin >> x; + vi fq(1e5 + 5); + vi dp(n); + dp[0] = st.size(); + for (int i = 0; i < n; i++) { + fq[v[i]]++; + } + for (int i = 1; i < n; i++) { + fq[v[i - 1]]--; + dp[i] = dp[i - 1] - (fq[v[i - 1]] == 0); + } + for (auto x : q) { + cout << dp[x - 1] << '\n'; + } +} + +int main () { + ios_base::sync_with_stdio(false); + cin.tie(NULL); + solve(); +} + + diff --git a/codeforces/SerejaAndSuffixes/main_input0.txt b/codeforces/SerejaAndSuffixes/main_input0.txt new file mode 100644 index 0000000..c1d3d92 --- /dev/null +++ b/codeforces/SerejaAndSuffixes/main_input0.txt @@ -0,0 +1,12 @@ +10 10 +1 2 3 4 1 2 3 4 100000 99999 +1 +2 +3 +4 +5 +6 +7 +8 +9 +10 diff --git a/codeforces/SerejaAndSuffixes/main_output0.txt b/codeforces/SerejaAndSuffixes/main_output0.txt new file mode 100644 index 0000000..7c6e911 --- /dev/null +++ b/codeforces/SerejaAndSuffixes/main_output0.txt @@ -0,0 +1,10 @@ +6 +6 +6 +6 +6 +5 +4 +3 +2 +1 -- cgit v1.2.3