Segcolt/CSES-solutions
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add a lot more of my solutions.

Browse Source 
I don't remember which ones doesn't pass though
This commit is contained in:
Segcolt
2025-09-12 14:51:10 -03:00
parent 2a94def2d6
commit 8b03716512
114 changed files with 14012 additions and 95 deletions
Download Patch File Download Diff File Expand all files Collapse all files

256
.gdb_history Normal file
View File

@@ -0,0 +1,256 @@
p b
next
run
next
p pref[i - a]
next
next
p vaar
p var
p pref[i]
p pref[i] - *var.begin()
next
break 94
run
next
p i - b + 1
p i - b + 1
p i
break 94
run
run
next
p i - b + 1
p a
p b
p i - b + 1
p i
p i - b - 1
break 94
run
next
next
p ans
p var
p pref[i - a]
break 94
run
next
p anms
p ans
next
p i - a
p pref[i - a]
p i
next
p ansd
p ans
next
p var
break 127
ru
p log
next
p log
p l
p r
p r - (1 << log)
sparse
p sparse
break 128
run
p sparse
continue
run
p sparse[l][log]
p l
p r
break 128
run
break 154
run
p seg[0 + fds.size()]
p segl[0 + fds.size()]
p segl[1 + fds.size()]
p segl[1 + fds.size()]
p segl[2 + fds.size()]
p segl[3 + fds.size()]
p segl[4 + fds.size()]
p segl[5 + fds.size()]
next
p i
p fds[i]
p sparsequery(i, i)
p ans
p i
next
p segr[8 + fds.size()]
p segr[7 + fds.size()]
p segr[8 + fds.size()]
p segr[9 + fds.size()]
break 178
ru
p segr
p segr[5 + fds.size()]
break 138
run
nextr
next
pi
p i
p i
p i
continue
necxt
nexty
next
continue
next
continue
next
next
p seg[6 + fds.size()]
break 139
run
continue
next
p ans
break 126
run
p count
p n
break 109
run
next
p ans
p count
break 90
run
next
p sum
next
p i
next
break 110
run
p sum
p count
break 86
run
break 87
run
next
next
p tmp
next
next
p i
p s.top().second
p
p s
next
p s
p tmp
next
p s
p i
p fds[i]
next
break 135
run
next
next
step
next
break 157
run
next
p pdis[i
p pdis[i
break 120
run
p parent
break 121
run
p reachable
break 121
run
p reachable
break 121
ru
run
p reachable
break 94
run
next
break 25
break 120
run
next
p j
next
p res
break 120
run
ruk
run
next
p i
next
p ans
next
p i
p res
break 183
run
1 2
1 3
p loop
p cyclesize
groupsize
p cyclepos
p groupsize
run
p loop
p groupsize
p groups
p group
p cyclechoice
next
next
p choice
next
p pos[a]
p ((cyclepos[b] - cyclepos[a]) % groupsize[g] + groupsize[g]) % groupsize[g]
p b
p ((cyclepos[b] - cyclepos[choice]) % groupsize[g] + groupsize[g]) % groupsize[g]
next
run
next
next
p j
run
run
continue
next
next
p ff
p F
step
next
p ff
step
break 1154
break 115
continue
p f
next
p f
p e.cap - e.flow
p e.cap
p e.flow
p f
p e.cap - e.flow
continue
p v
p f

105
CSES Problem Set/Advertisement.cpp Normal file
View File

@@ -0,0 +1,105 @@
/* Problem URL: https://cses.fi/problemset/task/1142/ */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
vl fds(n);
cin >> fds;
ll ans = 0;
stack<pair<ll, int>> s;
rep(i, n) {
rmax(ans, fds[i]);
int tmp = i;
while (!s.empty() && s.top().first >= fds[i]) {
rmax(ans, s.top().first * (i - s.top().second));
rmax(ans, fds[i] * (i - s.top().second + 1));
tmp = s.top().second;
s.pop();
}
s.emplace(fds[i], tmp);
}
while (!s.empty()) {
rmax(ans, s.top().first * (n - s.top().second));
s.pop();
}
cout << ans << '\n';
}

124
CSES Problem Set/Apartments.cpp Normal file
View File

@@ -0,0 +1,124 @@
/* Problem URL: https://cses.fi/problemset/task/1084 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (size_t i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (size_t i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
ll n, m, k;
cin >> n >> m >> k;
vl a(n);
cin >> a;
vl b(m);
cin >> b;
sortv(a);
sortv(b);
int i = 0;
int ans = 0;
repv(now, b) {
int low = i;
int high = a.size() - 1;
int choice = -1;
while (low <= high) {
int mid = (low + high) / 2;
if (a[mid] + k < now) {
low = mid + 1;
continue;
}
if (a[mid] - k > now) {
high = mid - 1;
continue;
}
choice = mid;
high = mid - 1;
}
if (choice == -1) {
continue;
}
i = choice + 1;
ans++;
}
cout << ans << '\n';
}

101
CSES Problem Set/Apple Division.cpp Normal file
View File

@@ -0,0 +1,101 @@
/* Problem URL: https://cses.fi/problemset/task/1623 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
ll ans = INT64_MAX >> 1;
int n;
cin >> n;
vl fds(n);
cin >> fds;
rep(i, (1 << n)) {
ll g1 = 0;
ll g2 = 0;
rep(j, n) {
if ((i >> j) & 1) {
g1 += fds[j];
} else {
g2 += fds[j];
}
}
rmin(ans, abs(g1 - g2));
}
cout << ans << '\n';
}

110
CSES Problem Set/Array Division.cpp Normal file
View File

@@ -0,0 +1,110 @@
/* Problem URL: https://cses.fi/problemset/task/1085 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, k;
cin >> n >> k;
vl fds(n);
cin >> fds;
ll low = *max_element(all(fds));
ll high = INT64_MAX >> 1;
ll ans = INT64_MAX >> 1;
while (low <= high) {
ll mid = (low + high) / 2;
int tmp = 1;
ll total = 0;
repv(i, fds) {
if (total + i > mid) {
total = 0;
tmp++;
}
total += i;
}
if (tmp > k) {
low = mid + 1;
continue;
}
ans = mid;
high = mid - 1;
}
cout << ans << '\n';
}

101
CSES Problem Set/Bit Strings.cpp Normal file
View File

@@ -0,0 +1,101 @@
/* Problem URL: https://cses.fi/problemset/task/1617 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
ll mod = 1e9 + 7;
ll fastpow(ll v, ll p)
{
ll ans = 1;
ll now = v;
rep(i, 20) {
if (p & (1 << i)) {
ans *= now;
ans %= mod;
}
now *= now;
now %= mod;
}
return ans;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
cout << fastpow(2, n) << '\n';
}

132
CSES Problem Set/Building Roads.cpp Normal file
View File

@@ -0,0 +1,132 @@
/* Problem URL: https://cses.fi/problemset/task/1666 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (size_t i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (size_t i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m;
cin >> n >> m;
vvi graph(n);
while (m--) {
int a, b;
cin >> a >> b;
a--, b--;
graph[a].push_back(b);
graph[b].push_back(a);
}
vi group(n);
int now = 1;
V<bool> vis(n, false);
function<void(int, int)> dfs = [&](int i, int g){
vis[i] = true;
group[i] = g;
for (auto j : graph[i]) {
if (vis[j]) {
continue;
}
dfs(j, g);
}
};
rep(i, n) {
if (!vis[i]) {
dfs(i, now);
now++;
}
}
V<pair<int, int>> ans;
now = 1;
int node = 0;
nrep(i, 1, n) {
if (group[i] > now) {
ans.emplace_back(node + 1, i + 1);
now = group[i];
node = i;
}
}
cout << ans.size() << '\n';
for (auto [i, j] : ans) {
cout << i << ' ' << j << '\n';
}
}

128
CSES Problem Set/Building Teams.cpp Normal file
View File

@@ -0,0 +1,128 @@
/* Problem URL: https://cses.fi/problemset/task/1668 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m;
cin >> n >> m;
vvi graph(n);
while (m--) {
int a, b;
cin >> a >> b;
a--, b--;
graph[a].push_back(b);
graph[b].push_back(a);
}
V<bool> group(n);
V<bool> vis(n);
function<bool(int, bool)> dfs = [&](int i, bool g) {
vis[i] = true;
group[i] = g;
repv(j, graph[i]) {
if (vis[j]) {
if (group[j] == g) {
return false;
}
continue;
}
if (!dfs(j, g ^ 1)) {
return false;
}
}
return true;
};
rep(i, n) {
if (vis[i]) {
continue;
}
if (!dfs(i, false)) {
cout << "IMPOSSIBLE\n";
return 0;
}
}
rep(i, n) {
cout << group[i] + 1 << ' ';
}
cout << '\n';
}

109
CSES Problem Set/Chessboard and Queens.cpp Normal file
View File

@@ -0,0 +1,109 @@
/* Problem URL: https://cses.fi/problemset/task/1624 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
V<string> fds(8);
cin >> fds;
vi perm = {0, 1, 2, 3, 4, 5, 6, 7};
int ans = 0;
do {
auto valid = [&](){
rep(i, 8) {
if (fds[i][perm[i]] == '*') {
return false;
}
rep(j, 8) {
if (i == j) {
continue;
}
if (abs(i - j) == abs(perm[i] - perm[j])) {
return false;
}
}
}
return true;
};
if (valid()) {
ans++;
}
} while (next_permutation(all(perm)));
cout << ans << '\n';
}

185
CSES Problem Set/Coin Collector.cpp Normal file
View File

@@ -0,0 +1,185 @@
/* Problem URL: https://cses.fi/problemset/task/1686 */
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;
template <class T, class comp = less<>>
using ordered_set = tree<T, null_type , comp , rb_tree_tag , tree_order_statistics_node_update>;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m;
cin >> n >> m;
vl val(n);
cin >> val;
vvi graph(n);
vvi inv(n);
while (m--) {
int a, b;
cin >> a >> b;
a--, b--;
graph[a].push_back(b);
inv[b].push_back(a);
}
vi order;
V<bool> vis(n);
function<void(int)> preorder = [&](int i) {
vis[i] = true;
for (auto j : graph[i]) {
if (vis[j]) {
continue;
}
preorder(j);
}
order.push_back(i);
};
rep(i, n) {
if (vis[i]) {
continue;
}
preorder(i);
}
reverse(all(order));
fill(all(vis), false);
vi group(n);
vl costs;
V<set<int>> cond_graph;
int g = 0;
function<void(int)> scc = [&](int i) {
vis[i] = true;
group[i] = g;
costs[g] += val[i];
for (auto j : inv[i]) {
if (vis[j]) {
if (group[j] != group[i]) {
cond_graph[group[j]].insert(group[i]);
}
continue;
}
scc(j);
}
};
repv(i, order) {
if (vis[i]) {
continue;
}
cond_graph.emplace_back();
costs.emplace_back();
scc(i);
g++;
}
vl memo(g, -1);
function<ll(int)> dfs = [&](int i) {
ll &ans = memo[i];
if (ans != -1) {
return ans;
}
ans = costs[i];
for (auto j : cond_graph[i]) {
rmax(ans, dfs(j) + costs[i]);
}
return ans;
};
ll ans = 0;
rep(i, g) {
rmax(ans, dfs(i));
}
cout << ans << '\n';
}

93
CSES Problem Set/Coin Piles.cpp Normal file
View File

@@ -0,0 +1,93 @@
/* Problem URL: https://cses.fi/problemset/task/1754 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int t;
cin >> t;
while (t--) {
ll a, b;
cin >> a >> b;
if (a < b) {
swap(a, b);
}
if ((2 * b - a) >= 0 && (2 * b - a) % 3 == 0) {
cout << "YES\n";
} else {
cout << "NO\n";
}
}
}

139
CSES Problem Set/Collecting Numbers II.cpp Normal file
View File

@@ -0,0 +1,139 @@
/* Problem URL: https://cses.fi/problemset/task/2217 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (size_t i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (size_t i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
const int inf = INT32_MAX >> 1;
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m;
cin >> n >> m;
vi adj(n + 2, inf);
adj[0] = -1;
adj[n + 1] = n;
vi nums(n);
cin >> nums;
ll rounds = 1;
rep(i, n) {
if (adj[nums[i] - 1] == inf) {
rounds++;
}
adj[nums[i]] = i;
}
vi prev = adj;
while (m--) {
int a, b;
cin >> a >> b;
a--, b--;
swap(adj[nums[a]], adj[nums[b]]);
swap(nums[a], nums[b]);
if (adj[nums[a] - 1] > adj[nums[a]] && prev[nums[a] - 1] < prev[nums[a]]) {
rounds++;
}
if (adj[nums[a] - 1] < adj[nums[a]] && prev[nums[a] - 1] > prev[nums[a]]) {
rounds--;
}
if (nums[b] != nums[a] + 1) {
if (adj[nums[b] - 1] > adj[nums[b]] && prev[nums[b] - 1] < prev[nums[b]]) {
rounds++;
}
if (adj[nums[b] - 1] < adj[nums[b]] && prev[nums[b] - 1] > prev[nums[b]]) {
rounds--;
}
}
if (adj[nums[a] + 1] < adj[nums[a]] && prev[nums[a] + 1] > prev[nums[a]]) {
rounds++;
}
if (adj[nums[a] + 1] > adj[nums[a]] && prev[nums[a] + 1] < prev[nums[a]]) {
rounds--;
}
if (nums[b] != nums[a] - 1) {
if (adj[nums[b] + 1] < adj[nums[b]] && prev[nums[b] + 1] > prev[nums[b]]) {
rounds++;
}
if (adj[nums[b] + 1] > adj[nums[b]] && prev[nums[b] + 1] < prev[nums[b]]) {
rounds--;
}
}
cout << rounds << '\n';
swap(prev[nums[a]], prev[nums[b]]);
}
}

93
CSES Problem Set/Collecting Numbers.cpp Normal file
View File

@@ -0,0 +1,93 @@
/* Problem URL: https://cses.fi/problemset/task/2216 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (size_t i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (size_t i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
V<bool> nums(n + 1, false);
nums[0] = true;
ll count = 1;
rep(i, n) {
int num;
cin >> num;
nums[num] = true;
if (!nums[num - 1]) {
count++;
}
}
cout << count << '\n';
}

104
CSES Problem Set/Common Divisors.cpp Normal file
View File

@@ -0,0 +1,104 @@
/* Problem URL: https://cses.fi/problemset/task/1081 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (size_t i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (size_t i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int divs[(ll)1e6 + 1];
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
rep(i, n) {
int num;
cin >> num;
const size_t lim = sqrt(num);
for (size_t j = 1; j <= lim; j++) {
if (num % j == 0) {
divs[j]++;
if (j != num / j) {
divs[num / j]++;
}
}
}
}
size_t fds = 1;
for (size_t i = 1e6; i >= fds; i--) {
if (divs[i] >= 2) {
cout << i << '\n';
break;
}
}
}

103
CSES Problem Set/Concert Tickets.cpp Normal file
View File

@@ -0,0 +1,103 @@
/* Problem URL: https://cses.fi/problemset/task/1091 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (size_t i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (size_t i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m;
cin >> n >> m;
multiset<ll> tickets;
while (n--) {
ll num;
cin >> num;
tickets.insert(num);
}
while (m--) {
ll man;
cin >> man;
auto itr = tickets.upper_bound(man);
if (itr == tickets.begin()) {
cout << "-1\n";
continue;
}
itr--;
cout << *itr << '\n';
tickets.erase(itr);
}
}

112
CSES Problem Set/Counting Rooms.cpp Normal file
View File

@@ -0,0 +1,112 @@
/* Problem URL: https://cses.fi/problemset/task/1192 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (size_t i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (size_t i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m;
cin >> n >> m;
V<string> mat(n);
cin >> mat;
V<V<bool>> vis(n, V<bool>(m, false));
function<void(int, int)> floodfill = [&](int i, int j)
{
vis[i][j] = true;
int add[4][2] = {{0, 1}, {0, -1}, {-1, 0}, {1, 0}};
for (auto &k : add) {
int id = i + k[0];
int jd = j + k[1];
if (id >= 0 && id < n && jd >= 0 && jd < m && !vis[id][jd] && mat[id][jd] != '#') {
floodfill(id, jd);
}
}
};
int ans = 0;
rep(i, n) {
rep(j, m) {
if (!vis[i][j] && mat[i][j] == '.') {
ans++;
floodfill(i, j);
}
}
}
cout << ans << '\n';
}

128
CSES Problem Set/Course Schedule.cpp Normal file
View File

@@ -0,0 +1,128 @@
/* Problem URL: https://cses.fi/problemset/task/1679 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m;
cin >> n >> m;
vvi graph(n);
while (m--) {
int a, b;
cin >> a >> b;
graph[a - 1].push_back(b - 1);
}
V<bool> frame(n);
V<bool> vis(n);
vi ans;
function<bool(int)> dfs = [&](int i) {
vis[i] = true;
frame[i] = true;
for (auto j : graph[i]) {
if (frame[j]) {
return true;
}
if (vis[j]) {
continue;
}
if (dfs(j)) {
return true;
}
}
ans.push_back(i + 1);
frame[i] = false;
return false;
};
rep(i, n) {
if (!vis[i]) {
if (dfs(i)) {
cout << "IMPOSSIBLE\n";
return 0;
}
}
}
reverse(all(ans));
cout << ans;
}

123
CSES Problem Set/Creating Strings II.cpp Normal file
View File

@@ -0,0 +1,123 @@
/* Problem URL: https://cses.fi/problemset/task/1715 */
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;
template <class T, class comp = less<>>
using ordered_set = tree<T, null_type , comp , rb_tree_tag , tree_order_statistics_node_update>;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
string a;
cin >> a;
const ll mod = 1e9 + 7;
vl count(26);
repv(i, a) {
count[i - 'a']++;
}
auto fpow = [&](ll a, ll p) {
ll ans = 1;
ll now = a;
rep(i, 30) {
if ((p >> i) & 1) {
ans *= now;
ans %= mod;
}
now *= now;
now %= mod;
}
return ans;
};
vl fact(a.size() + 1);
fact[0] = 1;
nrep(i, 1, a.size() + 1) {
fact[i] = (fact[i - 1] * i) % mod;
}
ll ans = fact[a.size()];
rep(i, 26) {
ans = (ans * fpow(fact[count[i]], mod - 2)) % mod;
}
cout << ans << '\n';
}

93
CSES Problem Set/Creating Strings.cpp Normal file
View File

@@ -0,0 +1,93 @@
/* Problem URL: https://cses.fi/problemset/task/1622 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
string a;
cin >> a;
sortv(a);
V<string> fds;
do {
fds.emplace_back(a);
} while (next_permutation(all(a)));
cout << fds.size() << '\n';
repv(i, fds) {
cout << i << '\n';
}
}

78
CSES Problem Set/Cut and Paste.cpp Normal file
View File

@@ -0,0 +1,78 @@
/* Problem URL: https://cses.fi/problemset/task/2072 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (size_t i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (size_t i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
}

130
CSES Problem Set/Cycle Finding.cpp Normal file
View File

@@ -0,0 +1,130 @@
/* Problem URL: https://cses.fi/problemset/task/1197 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m;
cin >> n >> m;
V<tuple<int, int, ll>> edges;
while (m--) {
int a, b, c;
cin >> a >> b >> c;
a--, b--;
edges.emplace_back(a, b, c);
}
vl dis(n, 0);
vi parent(n, -1);
rep(i, n - 1) {
for (auto [a, b, c] : edges) {
if (dis[b] > dis[a] + c) {
dis[b] = dis[a] + c;
parent[b] = a;
}
}
}
int now = -1;
for (auto [a, b, c] : edges) {
if (dis[b] > dis[a] + c) {
parent[b] = a;
now = b;
break;
}
}
if (now == -1) {
cout << "NO\n";
return 0;
}
rep(i, n) {
now = parent[now];
}
vi path = {now + 1};
int tmp = parent[now];
while (tmp != now) {
path.push_back(tmp + 1);
tmp = parent[tmp];
}
path.push_back(now + 1);
reverse(all(path));
cout << "YES\n" << path;
}

113
CSES Problem Set/De Bruijn Sequence.cpp Normal file
View File

@@ -0,0 +1,113 @@
/* Problem URL: https://cses.fi/problemset/task/1692 */
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;
template <class T, class comp = less<>>
using ordered_set = tree<T, null_type , comp , rb_tree_tag , tree_order_statistics_node_update>;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
#define INF INT64_MAX >> 1
vector<int> de_brujin(int n, int k, int lim = INF) {
if (k == 1) return vector<int>(lim == INF ? 1 : n, 0);
vector<int> l = {0}, ret; // l eh lyndon word
while (true) {
if (l.size() == 0) {
if (lim == INF) break;
l.push_back(0);
}
if (n % l.size() == 0) for (int i : l) {
ret.push_back(i);
if (ret.size() == n+lim-1) return ret;
}
int p = l.size();
while (l.size() < n) l.push_back(l[l.size()%p]);
while (l.size() and l.back() == k-1) l.pop_back();
if (l.size()) l.back()++;
}
return ret;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
auto ans = de_brujin(n, 2, 1 << n);
repv(i, ans) {
cout << i;
}
cout << '\n';
}

99
CSES Problem Set/Digit Queries.cpp Normal file
View File

@@ -0,0 +1,99 @@
/* Problem URL: https://cses.fi/problemset/task/2431 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int q;
cin >> q;
while (q--) {
ll k;
cin >> k;
k--;
ll now = 1;
int digs = 1;
while (k >= digs * now * 9) {
k -= digs * now * 9;
now *= 10;
digs++;
}
ll act = now;
act += (k / digs);
k %= digs;
string fds = to_string(act);
cout << fds[k] << '\n';
}
}

127
CSES Problem Set/Distinct Colors.cpp Normal file
View File

@@ -0,0 +1,127 @@
/* Problem URL: https://cses.fi/problemset/task/1139 */
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;
template <class T, class comp = less<>>
using ordered_set = tree<T, null_type , comp , rb_tree_tag , tree_order_statistics_node_update>;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
vi colors(n);
cin >> colors;
V<set<int>> c(n);
vi ans(n);
vvi graph(n);
rep(i, n - 1) {
int a, b;
cin >> a >> b;
a--, b--;
graph[a].push_back(b);
graph[b].push_back(a);
}
function<void(int, int)> dfs = [&](int i, int p) {
c[i].insert(colors[i]);
for (auto j : graph[i]) {
if (j == p) {
continue;
}
dfs(j, i);
if (c[j].size() > c[i].size()) {
swap(c[j], c[i]);
}
for (auto k : c[j]) {
c[i].insert(k);
}
}
ans[i] = c[i].size();
};
dfs(0, 0);
cout << ans;
}

88
CSES Problem Set/Distinct Numbers.cpp Normal file
View File

@@ -0,0 +1,88 @@
/* Problem URL: https://cses.fi/problemset/task/1621 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (size_t i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (size_t i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
set<int> fds;
while (n--) {
int f;
cin >> f;
fds.insert(f);
}
cout << fds.size() << '\n';
}

195
CSES Problem Set/Distinct Routes.cpp Normal file
View File

@@ -0,0 +1,195 @@
/* Problem URL: https://cses.fi/problemset/task/1711 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (size_t i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (size_t i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m;
cin >> n >> m;
vvi graph(n);
V<tuple<int, ll, ll>> edges;
while (m--) {
int a, b;
cin >> a >> b;
a--, b--;
graph[a].push_back(edges.size());
edges.emplace_back(b, 1, 0);
graph[b].push_back(edges.size());
edges.emplace_back(a, 0, 0);
}
int inf = INT32_MAX >> 1;
vi level(n);
auto bfs = [&](){
level[0] = 0;
queue<int> q;
q.push(0);
while (!q.empty()) {
auto a = q.front();
q.pop();
for (auto id : graph[a]) {
auto [b, c, u] = edges[id];
if (level[b] > level[a] + 1 && c != u) {
level[b] = level[a] + 1;
q.push(b);
}
}
}
return level[n - 1] != inf;
};
V<bool> vis(n);
function<ll(int, ll)> dfs = [&](int i, ll flux){
if (flux == 0) {
return 0LL;
}
if (i == n - 1) {
return flux;
}
for (auto id : graph[i]) {
auto &[j, c, u] = edges[id];
if (level[j] <= level[i] || vis[j]) {
continue;
}
ll ans = dfs(j, min(flux, c - u));
if (ans == 0) {
continue;
}
auto &[_, __, ud] = edges[id ^ 1];
u += ans;
ud -= ans;
return ans;
}
vis[i] = true;
return 0LL;
};
ll ans = 0;
while (true) {
fill(all(level), inf);
if (!bfs()) {
break;
}
fill(all(vis), false);
ll count = 0;
while ((count = dfs(0, inf))) {
ans += count;
}
}
cout << ans << '\n';
while (ans--) {
vi tmp;
function<void(int)> dfs = [&](int i){
tmp.push_back(i + 1);
for (auto id : graph[i]) {
auto &[j, c, u] = edges[id];
if (u == 1) {
u--;
dfs(j);
return;
}
}
};
dfs(0);
cout << tmp.size() << '\n';
cout << tmp;
tmp.clear();
}
}

176
CSES Problem Set/Distinct Values Queries.cpp Normal file
View File

@@ -0,0 +1,176 @@
/* Problem URL: https://cses.fi/problemset/task/1734 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (size_t i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (size_t i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, q;
cin >> n >> q;
vl tmp(n);
cin >> tmp;
vl fds(n);
set<ll> nums;
rep(i, n) {
nums.insert(tmp[i]);
}
int v = 0;
map<ll, int> var;
repv(i, nums) {
var[i] = v;
v++;
}
rep(i, n) {
fds[i] = var[tmp[i]];
}
int len = (int)sqrt(n) + 1;
vi prev(n, -1);
vi next(n, n);
vl blocks(len, 0);
V<bool> uniq(n, false);
rep(i, n) {
if (prev[fds[i]] == -1) {
uniq[i] = true;
prev[fds[i]] = i;
blocks[i / len]++;
continue;
}
next[prev[fds[i]]] = i;
prev[fds[i]] = i;
}
V<tuple<int, int, size_t>> query(q);
size_t act = 0;
for (auto &[l, r, i] : query) {
cin >> l >> r;
l--, r--;
i = act;
act++;
}
sortv(query);
int ql = 0;
vl ans(q);
for (auto [l, r, i] : query) {
while (ql < l) {
blocks[ql / len]--;
uniq[next[ql]] = true;
blocks[next[ql] / len]++;
ql++;
}
if (l / len == r / len) {
ll sum = 0;
for (size_t i = l; i <= r; i++) {
if (uniq[i]) {
sum++;
}
}
ans[i] = sum;
continue;
}
ll sum = 0;
int bl = l / len;
int br = r / len;
for (int i = l, end = (bl + 1) * len - 1; i <= end; i++) {
if (uniq[i]) {
sum++;
}
}
for (int i = bl + 1; i <= br - 1; i++) {
sum += blocks[i];
}
for (int i = br * len; i <= r; i++) {
if (uniq[i]) {
sum++;
}
}
ans[i] = sum;
}
rep(i, q) {
cout << ans[i] << '\n';
}
}

102
CSES Problem Set/Distinct Values Subarrays.cpp Normal file
View File

@@ -0,0 +1,102 @@
/* Problem URL: https://cses.fi/problemset/task/3420 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
set<ll> seen;
vl fds(n);
cin >> fds;
int i = 0;
int j = 0;
ll ans = 0;
while (i < n) {
while (seen.count(fds[i])) {
seen.erase(fds[j]);
j++;
}
seen.insert(fds[i]);
i++;
ans += i - j;
}
cout << ans << '\n';
}

101
CSES Problem Set/Distinct Values Subsequences.cpp Normal file
View File

@@ -0,0 +1,101 @@
/* Problem URL: https://cses.fi/problemset/task/3421 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
map<ll, ll> var;
while (n--) {
ll num;
cin >> num;
var[num]++;
}
ll mod = 1e9 + 7;
ll ans = 0;
ll now = 0;
repv(i, var) {
ans += i.second;
ans += i.second * now;
ans %= mod;
now = (now * i.second + now + i.second) % mod;
}
cout << ans << '\n';
}

117
CSES Problem Set/Distributing Apples.cpp Normal file
View File

@@ -0,0 +1,117 @@
/* Problem URL: https://cses.fi/problemset/task/1716 */
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;
template <class T, class comp = less<>>
using ordered_set = tree<T, null_type , comp , rb_tree_tag , tree_order_statistics_node_update>;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m;
cin >> n >> m;
const ll mod = 1e9 + 7;
vl fact(2e6 + 1);
fact[0] = 1;
nrep(i, 1, fact.size()) {
fact[i] = (fact[i - 1] * i) % mod;
}
auto fpow = [&](ll a, ll b) {
ll ans = 1;
ll now = a;
rep(i, 30) {
if ((b >> i) & 1) {
ans *= now;
ans %= mod;
}
now *= now;
now %= mod;
}
return ans;
};
auto comb = [&](ll n, ll k) {
return (fact[n] * fpow((fact[k] * fact[n - k]) % mod, mod - 2)) % mod;
};
cout << comb(n + m - 1, m) << '\n';
}

173
CSES Problem Set/Download Speed.cpp Normal file
View File

@@ -0,0 +1,173 @@
/* Problem URL: https://cses.fi/problemset/task/1694 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (size_t i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (size_t i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m;
cin >> n >> m;
vvi graph(n);
V<tuple<int, ll, ll>> edges;
while (m--) {
ll a, b, c;
cin >> a >> b >> c;
a--, b--;
graph[a].push_back(edges.size());
edges.emplace_back(b, c, 0);
graph[b].push_back(edges.size());
edges.emplace_back(a, 0, 0);
}
int inf = INT32_MAX >> 1;
auto flux = [&](){
vi level(n);
auto bfs = [&](){
queue<int> q;
level[0] = 0;
q.push(0);
while (!q.empty()) {
int i = q.front();
q.pop();
for (auto id : graph[i]) {
auto [j, c, u] = edges[id];
if (level[i] + 1 < level[j] && c != u) {
level[j] = level[i] + 1;
q.push(j);
}
}
}
return level[n - 1] != inf;
};
V<bool> vis(n);
function<ll(int, ll)> dfs = [&](int i, ll flux){
if (flux == 0) {
return 0LL;
}
if (i == n - 1) {
return flux;
}
for (auto id : graph[i]) {
auto &[j, c, u] = edges[id];
if (level[i] >= level[j] || vis[j]) {
continue;
}
ll ans = dfs(j, min(flux, c - u));
if (ans == 0) {
continue;
}
auto &[_, __, ud] = edges[id ^ 1];
u += ans;
ud -= ans;
return ans;
}
vis[i] = true;
return 0LL;
};
ll ans = 0;
while (true) {
fill(all(level), inf);
if (!bfs()) {
break;
}
fill(all(vis), false);
ll pushed = 0;
while ((pushed = dfs(0, inf))) {
ans += pushed;
}
}
return ans;
};
cout << flux() << '\n';
}

103
CSES Problem Set/Exponentiation II.cpp Normal file
View File

@@ -0,0 +1,103 @@
/* Problem URL: https://cses.fi/problemset/task/1712 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (size_t i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (size_t i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
ll fastpow(ll n, ll p, ll mod)
{
ll total = n;
ll ans = 1;
for (size_t i = 0; i < 30; i++) {
if (p & (1 << i)) {
ans *= total;
ans %= mod;
}
total *= total;
total %= mod;
}
return ans;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int q;
cin >> q;
while (q--) {
ll a, b, c;
cin >> a >> b >> c;
ll p = fastpow(b, c, 1e9 + 6);
cout << fastpow(a, p, 1e9 + 7) << '\n';
}
}

108
CSES Problem Set/Exponentiation.cpp Normal file
View File

@@ -0,0 +1,108 @@
/* Problem URL: https://cses.fi/problemset/task/1095 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (size_t i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (size_t i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
ll m = 1e9 + 7;
ll fastpow(ll a, ll p)
{
if (p == 0) {
return 1;
}
ll ans = 1;
ll total = a;
for (size_t i = 0; i < 30; i++) {
if (p & (1 << i)) {
ans *= total;
ans %= m;
}
total *= total;
total %= m;
}
return ans;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
while (n--) {
ll a, b;
cin >> a >> b;
cout << fastpow(a, b) << '\n';
}
}

108
CSES Problem Set/Factory Machines.cpp Normal file
View File

@@ -0,0 +1,108 @@
/* Problem URL: https://cses.fi/problemset/task/1620 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (size_t i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (size_t i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
ll n, t;
cin >> n >> t;
vl machines(n);
cin >> machines;
ll ans = 0;
ll low = 0;
ll high = INT64_MAX;
while (low <= high) {
ll mid = (high - low) / 2 + low;
ll sum = 0;
repv(i, machines) {
sum += mid / i;
if (sum >= t) {
break;
}
}
if (sum >= t) {
ans = mid;
high = mid - 1;
continue;
}
low = mid + 1;
}
cout << ans << '\n';
}

110
CSES Problem Set/Ferris Wheel.cpp Normal file
View File

@@ -0,0 +1,110 @@
/* Problem URL: https://cses.fi/problemset/task/1090 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (size_t i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (size_t i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
ll n, x;
cin >> n >> x;
vl fds(n);
cin >> fds;
int ans = 0;
ll now = 0;
sortv(fds);
int i = 0;
int j = n - 1;
int get = 0;
while (i <= j) {
if (i == j) {
ans++;
break;
}
if (fds[i] + fds[j] <= x) {
i++;
j--;
} else {
j--;
}
ans++;
}
cout << ans << '\n';
}

128
CSES Problem Set/Finding a Centroid.cpp Normal file
View File

@@ -0,0 +1,128 @@
/* Problem URL: https://cses.fi/problemset/task/2079 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
vvi graph(n);
rep(i, n - 1) {
int a, b;
cin >> a >> b;
a--, b--;
graph[a].push_back(b);
graph[b].push_back(a);
}
vi sizes(n);
function<int(int, int)> calc = [&](int i, int p) {
sizes[i] = 1;
for (auto j : graph[i]) {
if (j == p) {
continue;
}
sizes[i] += calc(j, i);
}
return sizes[i];
};
calc(0, 0);
function<int(int)> dfs = [&](int i) {
sizes[i] = 1;
int choice = -1;
for (auto j : graph[i]) {
if (sizes[j] > n / 2) {
choice = j;
continue;
}
sizes[i] += sizes[j];
}
if (choice == -1) {
return i;
}
return dfs(choice);
};
cout << dfs(0) + 1 << '\n';
}

149
CSES Problem Set/Flight Routes Check.cpp Normal file
View File

@@ -0,0 +1,149 @@
/* Problem URL: https://cses.fi/problemset/task/1682 */
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;
template <class T, class comp = less<>>
using ordered_set = tree<T, null_type , comp , rb_tree_tag , tree_order_statistics_node_update>;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m;
cin >> n >> m;
vvi graph(n);
vvi inv(n);
while (m--) {
int a, b;
cin >> a >> b;
a--, b--;
graph[a].push_back(b);
inv[b].push_back(a);
}
V<bool> vis(n);
stack<int> s;
function<void(int)> postorder = [&](int i) {
vis[i] = true;
for (auto j : graph[i]) {
if (vis[j]) {
continue;
}
postorder(j);
}
s.push(i);
};
rep(i, n) {
if (vis[i]) {
continue;
}
postorder(i);
}
int choice = s.top();
fill(all(vis), false);
function<void(int)> dfs = [&](int i) {
vis[i] = true;
for (auto j : inv[i]) {
if (vis[j]) {
continue;
}
dfs(j);
}
};
dfs(choice);
rep(i, n) {
if (!vis[i]) {
cout << "NO\n";
cout << i + 1 << ' ' << choice + 1 << '\n';
return 0;
}
}
cout << "YES\n";
}

124
CSES Problem Set/Flight Routes.cpp Normal file
View File

@@ -0,0 +1,124 @@
/* Problem URL: https://cses.fi/problemset/task/1196 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m, k;
cin >> n >> m >> k;
V<V<pair<int, ll>>> graph(n);
while (m--) {
int a, b, c;
cin >> a >> b >> c;
a--, b--;
graph[a].emplace_back(b, c);
}
ll inf = INT64_MAX >> 1;
V<multiset<ll>> dis(n, multiset<ll>({inf}));
auto dijkstra = [&]() {
dis[0].insert(0);
priority_queue<pair<ll, int>, V<pair<ll, int>>, greater<>> pq;
pq.emplace(0, 0);
while (!pq.empty()) {
auto [c, i] = pq.top();
pq.pop();
if (c > *prev(dis[i].end())) {
continue;
}
for (auto [j, cc] : graph[i]) {
if (dis[j].size() < k || c + cc < *prev(dis[j].end())) {
dis[j].insert(c + cc);
pq.emplace(c + cc, j);
if (dis[j].size() > k) {
dis[j].erase(prev(dis[j].end()));
}
}
}
}
};
dijkstra();
for (auto i : dis.back()) {
cout << i << ' ';
}
cout << '\n';
}

136
CSES Problem Set/Forest Queries.cpp Normal file
View File

@@ -0,0 +1,136 @@
/* Problem URL: https://cses.fi/problemset/task/1652 */
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;
template <class T, class comp = less<>>
using ordered_set = tree<T, null_type , comp , rb_tree_tag , tree_order_statistics_node_update>;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, q;
cin >> n >> q;
vvi count(n, vi(n));
rep(i, n) {
rep(j, n) {
char c;
cin >> c;
if (c == '*') {
count[i][j] = 1;
}
}
}
rep(i, n) {
rep(j, n) {
if (i > 0) {
count[i][j] += count[i - 1][j];
}
if (j > 0) {
count[i][j] += count[i][j - 1];
}
if (i > 0 && j > 0) {
count[i][j] -= count[i - 1][j - 1];
}
}
}
while (q--) {
int x1, y1, x2, y2;
cin >> x1 >> y1 >> x2 >> y2;
x1--, y1--, x2--, y2--;
int ans = count[x2][y2];
if (x1 > 0) {
ans -= count[x1 - 1][y2];
}
if (y1 > 0) {
ans -= count[x2][y1 - 1];
}
if (x1 > 0 && y1 > 0) {
ans += count[x1 - 1][y1 - 1];
}
cout << ans << '\n';
}
}

114
CSES Problem Set/Game Routes.cpp Normal file
View File

@@ -0,0 +1,114 @@
/* Problem URL: https://cses.fi/problemset/task/1681 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m;
cin >> n >> m;
vvi graph(n);
while (m--) {
int a, b;
cin >> a >> b;
graph[a - 1].push_back(b - 1);
}
const ll mod = 1e9 + 7;
vl dp(n, -1);
function<ll(int)> dfs = [&](int i) {
if (i == n - 1) {
dp[i] = 1;
return dp[i];
}
if (dp[i] != -1) {
return dp[i];
}
dp[i] = 0;
for (auto j : graph[i]) {
dp[i] += dfs(j);
dp[i] %= mod;
}
return dp[i];
};
cout << dfs(0) << '\n';
}

207
CSES Problem Set/Giant Pizza.cpp Normal file
View File

@@ -0,0 +1,207 @@
/* Problem URL: https://cses.fi/problemset/task/1684 */
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;
template <class T, class comp = less<>>
using ordered_set = tree<T, null_type , comp , rb_tree_tag , tree_order_statistics_node_update>;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m;
cin >> m >> n;
n <<= 1;
vvi graph(n);
vvi inv(n);
while (m--) {
char op1, op2;
int a, b;
cin >> op1 >> a >> op2 >> b;
a--, b--;
if (op1 == '-') {
a += n >> 1;
}
if (op2 == '-') {
b += n >> 1;
}
graph[a].push_back((b + (n >> 1)) % n);
graph[b].push_back((a + (n >> 1)) % n);
inv[(b + (n >> 1)) % n].push_back(a);
inv[(a + (n >> 1)) % n].push_back(b);
}
V<bool> vis(n);
vi top;
function<void(int)> preorder = [&](int i) {
vis[i] = true;
for (auto j : graph[i]) {
if (vis[j]) {
continue;
}
preorder(j);
}
top.push_back(i);
};
rep(i, n) {
if (vis[i]) {
continue;
}
preorder(i);
}
reverse(all(top));
fill(all(vis), false);
vi group(n);
vi representant(n);
int g = 0;
function<void(int)> scc = [&](int i) {
vis[i] = true;
group[i] = g;
for (auto j : inv[i]) {
if (vis[j]) {
continue;
}
scc(j);
}
};
repv(i, top) {
if (vis[i]) {
continue;
}
scc(i);
representant.push_back(i);
g++;
}
rep(i, n >> 1) {
if (group[i] == group[i + (n >> 1)]) {
cout << "IMPOSSIBLE\n";
return 0;
}
}
fill(all(vis), false);
reverse(all(representant));
string ans(n >> 1, ' ');
function<void(int)> getans = [&](int i) {
vis[i] = true;
if (ans[i % (n >> 1)] == ' ') {
ans[i % (n >> 1)] = i < (n >> 1) ? '-' : '+';
}
repv(j, graph[i]) {
if (vis[j] || group[j] != group[i]) {
continue;
}
getans(j);
}
};
repv(i, representant) {
if (vis[i]) {
continue;
}
getans(i);
}
repv(i, ans) {
cout << i << ' ';
}
cout << '\n';
}

89
CSES Problem Set/Gray Code.cpp Normal file
View File

@@ -0,0 +1,89 @@
/* Problem URL: https://cses.fi/problemset/task/2205 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
rep(i, (1 << n)) {
int act = i ^ (i >> 1);
for (int j = n - 1; j >= 0; j--) {
int bit = (act >> j) & 1;
cout << (char)(bit + '0');
}
cout << '\n';
}
}

100
CSES Problem Set/Grid Coloring I.cpp Normal file
View File

@@ -0,0 +1,100 @@
/* Problem URL: https://cses.fi/problemset/task/3311 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m;
cin >> n >> m;
V<string> fds(n);
cin >> fds;
V<string> ans(n, string(m, ' '));
rep(i, n) {
rep(j, m) {
int now = 0;
char act;
while (act = now + 'A', act == fds[i][j] || (i > 0 && act == ans[i - 1][j]) || (j > 0 && act == ans[i][j - 1])) {
now++;
}
ans[i][j] = act;
}
}
repv(i, ans) {
cout << i << '\n';
}
}

145
CSES Problem Set/Grid Path Description.cpp Normal file
View File

@@ -0,0 +1,145 @@
/* Problem URL: https://cses.fi/problemset/task/1625 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
char a[49];
bool frame[7][7];
int func(int i, int j, int pos)
{
if (pos >= 48) {
return (i == 6 && j == 0);
}
if (i == 6 && j == 0) {
return 0;
}
if ((i == 0 || frame[i - 1][j]) && (i == 6 || frame[i + 1][j]) && (j != 0 && !frame[i][j - 1]) && (j != 6 && !frame[i][j + 1])) {
return 0;
}
if ((j == 0 || frame[i][j - 1]) && (j == 6 || frame[i][j + 1]) && (i != 0 && !frame[i - 1][j]) && (i != 6 && !frame[i + 1][j])) {
return 0;
}
frame[i][j] = true;
if (a[pos] == '?') {
int add[][4] = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
int ans = 0;
repv(k, add) {
int id = i + k[0];
int jd = j + k[1];
if (id >= 0 && id < 7 && jd >= 0 && jd < 7 && !frame[id][jd]) {
ans += func(id, jd, pos + 1);
}
}
frame[i][j] = false;
return ans;
}
int id = i;
int jd = j;
switch (a[pos]) {
case 'U':
id--;
break;
case 'D':
id++;
break;
case 'L':
jd--;
break;
case 'R':
jd++;
}
int ans = 0;
if (id >= 0 && id < 7 && jd >= 0 && jd < 7 && !frame[id][jd]) {
ans = func(id, jd, pos + 1);
}
frame[i][j] = false;
return ans;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
scanf("%s", a);
printf("%d\n", func(0, 0, 0));
}

121
CSES Problem Set/Hamiltonian Flights.cpp Normal file
View File

@@ -0,0 +1,121 @@
/* Problem URL: https://cses.fi/problemset/task/1690 */
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;
template <class T, class comp = less<>>
using ordered_set = tree<T, null_type , comp , rb_tree_tag , tree_order_statistics_node_update>;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m;
cin >> n >> m;
vvi graph(n);
while (m--) {
int a, b;
cin >> a >> b;
graph[a - 1].push_back(b - 1);
}
const ll mod = 1e9 + 7;
vvl memo(n, vl(1 << n, -1));
function<ll(int, ll)> dp = [&](int i, ll mask) -> ll {
if (i == n - 1) {
return mask == 0;
}
ll &ans = memo[i][mask];
if (ans != -1) {
return ans;
}
ans = 0;
repv(j, graph[i]) {
if (mask & (1 << j)) {
ans += dp(j, mask ^ (1 << j));
ans %= mod;
}
}
return ans;
};
cout << dp(0, ((1 << n) - 1) ^ 1) << '\n';
}

147
CSES Problem Set/High Score.cpp Normal file
View File

@@ -0,0 +1,147 @@
/* Problem URL: https://cses.fi/problemset/task/1673 */
#include <bits/stdc++.h>
#include <algorithm>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m;
cin >> n >> m;
V<tuple<int, int, ll>> edges;
V<V<pair<ll, int>>> graph(n);
while (m--) {
int a, b, c;
cin >> a >> b >> c;
a--, b--;
edges.emplace_back(a, b, -c);
graph[a].emplace_back(b, c);
}
V<bool> reachable(n, false);
V<bool> vis(n, false);
reachable.back() = true;
function<bool(int)> dfs = [&](int i) -> bool {
vis[i] = true;
for (auto [j, c] : graph[i]) {
if (reachable[j]) {
reachable[i] = true;
}
if (vis[j]) {
continue;
}
if (dfs(j)) {
reachable[i] = true;
}
}
return reachable[i];
};
dfs(0);
auto bellman = [&]() -> ll {
ll inf = INT64_MAX >> 1;
vl dis(n, inf);
dis[0] = 0;
rep(i, n - 1) {
for (auto [a, b, c] : edges) {
if (dis[a] < inf) {
rmin(dis[b], dis[a] + c);
}
}
}
for (auto [a, b, c] : edges) {
if (dis[a] < inf && dis[b] > dis[a] + c) {
if (reachable[b]) {
return -1;
}
dis[b] = dis[a] + c;
}
}
return -dis.back();
};
cout << bellman() << '\n';
}

131
CSES Problem Set/Increasing Subsequence II.cpp Normal file
View File

@@ -0,0 +1,131 @@
/* Problem URL: https://cses.fi/problemset/task/1748 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
map<int, int> var;
vi fds(n);
repv(i, fds) {
cin >> i;
var[i] = 0;
}
int v = 0;
repv(i, var) {
i.second = v;
v++;
}
repv(i, fds) {
i = var[i];
}
while (__builtin_popcount(fds.size()) != 1) {
fds.push_back(0);
}
vl seg(fds.size() * 2);
ll mod = 1e9 + 7;
auto update = [&](int i, ll v) {
seg[i + fds.size()] += v;
seg[i + fds.size()] %= mod;
for (i = (i + fds.size()) / 2; i > 0; i /= 2) {
seg[i] = (seg[i * 2] + seg[i * 2 + 1]) % mod;
}
};
function<ll(int, int, int, int, int)> search = [&](int i, int l, int r, int tl, int tr) -> ll {
if (l > tr || r < tl) {
return 0;
}
if (l >= tl && r <= tr) {
return seg[i];
}
int mid = (l + r) / 2;
return (search(i * 2, l, mid, tl, tr) + search(i * 2 + 1, mid + 1, r, tl, tr)) % mod;
};
rep(i, n) {
ll tmp = search(1, 0, fds.size() - 1, 0, fds[i] - 1) + 1;
update(fds[i], tmp);
}
cout << search(1, 0, n - 1, 0, n - 1) << '\n';
}

195
CSES Problem Set/Investigation.cpp Normal file
View File

@@ -0,0 +1,195 @@
/* Problem URL: https://cses.fi/problemset/task/1202 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m;
cin >> n >> m;
V<V<pair<int, ll>>> graph(n);
V<V<pair<int, ll>>> rev(n);
while (m--) {
int a, b, c;
cin >> a >> b >> c;
a--, b--;
graph[a].emplace_back(b, c);
rev[b].emplace_back(a, c);
}
ll inf = INT64_MAX >> 1;
vl dis(n, inf);
auto dijkstra = [&]() {
priority_queue<pair<ll, int>, V<pair<ll, int>>, greater<>> pq;
dis[0] = 0;
pq.emplace(0, 0);
while (!pq.empty()) {
auto [c, i] = pq.top();
pq.pop();
if (c > dis[i]) {
continue;
}
for (auto [j, cc] : graph[i]) {
if (dis[j] > c + cc) {
dis[j] = c + cc;
pq.emplace(dis[j], j);
}
}
}
return dis.back();
};
ll ans = dijkstra();
vl dp1(n, -1);
ll mod = 1e9 + 7;
function<ll(int, ll)> dfs1 = [&](int i, ll c) {
if (i == 0) {
dp1[i] = 1;
return dp1[i];
}
if (dp1[i] != -1) {
return dp1[i];
}
dp1[i] = 0;
for (auto [j, cc] : rev[i]) {
if (dis[j] == c - cc) {
dp1[i] += dfs1(j, c - cc);
dp1[i] %= mod;
}
}
return dp1[i];
};
vi dp2(n, -1);
function<int(int, ll)> dfsmax = [&](int i, ll c) {
if (i == 0) {
dp2[i] = 1;
return dp2[i];
}
if (dp2[i] != -1) {
return dp2[i];
}
dp2[i] = 0;
for (auto [j, cc] : rev[i]) {
if (dis[j] == c - cc) {
int res = dfsmax(j, c - cc);
if (res > 0) {
rmax(dp2[i], res + 1);
}
}
}
return dp2[i];
};
vi dp3(n, -1);
function<int(int, ll)> dfsmin = [&](int i, ll c) {
if (i == 0) {
dp3[i] = 1;
return dp3[i];
}
if (dp3[i] != -1) {
return dp3[i];
}
dp3[i] = INT32_MAX >> 1;
for (auto [j, cc] : rev[i]) {
if (dis[j] == c - cc) {
rmin(dp3[i], dfsmin(j, c - cc) + 1);
}
}
return dp3[i];
};
cout << ans << ' ' << dfs1(n - 1, ans) << ' ' << dfsmin(n - 1, ans) - 1 << ' ' << dfsmax(n - 1, ans) - 1 << '\n';
}

100
CSES Problem Set/Josephus Problem I.cpp Normal file
View File

@@ -0,0 +1,100 @@
/* Problem URL: https://cses.fi/problemset/task/2162 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (size_t i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (size_t i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
list<int> children;
rep(i, n) {
children.push_back(i + 1);
}
auto itr = children.begin();
while (!children.empty()) {
itr++;
if (itr == children.end()) {
itr = children.begin();
}
cout << *itr << ' ';
itr = children.erase(itr);
if (itr == children.end()) {
itr = children.begin();
}
}
cout << '\n';
}

100
CSES Problem Set/Josephus Problem II.cpp Normal file
View File

@@ -0,0 +1,100 @@
/* Problem URL: https://cses.fi/problemset/task/2163 */
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#include <ext/pb_ds/detail/standard_policies.hpp>
using namespace __gnu_pbds;
using namespace std;
typedef tree<int, null_type, less<>, rb_tree_tag, tree_order_statistics_node_update> ordered_set;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (size_t i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (size_t i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
ll n, k;
cin >> n >> k;
ll now = 0;
ordered_set set;
rep(i, n) {
set.insert(i + 1);
}
while (!set.empty()) {
now = (now + k) % set.size();
auto itr = set.find_by_order(now);
cout << *itr << ' ';
set.erase(itr);
}
cout << '\n';
}

92
CSES Problem Set/Josephus Queries.cpp Normal file
View File

@@ -0,0 +1,92 @@
/* Problem URL: https://cses.fi/problemset/task/2164 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (size_t i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (size_t i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int q;
cin >> q;
while (q--) {
ll n, k;
cin >> n >> k;
ll count = 1;
while (k > n / 2 + n % 2) {
k -= n / 2 + n % 2;
n /= 2;
count <<= 1;
}
cout << k * count << '\n';
}
}

105
CSES Problem Set/Knight Moves Grid.cpp Normal file
View File

@@ -0,0 +1,105 @@
/* Problem URL: https://cses.fi/problemset/task/3217 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
int inf = INT32_MAX >> 1;
vvi dis(n, vi(n, inf));
dis[0][0] = 0;
queue<pair<int, int>> q;
q.emplace(0, 0);
while (!q.empty()) {
auto [i, j] = q.front();
q.pop();
int add[][2] = {{2, 1}, {-2, 1}, {1, 2}, {-1, 2}, {2, -1}, {1, -2}, {-2, -1}, {-1, -2}};
repv(k, add) {
int id = i + k[0];
int jd = j + k[1];
if (id >= 0 && id < n && jd >= 0 && jd < n && dis[id][jd] > dis[i][j] + 1) {
dis[id][jd] = dis[i][j] + 1;
q.emplace(id, jd);
}
}
}
cout << dis;
}

173
CSES Problem Set/Knight's Tour.cpp Normal file
View File

@@ -0,0 +1,173 @@
/* Problem URL: https://cses.fi/problemset/task/1689 */
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;
template <class T, class comp = less<>>
using ordered_set = tree<T, null_type , comp , rb_tree_tag , tree_order_statistics_node_update>;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int x, y;
cin >> x >> y;
x--, y--;
int n = 8;
vvi ans(n, vi(n));
vvi count(n, vi(n));
int add[][2] = {{1, 2}, {1, -2}, {-1, 2}, {-1, -2}, {2, 1}, {2, -1}, {-2, 1}, {-2, -1}};
auto check = [&](int i, int j) {
return i >= 0 && i < n && j >= 0 && j < n;
};
rep(i, n) {
rep(j, n) {
repv(k, add) {
int id = i + k[0];
int jd = j + k[1];
if (check(id, jd)) {
count[id][jd]++;
}
}
}
}
swap(x, y);
int oo = INT32_MAX >> 1;
int curx = x;
int cury = y;
ans[x][y] = 1;
function<bool(int, int)> dfs = [&](int i, int j) {
if (ans[i][j] == 64) {
return true;
}
int prev = count[i][j];
count[i][j] = oo;
priority_queue<tuple<int, int, int>, V<tuple<int, int, int>>, greater<>> pq;
repv(k, add) {
int id = i + k[0];
int jd = j + k[1];
if (check(id, jd)) {
count[id][jd]--;
pq.emplace(count[id][jd], id, jd);
}
}
while (!pq.empty()) {
auto [ty, id, jd] = pq.top();
pq.pop();
if (ty > 8) {
break;
}
ans[id][jd] = ans[i][j] + 1;
if (dfs(id, jd)) {
return true;
}
}
repv(k, add) {
int id = i + k[0];
int jd = j + k[1];
if (check(id, jd)) {
count[id][jd]++;
}
}
count[i][j] = prev;
return false;
};
dfs(x, y);
cout << ans;
}

163
CSES Problem Set/Labyrinth.cpp Normal file
View File

@@ -0,0 +1,163 @@
/* Problem URL: https://cses.fi/problemset/task/1193 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m;
cin >> n >> m;
V<string> fds(n);
cin >> fds;
int is;
int js;
rep(i, n) {
rep(j, m) {
if (fds[i][j] == 'A') {
is = i;
js = j;
goto leave;
}
}
}
leave:
string ans;
auto bfs = [&](int i, int j) {
queue<pair<int, int>> q;
q.emplace(i, j);
while (!q.empty()) {
auto [i, j] = q.front();
q.pop();
int add[][3] = {{0, 1, 0}, {1, 0, 1}, {0, -1, 2}, {-1, 0, 3}};
for (auto &k : add) {
int id = i + k[0];
int jd = j + k[1];
if (id >= 0 && id < n && jd >= 0 && jd < m) {
if (fds[id][jd] == 'B') {
char var[] = {'R', 'D', 'L', 'U'};
ans.push_back(var[k[2]]);
return make_pair(i, j);
}
if (fds[id][jd] == '.') {
fds[id][jd] = k[2];
q.emplace(id, jd);
}
}
}
}
return make_pair(-1, -1);
};
auto [i, j] = bfs(is, js);
if (i == -1) {
cout << "NO\n";
return 0;
}
cout << "YES\n";
while (fds[i][j] != 'A') {
switch(fds[i][j]) {
case 0:
ans.push_back('R');
j--;
break;
case 1:
ans.push_back('D');
i--;
break;
case 2:
ans.push_back('L');
j++;
break;
case 3:
ans.push_back('U');
i++;
break;
}
}
reverse(all(ans));
cout << ans.size() << '\n' << ans << '\n';
}

104
CSES Problem Set/List Removals.cpp Normal file
View File

@@ -0,0 +1,104 @@
/* Problem URL: https://cses.fi/problemset/task/1749 */
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;
template <class T, class comp = less<>>
using ordered_set = tree<T, null_type , comp , rb_tree_tag , tree_order_statistics_node_update>;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
ordered_set<pair<int, int>> ord;
rep(i, n) {
int x;
cin >> x;
ord.insert({i, x});
}
while (n--) {
int i;
cin >> i;
i--;
auto itr = ord.find_by_order(i);
cout << itr->second << " \n"[n == 0];
ord.erase(itr);
}
}

126
CSES Problem Set/Longest Common Subsequence.cpp Normal file
View File

@@ -0,0 +1,126 @@
/* Problem URL: https://cses.fi/problemset/task/3403 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m;
cin >> n >> m;
vi a(n);
vi b(m);
cin >> a >> b;
vvi dp(n + 1, vi(m + 1));
nrep(i, 1, n + 1) {
nrep(j, 1, m + 1) {
if (a[i - 1] == b[j - 1]) {
dp[i][j] = dp[i - 1][j - 1] + 1;
continue;
}
dp[i][j] = max(dp[i - 1][j], dp[i][j - 1]);
}
}
cout << dp[n][m] << '\n';
// if (dp[n][m] == 0) {
// return 0;
// }
vi ans(dp[n][m]);
int act = dp[n][m] - 1;
int i = n;
int j = m;
while (i > 0 && j > 0) {
if (a[i - 1] == b[j - 1]) {
ans[act] = a[i - 1];
i--;
j--;
act--;
continue;
}
if (dp[i - 1][j] > dp[i][j - 1]) {
i--;
continue;
}
j--;
}
cout << ans;
}

132
CSES Problem Set/Longest Flight Route.cpp Normal file
View File

@@ -0,0 +1,132 @@
/* Problem URL: https://cses.fi/problemset/task/1680 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m;
cin >> n >> m;
vvi graph(n);
while (m--) {
int a, b;
cin >> a >> b;
graph[a - 1].push_back(b - 1);
}
vi dp(n, -1);
function<int(int)> dfs = [&](int i) {
if (i == n - 1) {
dp[i] = 1;
return dp[i];
}
if (dp[i] != -1) {
return dp[i];
}
dp[i] = 0;
for (auto j : graph[i]) {
int res = dfs(j);
if (res > 0) {
rmax(dp[i], res + 1);
}
}
return dp[i];
};
if (dfs(0) == 0) {
cout << "IMPOSSIBLE\n";
return 0;
}
vi ans = {1};
int now = 0;
while (now != n - 1) {
for (auto j : graph[now]) {
if (dp[j] == dp[now] - 1) {
ans.push_back(j + 1);
now = j;
break;
}
}
}
cout << ans.size() << '\n' << ans;
}

130
CSES Problem Set/Mail Delivery.cpp Normal file
View File

@@ -0,0 +1,130 @@
/* Problem URL: https://cses.fi/problemset/task/1691 */
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;
template <class T, class comp = less<>>
using ordered_set = tree<T, null_type , comp , rb_tree_tag , tree_order_statistics_node_update>;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m;
cin >> n >> m;
V<V<pair<int, int>>> graph(n);
V<bool> used(m);
rep(i, m) {
int a, b;
cin >> a >> b;
a--, b--;
graph[a].emplace_back(b, i);
graph[b].emplace_back(a, i);
}
rep(i, n) {
if (graph[i].size() & 1) {
cout << "IMPOSSIBLE\n";
return 0;
}
}
vi ans;
stack<int> euler;
euler.push(0);
while (!euler.empty()) {
int i = euler.top();
while (!graph[i].empty() && used[graph[i].back().second]) {
graph[i].pop_back();
}
if (graph[i].empty()) {
ans.push_back(i + 1);
euler.pop();
continue;
}
used[graph[i].back().second] = true;
euler.push(graph[i].back().first);
graph[i].pop_back();
}
if (ans.size() != m + 1) {
cout << "IMPOSSIBLE\n";
return 0;
}
cout << ans;
}

104
CSES Problem Set/Maximum Subarray Sum II.cpp Normal file
View File

@@ -0,0 +1,104 @@
/* Problem URL: https://cses.fi/problemset/task/1644 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, a, b;
cin >> n >> a >> b;
vl fds(n);
cin >> fds;
vl pref(n + 1);
rep(i, n) {
pref[i + 1] = pref[i] + fds[i];
}
ll ans = INT64_MIN >> 1;
int i = a;
multiset<ll> var;
while (i <= n) {
if (i - b - 1 >= 0) {
var.erase(var.find(pref[i - b - 1]));
}
var.insert(pref[i - a]);
rmax(ans, pref[i] - *var.begin());
i++;
}
cout << ans << '\n';
}

102
CSES Problem Set/Maximum Subarray Sum.cpp Normal file
View File

@@ -0,0 +1,102 @@
/* Problem URL: https://cses.fi/problemset/task/1643 */
#include <bits/stdc++.h>
#include <algorithm>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (size_t i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (size_t i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
vl fds(n);
cin >> fds;
ll now = 0;
ll ans = 0;
ll maximus = INT64_MIN;
repv(i, fds) {
now += i;
rmax(now, 0LL);
rmax(ans, now);
rmax(maximus, i);
}
if (maximus < 0) {
cout << maximus << '\n';
return 0;
}
cout << ans << '\n';
}

150
CSES Problem Set/Meet in the Middle.cpp Normal file
View File

@@ -0,0 +1,150 @@
/* Problem URL: https://cses.fi/problemset/task/1628/ */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
ll s;
cin >> n >> s;
vl fds(n);
cin >> fds;
int mid = n / 2;
vl count;
function<void(int, ll)> func1 = [&](int i, ll sum) {
if (sum > s) {
return;
}
if (i >= mid) {
count.push_back(sum);
return;
}
func1(i + 1, sum);
func1(i + 1, sum + fds[i]);
};
ll ans = 0;
vl sum;
vl c;
function<void(int, ll)> func2 = [&](int i, ll ss) {
if (ss > s) {
return;
}
if (i >= n - mid) {
// ans += count[s - sum];
auto itr = lower_bound(all(sum), s - ss);
if (itr == sum.end() || *itr != s - ss) {
return;
}
ans += c[itr - sum.begin()];
return;
}
func2(i + 1, ss);
func2(i + 1, ss + fds[i + mid]);
};
func1(0, 0);
sortv(count);
ll prev = count[0];
int co = 1;
nrep(i, 1, count.size()) {
if (prev == count[i]) {
co++;
continue;
}
sum.push_back(prev);
c.push_back(co);
prev = count[i];
co = 1;
}
sum.push_back(prev);
c.push_back(co);
func2(0, 0);
cout << ans << '\n';
}

131
CSES Problem Set/Message Route.cpp Normal file
View File

@@ -0,0 +1,131 @@
/* Problem URL: https://cses.fi/problemset/task/1667 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m;
cin >> n >> m;
vvi graph(n);
while (m--) {
int a, b;
cin >> a >> b;
a--, b--;
graph[a].push_back(b);
graph[b].push_back(a);
}
vi p(n, -1);
auto bfs = [&]() {
p[0] = 0;
queue<int> q;
q.push(0);
while (!q.empty()) {
int i = q.front();
q.pop();
for (auto j : graph[i]) {
if (p[j] != -1) {
continue;
}
p[j] = i;
q.push(j);
}
}
};
bfs();
if (p[n - 1] == -1) {
cout << "IMPOSSIBLE\n";
return 0;
}
vi ans;
ans.push_back(n);
int i = p[n - 1];
while (i != 0) {
ans.push_back(i + 1);
i = p[i];
}
ans.push_back(1);
reverse(all(ans));
cout << ans.size() << '\n' << ans;
}

134
CSES Problem Set/Mex Grid Construction.cpp Normal file
View File

@@ -0,0 +1,134 @@
/* Problem URL: https://cses.fi/problemset/task/3419 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
vvi ans(n, vi(n));
ans[0][0] = 0;
V<V<set<int>>> mexleft(n, V<set<int>>(n));
V<V<set<int>>> mexup(n, V<set<int>>(n));
mexleft[0][0].insert(0);
mexup[0][0].insert(0);
auto find_mex = [](set<int> &a) {
int now = 0;
auto itr = a.begin();
while (itr != a.end() && now == *itr) {
itr++;
now++;
}
return now;
};
rep(i, n) {
rep(j, n) {
if (i == 0 && j == 0) {
continue;
}
if (i > 0) {
swap(mexup[i][j], mexup[i - 1][j]);
}
if (j > 0) {
swap(mexleft[i][j], mexleft[i][j - 1]);
}
set<int> tmp;
repv(k, mexup[i][j]) {
tmp.insert(k);
}
repv(k, mexleft[i][j]) {
tmp.insert(k);
}
int mex = find_mex(tmp);
ans[i][j] = mex;
mexup[i][j].insert(mex);
mexleft[i][j].insert(mex);
}
}
rep(i, n) {
rep(j, n) {
cout << ans[i][j] << " \n"[j == n - 1];
}
}
}

125
CSES Problem Set/Minimal Grid Path.cpp Normal file
View File

@@ -0,0 +1,125 @@
/* Problem URL: https://cses.fi/problemset/task/3359 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
V<string> fds(n);
cin >> fds;
string ans(2 * n - 1, 'Z' + 1);
queue<pair<int, int>> q;
V<V<bool>> vis(n, V<bool>(n, false));
vis[0][0] = true;
q.emplace(0, 0);
queue<pair<int, int>> tmp;
rep(_, n * 2 - 1) {
while (!q.empty()) {
auto [i, j] = q.front();
q.pop();
int pos = i + j;
rmin(ans[pos], fds[i][j]);
tmp.emplace(i, j);
}
while (!tmp.empty()) {
auto [i, j] = tmp.front();
tmp.pop();
if (ans[i + j] < fds[i][j]) {
continue;
}
int add[][2] = {{0, 1}, {1, 0}};
for (auto &k : add) {
int id = i + k[0];
int jd = j + k[1];
if (id < n && jd < n && !vis[id][jd]) {
vis[id][jd] = true;
q.emplace(id, jd);
}
}
}
}
cout << ans << '\n';
}

95
CSES Problem Set/Missing Coin Sum.cpp Normal file
View File

@@ -0,0 +1,95 @@
/* Problem URL: https://cses.fi/problemset/task/2183 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (size_t i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (size_t i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
vl fds(n);
cin >> fds;
sortv(fds);
if (fds[0] > 1) {
cout << "1\n";
return 0;
}
ll sum = 1;
for (size_t i = 1; i < n && fds[i] <= sum + 1; i++) {
sum += fds[i];
}
cout << sum + 1 << '\n';
}

196
CSES Problem Set/Monsters.cpp Normal file
View File

@@ -0,0 +1,196 @@
/* Problem URL: https://cses.fi/problemset/task/1194 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m;
cin >> n >> m;
V<string> fds(n);
cin >> fds;
int si, sj;
int inf = INT32_MAX >> 1;
queue<pair<int, int>> mq;
vvi mdis(n, vi(m, inf));
rep(i, n){
rep(j, m) {
if (fds[i][j] == 'A') {
si = i;
sj = j;
continue;
}
if (fds[i][j] == 'M') {
mq.emplace(i, j);
mdis[i][j] = 0;
}
}
}
auto mbfs = [&](){
while (!mq.empty()) {
auto [i, j] = mq.front();
mq.pop();
int add[][2] = {{0, 1}, {0, -1}, {-1, 0}, {1, 0}};
repv(k, add) {
int id = i + k[0];
int jd = j + k[1];
if (id >= 0 && id < n && jd >= 0 && jd < m && fds[id][jd] != '#' && mdis[id][jd] > mdis[i][j] + 1) {
mdis[id][jd] = mdis[i][j] + 1;
mq.emplace(id, jd);
}
}
}
};
mbfs();
vvi pdis(n, vi(m, inf));
auto bfs = [&](){
pdis[si][sj] = 0;
queue<pair<int, int>> q;
q.emplace(si, sj);
while (!q.empty()) {
auto [i, j] = q.front();
q.pop();
int add[][2] = {{0, 1}, {0, -1}, {-1, 0}, {1, 0}};
repv(k, add) {
int id = i + k[0];
int jd = j + k[1];
if (id >= 0 && id < n && jd >= 0 && jd < m && fds[id][jd] == '.' && pdis[id][jd] > pdis[i][j] + 1) {
pdis[id][jd] = pdis[i][j] + 1;
q.emplace(id, jd);
}
}
}
};
bfs();
string ans;
function<bool(int, int)> dfs = [&](int i, int j) {
if (i < 0 || i >= n || j < 0 || j >= m) {
return true;
}
if (pdis[i][j] >= mdis[i][j] || fds[i][j] == '#' || fds[i][j] == 'M') {
return false;
}
int add[][3] = {{0, 1, 'R'}, {0, -1, 'L'}, {-1, 0, 'U'}, {1, 0, 'D'}};
repv(k, add) {
int id = i + k[0];
int jd = j + k[1];
if (id >= 0 && id < n && jd >= 0 && jd < m && pdis[id][jd] <= pdis[i][j]) {
continue;
}
ans.push_back(k[2]);
if (dfs(id, jd)) {
return true;
}
ans.pop_back();
}
return false;
};
if (!dfs(si, sj)) {
cout << "NO\n";
return 0;
}
ans.pop_back();
cout << "YES\n" << ans.size() << '\n';
cout << ans << '\n';
}

190
CSES Problem Set/Mountain Range.cpp Normal file
View File

@@ -0,0 +1,190 @@
/* Problem URL: https://cses.fi/problemset/task/3314 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
vl fds(n);
cin >> fds;
while (__builtin_popcount(fds.size()) != 1) {
fds.push_back(0);
}
vi seg(fds.size() * 2, 1);
auto update = [&](int i, int v) {
seg[i + fds.size()] = v;
for (int j = (i + fds.size()) / 2; j > 0; j /= 2) {
seg[j] = max(seg[j * 2], seg[j * 2 + 1]);
}
};
function<int(int, int, int, int, int)> search = [&](int i, int l, int r, int tl, int tr) {
if (l > tr || r < tl) {
return 0;
}
if (l >= tl && r <= tr) {
return seg[i];
}
int mid = (l + r) / 2;
return max(search(i * 2, l, mid, tl, tr), search(i * 2 + 1, mid + 1, r, tl, tr));
};
priority_queue<pair<ll, ll>, V<pair<ll, ll>>, greater<>> order;
vvl sparse(n, vl(20));
rep(i, n) {
sparse[i][0] = fds[i];
}
nrep(j, 1, 20) {
for (int i = 0; i + (1 << (j - 1)) < n; i++) {
sparse[i][j] = max(sparse[i][j - 1], sparse[i + (1 << (j - 1))][j - 1]);
}
}
auto sparsequery = [&](int l, int r) {
if (l == r) {
return sparse[l][0];
}
int log = 31 - __builtin_clz(r - l + 1);
return max(sparse[l][log], sparse[r - (1 << log) + 1][log]);
};
rep(i, n) {
order.emplace(fds[i], i);
}
while (!order.empty()) {
int i = order.top().second;
order.pop();
int ans1 = 0;
int ans2 = 0;
if (i > 0 && fds[i - 1] < fds[i]) {
int ans = 0;
int low = 0;
int high = i - 1;
while (low <= high) {
int mid = (low + high) / 2;
if (sparsequery(mid, i - 1) >= fds[i]) {
ans = mid + 1;
low = mid + 1;
continue;
}
high = mid - 1;
}
if (ans != i) {
ans1 = search(1, 0, fds.size() - 1, ans, i - 1);
}
}
if (i < n - 1 && fds[i + 1] < fds[i]) {
int ans = n - 1;
int low = i + 1;
int high = n - 1;
while (low <= high) {
int mid = (low + high) / 2;
if (sparsequery(i + 1, mid) >= fds[i]) {
ans = mid - 1;
high = mid - 1;
continue;
}
low = mid + 1;
}
if (ans != i) {
ans2 = search(1, 0, fds.size() - 1, i + 1, ans);
}
}
update(i, max(ans1, ans2) + 1);
}
cout << seg[1] << '\n';
}

106
CSES Problem Set/Movie Festival II.cpp Normal file
View File

@@ -0,0 +1,106 @@
/* Problem URL: https://cses.fi/problemset/task/1632 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, k;
cin >> n >> k;
V<pair<ll, ll>> fds(n);
repv(i, fds) {
cin >> i.second >> i.first;
}
sortv(fds);
multiset<ll> ends;
rep(i, k) {
ends.insert(0);
}
int ans = 0;
repv(i, fds) {
auto itr = ends.upper_bound(i.second);
if (itr == ends.begin()) {
continue;
}
ans++;
ends.erase(prev(itr));
ends.insert(i.first);
}
cout << ans << '\n';
}

101
CSES Problem Set/Movie Festival.cpp Normal file
View File

@@ -0,0 +1,101 @@
/* Problem URL: https://cses.fi/problemset/task/1629 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (size_t i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (size_t i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
V<pair<ll, ll>> dura;
while (n--) {
ll a, b;
cin >> a >> b;
dura.emplace_back(b, a);
}
sortv(dura);
int ans = 1;
int nowe = dura[0].first;
nrep(i, 1, dura.size()) {
if (dura[i].second >= nowe) {
ans++;
nowe = dura[i].first;
}
}
cout << ans << '\n';
}

100
CSES Problem Set/Nearest Smaller Values.cpp Normal file
View File

@@ -0,0 +1,100 @@
/* Problem URL: https://cses.fi/problemset/task/1645 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
vl fds(n);
cin >> fds;
stack<pair<int, int>> s;
rep(i, fds.size()) {
while (!s.empty() && s.top().first >= fds[i]) {
s.pop();
}
if (s.empty()) {
cout << "0 ";
s.emplace(fds[i], i);
continue;
}
cout << s.top().second + 1 << ' ';
s.emplace(fds[i], i);
}
cout << '\n';
}

112
CSES Problem Set/Nested Ranges Check.cpp Normal file
View File

@@ -0,0 +1,112 @@
/* Problem URL: https://cses.fi/problemset/task/2168 */
#include <bits/stdc++.h>
#include <iterator>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (size_t i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (size_t i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
int prev = n;
V<pair<int, pair<ll, ll>>> ranges(n);
rep(i, n) {
cin >> ranges[i].second.first >> ranges[i].second.second;
ranges[i].first = i;
}
sort(all(ranges), [&](pair<int, pair<ll, ll>> &a, pair<int, pair<ll, ll>> &b){
if (a.second.first == b.second.first) {
return a.second.second > b.second.second;
}
return a.second.first < b.second.first;
});
V<bool> ans1(prev);
V<bool> ans2(prev);
ll minimus = INT64_MAX;
for (int i = n - 1; i >= 0; i--) {
ans1[ranges[i].first] = ranges[i].second.second >= minimus;
rmin(minimus, ranges[i].second.second);
}
ll maximus = 0;
rep(i, n) {
ans2[ranges[i].first] = ranges[i].second.second <= maximus;
rmax(maximus, ranges[i].second.second);
}
cout << ans1 << ans2;
}

141
CSES Problem Set/Nested Ranges Count.cpp Normal file
View File

@@ -0,0 +1,141 @@
/* Problem URL: https://cses.fi/problemset/task/2169 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (size_t i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (size_t i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
V<pair<int, pair<ll, ll>>> ranges(n);
map<ll, int> var;
rep(i, n) {
cin >> ranges[i].second.first >> ranges[i].second.second;
ranges[i].first = i;
var[ranges[i].second.first] = 0;
var[ranges[i].second.second] = 0;
}
int v = 0;
repv(i, var) {
i.second = v;
v++;
}
repv(i, ranges) {
i.second.first = var[i.second.first];
i.second.second = var[i.second.second];
}
sort(all(ranges), [&](pair<int, pair<ll, ll>> &a, pair<int, pair<ll, ll>> &b){
if (a.second.first == b.second.first) {
return a.second.second > b.second.second;
}
return a.second.first < b.second.first;
});
vi bit(var.size());
auto add = [&](int x){
for (; x < var.size(); x |= (x + 1)) {
bit[x]++;
}
};
auto get = [&](int x){
int ans = 0;
for (; x >= 0; x = (x & (x + 1)) - 1) {
ans += bit[x];
}
return ans;
};
vi ans1(n);
vi ans2(n);
for (int i = n - 1; i >= 0; i--) {
ans1[ranges[i].first] = get(ranges[i].second.second);
add(ranges[i].second.second);
}
fill(all(bit), 0);
rep(i, n) {
ans2[ranges[i].first] = get(var.size() - 1) - get(ranges[i].second.second - 1);
add(ranges[i].second.second);
}
cout << ans1 << ans2;
}

199
CSES Problem Set/New Roads Queries.cpp Normal file
View File

@@ -0,0 +1,199 @@
/* Problem URL: https://cses.fi/problemset/task/2101/ */
#include <bits/stdc++.h>
#include <algorithm>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (size_t i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (size_t i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m, q;
cin >> n >> m >> q;
V<tuple<int, int, int>> edges;
rep(i, m) {
int a, b;
cin >> a >> b;
a--, b--;
edges.emplace_back(i, a, b);
}
vi dsu(n);
rep(i, n) {
dsu[i] = i;
}
function<int(int)> find_p = [&](int i){
if (dsu[i] == i) {
return i;
}
return dsu[i] = find_p(dsu[i]);
};
auto make_pair = [&](int a, int b) {
a = find_p(a);
b = find_p(b);
if (a == b) {
return false;
}
dsu[b] = a;
return true;
};
V<V<pair<int, int>>> graph(n);
for (auto [i, a, b] : edges) {
if (make_pair(a, b)) {
graph[a].emplace_back(b, i + 1);
graph[b].emplace_back(a, i + 1);
}
}
vvi par(n, vi(20, 0));
vvi maximus(n, vi(20, 0));
vi depth(n, 0);
function<void(int, int, int)> dfs = [&](int i, int p, int big){
par[i][0] = p;
maximus[i][0] = big;
depth[i] = depth[p] + 1;
nrep(j, 1, 20) {
par[i][j] = par[par[i][j - 1]][j - 1];
maximus[i][j] = max(maximus[i][j - 1], maximus[par[i][j - 1]][j - 1]);
}
for (auto [u, v] : graph[i]) {
if (u == p) {
continue;
}
dfs(u, i, v);
}
};
auto lca = [&](int a, int b){
if (depth[a] > depth[b]) {
swap(a, b);
}
int ans = 0;
int diff = depth[b] - depth[a];
for (int i = 19; i >= 0; i--) {
if (diff & (1 << i)) {
rmax(ans, maximus[b][i]);
b = par[b][i];
}
}
if (a == b) {
return ans;
}
for (int i = 19; i >= 0; i--) {
if (par[a][i] != par[b][i]) {
rmax(ans, maximus[a][i]);
rmax(ans, maximus[b][i]);
a = par[a][i];
b = par[b][i];
}
}
return max({ans, maximus[a][0], maximus[b][0]});
};
rep(i, n) {
if (dsu[i] == i) {
fill(all(par[i]), i);
dfs(i, i, 0);
}
}
while (q--) {
int a, b;
cin >> a >> b;
a--, b--;
if (find_p(a) != find_p(b)) {
cout << "-1\n";
continue;
}
cout << lca(a, b) << '\n';
}
}

121
CSES Problem Set/Palindrome Reorder.cpp Normal file
View File

@@ -0,0 +1,121 @@
/* Problem URL: https://cses.fi/problemset/task/1755 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
string a;
cin >> a;
vi count(26);
repv(i, a) {
count[i - 'A']++;
}
int odds = 0;
int odd = -1;
rep(i, 26) {
if (count[i] & 1) {
odds++;
odd = i;
}
}
if (odds > 1) {
cout << "NO SOLUTION\n";
return 0;
}
if (odd != -1) {
count[odd]--;
}
rep(i, 26) {
rep(j, count[i] / 2) {
cout << (char)(i + 'A');
}
}
if (odd != -1) {
cout << (char)(odd + 'A');
}
for (int i = 25; i >= 0; i--) {
nrep(j, count[i] / 2, count[i]) {
cout << (char)(i + 'A');
}
}
cout << '\n';
}

166
CSES Problem Set/Planets Cycles.cpp Normal file
View File

@@ -0,0 +1,166 @@
/* Problem URL: https://cses.fi/problemset/task/1751 */
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;
template <class T, class comp = less<>>
using ordered_set = tree<T, null_type , comp , rb_tree_tag , tree_order_statistics_node_update>;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
vi next(n);
repv(i, next) {
cin >> i;
i--;
}
V<bool> vis(n);
V<bool> frame(n);
V<bool> cycle(n);
vi cyclesize(n);
auto getcycle = [&](int i) {
set<int> rem;
frame[i] = true;
vis[i] = true;
rem.insert(i);
int j = next[i];
while (!vis[j] && !frame[j]) {
vis[j] = true;
frame[j] = true;
rem.insert(j);
j = next[j];
}
if (!frame[j]) {
repv(i, rem) {
frame[i] = false;
}
return;
}
int now = 1;
i = j;
j = next[j];
cycle[i] = true;
while (j != i) {
cycle[j] = true;
now++;
j = next[j];
}
cyclesize[i] = now;
j = next[j];
while (j != i) {
cyclesize[j] = now;
j = next[j];
}
repv(i, rem) {
frame[i] = false;
}
};
rep(i, n) {
getcycle(i);
}
vi memo(n, -1);
rep(i, n) {
if (cycle[i]) {
memo[i] = cyclesize[i];
}
}
function<int(int)> dfs = [&](int i) {
int &ans = memo[i];
if (ans != -1) {
return ans;
}
ans = dfs(next[i]) + 1;
return ans;
};
rep(i, n) {
cout << dfs(i) << ' ';
}
cout << '\n';
}

116
CSES Problem Set/Planets Queries I.cpp Normal file
View File

@@ -0,0 +1,116 @@
/* Problem URL: https://cses.fi/problemset/task/1750 */
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;
template <class T, class comp = less<>>
using ordered_set = tree<T, null_type , comp , rb_tree_tag , tree_order_statistics_node_update>;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, q;
cin >> n >> q;
vvi parent(30, vi(n));
rep(i, n) {
int a;
cin >> a;
parent[0][i] = a - 1;
}
nrep(i, 1, 30) {
rep(j, n) {
parent[i][j] = parent[i - 1][parent[i - 1][j]];
}
}
while (q--) {
int i;
ll k;
cin >> i >> k;
i--;
rep(j, 30) {
if ((k >> j) & 1) {
i = parent[j][i];
}
}
cout << i + 1 << '\n';
}
}

240
CSES Problem Set/Planets Queries II.cpp Normal file
View File

@@ -0,0 +1,240 @@
/* Problem URL: https://cses.fi/problemset/task/1160 */
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;
template <class T, class comp = less<>>
using ordered_set = tree<T, null_type , comp , rb_tree_tag , tree_order_statistics_node_update>;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, q;
cin >> n >> q;
vi parent(n);
rep(i, n) {
int a;
cin >> a;
parent[i] = a - 1;
}
V<bool> loop(n);
vi group(n);
vi cyclepos(n);
V<bool> vis(n);
V<bool> frame(n);
vi groupsize;
vi cyclechoice(n);
int g = 0;
auto func = [&](int i) {
set<int> used;
while (!vis[i] && !frame[i]) {
vis[i] = true;
frame[i] = true;
used.insert(i);
i = parent[i];
}
if (frame[i]) {
loop[i] = true;
group[i] = g;
cyclechoice[i] = i;
int now = 1;
int j = parent[i];
while (j != i) {
cyclepos[j] = now;
now++;
loop[j] = true;
group[j] = g;
cyclechoice[j] = j;
j = parent[j];
}
g++;
groupsize.push_back(now);
}
for (auto j : used) {
frame[j] = false;
}
};
rep(i, n) {
if (vis[i]) {
continue;
}
func(i);
}
fill(all(vis), false);
vi pos(n);
function<void(int)> dfs = [&](int i) {
vis[i] = true;
if (!vis[parent[i]] && !loop[parent[i]]) {
dfs(parent[i]);
}
pos[i] = pos[parent[i]] + 1;
cyclechoice[i] = cyclechoice[parent[i]];
};
rep(i, n) {
if (!loop[i] && !vis[i]) {
dfs(i);
}
}
vvi lift(17, vi(n));
vi valids;
rep(i, n) {
if (!loop[i]) {
valids.push_back(i);
lift[0][i] = parent[i];
}
}
nrep(i, 1, 17) {
repv(j, valids) {
lift[i][j] = lift[i - 1][lift[i - 1][j]];
}
}
while (q--) {
int a, b;
cin >> a >> b;
a--, b--;
if (loop[a] && loop[b]) {
if (group[a] != group[b]) {
cout << "-1\n";
continue;
}
int g = group[a];
cout << ((cyclepos[b] - cyclepos[a]) % groupsize[g] + groupsize[g]) % groupsize[g] << '\n';
continue;
}
if (loop[a] && !loop[b]) {
cout << "-1\n";
continue;
}
if (!loop[a] && loop[b]) {
int choice = cyclechoice[a];
if (group[choice] != group[b]) {
cout << "-1\n";
continue;
}
int g = group[choice];
cout << pos[a] + ((cyclepos[b] - cyclepos[choice]) % groupsize[g] + groupsize[g]) % groupsize[g] << '\n';
continue;
}
int diff = pos[a] - pos[b];
if (diff < 0) {
cout << "-1\n";
continue;
}
rep(i, 17) {
if ((diff >> i) & 1) {
a = lift[i][a];
}
}
if (a != b) {
cout << "-1\n";
continue;
}
cout << diff << '\n';
}
}

154
CSES Problem Set/Planets and Kingdoms.cpp Normal file
View File

@@ -0,0 +1,154 @@
/* Problem URL: https://cses.fi/problemset/task/1683 */
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;
template <class T, class comp = less<>>
using ordered_set = tree<T, null_type , comp , rb_tree_tag , tree_order_statistics_node_update>;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m;
cin >> n >> m;
vvi graph(n);
vvi inv(n);
while (m--) {
int a, b;
cin >> a >> b;
a--, b--;
graph[a].push_back(b);
inv[b].push_back(a);
}
V<bool> vis(n);
stack<int> s;
function<void(int)> postorder = [&](int i) {
vis[i] = true;
for (auto j : graph[i]) {
if (vis[j]) {
continue;
}
postorder(j);
}
s.push(i);
};
rep(i, n) {
if (vis[i]) {
continue;
}
postorder(i);
}
vi group(n);
int g = 1;
fill(all(vis), false);
function<void(int)> dfs = [&](int i) {
vis[i] = true;
group[i] = g;
for (auto j : inv[i]) {
if (vis[j]) {
continue;
}
dfs(j);
}
};
while (!s.empty()) {
int now = s.top();
s.pop();
if (vis[now]) {
continue;
}
dfs(now);
g++;
}
cout << g - 1 << '\n';
cout << group;
}

101
CSES Problem Set/Playlist.cpp Normal file
View File

@@ -0,0 +1,101 @@
/* Problem URL: https://cses.fi/problemset/task/1141 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (size_t i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (size_t i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
map<ll, int> appear;
int n;
cin >> n;
int ans = 0;
int now = 0;
int time = 0;
rep(i, n) {
ll nm;
cin >> nm;
now++;
if (appear.count(nm) && appear[nm] >= time) {
now -= appear[nm] - time + 1;
time = appear[nm] + 1;
}
rmax(ans, now);
appear[nm] = i;
}
cout << ans << '\n';
}

174
CSES Problem Set/Police Chase.cpp Normal file
View File

@@ -0,0 +1,174 @@
/* Problem URL: https://cses.fi/problemset/task/1695 */
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;
template <class T, class comp = less<>>
using ordered_set = tree<T, null_type , comp , rb_tree_tag , tree_order_statistics_node_update>;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
const int INF = INT32_MAX >> 1;
struct dinitz {
const bool scaling = false; // com scaling -> O(nm log(MAXCAP)),
int lim; // com constante alta
struct edge {
int to, cap, rev, flow;
bool res;
edge(int to_, int cap_, int rev_, bool res_)
: to(to_), cap(cap_), rev(rev_), flow(0), res(res_) {}
};
vector<vector<edge>> g;
vector<int> lev, beg;
ll F;
dinitz(int n) : g(n), F(0) {}
void add(int a, int b, int c) {
g[a].emplace_back(b, c, g[b].size(), false);
g[b].emplace_back(a, 0, g[a].size()-1, true);
}
bool bfs(int s, int t) {
lev = vector<int>(g.size(), -1); lev[s] = 0;
beg = vector<int>(g.size(), 0);
queue<int> q; q.push(s);
while (q.size()) {
int u = q.front(); q.pop();
for (auto& i : g[u]) {
if (lev[i.to] != -1 or (i.flow == i.cap)) continue;
if (scaling and i.cap - i.flow < lim) continue;
lev[i.to] = lev[u] + 1;
q.push(i.to);
}
}
return lev[t] != -1;
}
int dfs(int v, int s, int f = INF) {
if (!f or v == s) return f;
for (int& i = beg[v]; i < g[v].size(); i++) {
auto& e = g[v][i];
if (lev[e.to] != lev[v] + 1) continue;
int foi = dfs(e.to, s, min(f, e.cap - e.flow));
if (!foi) continue;
e.flow += foi, g[e.to][e.rev].flow -= foi;
return foi;
}
return 0;
}
ll max_flow(int s, int t) {
for (lim = scaling ? (1<<30) : 1; lim; lim /= 2)
while (bfs(s, t)) while (int ff = dfs(s, t)) F += ff;
return F;
}
};
// Recupera as arestas do corte s-t
vector<pair<int, int>> get_cut(dinitz& g, int s, int t) {
g.max_flow(s, t);
vector<pair<int, int>> cut;
vector<int> vis(g.g.size(), 0), st = {s};
vis[s] = 1;
while (st.size()) {
int u = st.back(); st.pop_back();
for (auto e : g.g[u]) if (!vis[e.to] and e.flow < e.cap)
vis[e.to] = 1, st.push_back(e.to);
}
for (int i = 0; i < g.g.size(); i++) for (auto e : g.g[i])
if (vis[i] and !vis[e.to] and !e.res) cut.emplace_back(i, e.to);
return cut;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m;
cin >> n >> m;
dinitz din(n);
while (m--) {
int a, b;
cin >> a >> b;
a--, b--;
din.add(a, b, 1);
din.add(b, a, 1);
}
V<pair<int, int>> ans = get_cut(din, 0, n - 1);
cout << ans.size() << '\n';
for (auto [a, b] : ans) {
cout << a + 1 << ' ' << b + 1 << '\n';
}
}

94
CSES Problem Set/Projects.cpp
View File

@@ -1,9 +1,7 @@
/* Problem URL: https://cses.fi/problemset/task/1140 */ /* Problem URL: https://cses.fi/problemset/task/1140 */
#include <bits/stdc++.h> #include <bits/stdc++.h>
using namespace std; using namespace std;
#define V vector #define V vector
#define rmin(a, b) a = min(a, b) #define rmin(a, b) a = min(a, b)
@@ -69,52 +67,6 @@ auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
return is; return is;
} }
class segtree {
vl seg;
ll _query(int i, int l, int r, int ql, int qr) {
if (l > qr || r < ql) {
return -1;
}
if (l >= ql && r <= qr) {
return seg[i];
}
int mid = (l + r) / 2;
return max(_query(i * 2, l, mid, ql, qr), _query(i * 2 + 1, mid + 1, r, ql, qr));
}
public:
segtree() = delete;
segtree(vl &items): seg(items.size() * 2) {
size_t n = items.size();
for (size_t i = n; i < n * 2; i++) {
seg[i] = items[i - n];
}
for (size_t i = n - 1; i > 0; i--) {
seg[i] = max(seg[i * 2], seg[i * 2 + 1]);
}
}
void update(ll v, int i) {
i += seg.size() / 2;
if (seg[i] >= v) {
return;
}
seg[i] = v;
i /= 2;
for (; i > 0; i /= 2) {
seg[i] = max(seg[i * 2], seg[i * 2 + 1]);
}
}
ll query(int i) {
return _query(1, 0, seg.size() / 2 - 1, 0, i);
}
};
int main() int main()
{ {
ios::sync_with_stdio(false); ios::sync_with_stdio(false);
@@ -122,39 +74,33 @@ int main()
int n; int n;
cin >> n; cin >> n;
V<tuple<ll, ll, ll>> fds(n); V<tuple<ll, ll, ll>> q(n);
set<ll> lol;
for (auto &[i, e, v] : fds) { map<ll, int> var;
cin >> i >> e >> v; for (auto &[x, y, z] : q) {
lol.insert(e); cin >> x >> y >> z;
var[x] = 0;
var[y] = 0;
} }
pair<vl, vl> ends(vl(lol.size(), 0), vl(lol.size())); int v = 0;
auto itr = lol.begin(); for (auto &i : var) {
for (size_t i = 0; itr != lol.end(); i++, itr++) { i.second = v;
ends.second[i] = *itr; v++;
} }
sort(fds.begin(), fds.end()); V<V<pair<int, ll>>> proj(var.size());
for (auto [s, e, v] : q) {
ll neutral = 1e9 + 1; proj[var[e]].emplace_back(var[s], v);
while (__builtin_popcount(ends.second.size()) != 1) {
ends.first.push_back(-1);
ends.second.push_back(neutral++);
} }
segtree seg(ends.first); vl dp(var.size() + 1);
nrep(i, 1, dp.size()) {
for (auto [i, f, v] : fds) { dp[i] += dp[i - 1];
auto itr = lower_bound(ends.second.begin(), ends.second.end(), i); for (auto [s, v] : proj[i - 1]) {
if (itr == ends.second.begin()) { rmax(dp[i], dp[s] + v);
seg.update(v, lower_bound(ends.second.begin(), ends.second.end(), f) - ends.second.begin());
continue;
} }
ll up = seg.query(itr - ends.second.begin() - 1) + v;
seg.update(up, lower_bound(ends.second.begin(), ends.second.end(), f) - ends.second.begin());
} }
cout << seg.query(ends.first.size() - 1) << '\n'; cout << dp.back() << '\n';
} }

116
CSES Problem Set/Raab Game I.cpp Normal file
View File

@@ -0,0 +1,116 @@
/* Problem URL: https://cses.fi/problemset/task/3399 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int t;
cin >> t;
while (t--) {
int n, a, b;
cin >> n >> a >> b;
vi ans1(n);
rep(i, n) {
ans1[i] = i + 1;
}
if (a == 0 && b == 0) {
cout << "YES\n" << ans1 << ans1;
continue;
}
if (a + b == 1 || a == 0 || b == 0 || a + b > n) {
cout << "NO\n";
continue;
}
vi ans2(n);
int i = 0;
for (; i < b; i++) {
ans2[i] = i + a + 1;
}
for (; i < b + a; i++) {
ans2[i] = i - b + 1;
}
for (; i < n; i++) {
ans2[i] = ans1[i];
}
cout << "YES\n";
cout << ans1 << ans2;
}
}

93
CSES Problem Set/Reading Books.cpp Normal file
View File

@@ -0,0 +1,93 @@
/* Problem URL: https://cses.fi/problemset/task/1631 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (size_t i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (size_t i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
vl fds(n);
cin >> fds;
sortv(fds);
ll sum = 0;
vl pref(n + 1);
rep(i, n) {
pref[i + 1] = pref[i] + fds[i];
}
cout << (fds.back() >= pref.back() - fds.back() ? fds.back() * 2 : pref.back()) << '\n';
}

109
CSES Problem Set/Restaurant Customers.cpp Normal file
View File

@@ -0,0 +1,109 @@
/* Problem URL: https://cses.fi/problemset/task/1619 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (size_t i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (size_t i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
V<pair<int, int>> query(n);
map<int, int> var;
for (auto &[a, b] : query) {
cin >> a >> b;
var[a] = 0;
var[b] = 0;
}
int v = 0;
for (auto &i : var) {
i.second = v;
v++;
}
vi ans(var.size() + 1, 0);
for (auto [a, b] : query) {
ans[var[a]]++;
ans[var[b] + 1]--;
}
int maximus = 0;
int count = 0;
repv(i, ans) {
count += i;
rmax(maximus, count);
}
cout << maximus << '\n';
}

126
CSES Problem Set/Room Allocation.cpp Normal file
View File

@@ -0,0 +1,126 @@
/* Problem URL: https://cses.fi/problemset/task/1164 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (size_t i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (size_t i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>; using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
V<tuple<int, ll, ll>> cus(n);
map<ll, int> var;
size_t ind = 0;
for (auto &[i, a, b] : cus) {
cin >> a >> b;
var[a] = 0;
var[b] = 0;
i = ind;
ind++;
}
int v = 0;
for (auto &i : var) {
i.second = v;
v++;
}
V<tuple<ll, int, int>> act;
for (auto &[i, a, b] : cus) {
a = var[a];
b = var[b];
act.emplace_back(a, 0, i);
act.emplace_back(b, 1, i);
}
int k = 1;
priority_queue<int, vi, greater<>> pos;
for (size_t i = 1; i <= n; i++) {
pos.push(i);
}
vi ans(n);
sortv(act);
for (auto [a, yes, i] : act) {
if (!yes) {
ans[i] = pos.top();
pos.pop();
rmax(k, ans[i]);
} else {
pos.push(ans[i]);
}
}
cout << k << '\n';
cout << ans;
}

151
CSES Problem Set/Round Trip II.cpp Normal file
View File

@@ -0,0 +1,151 @@
/* Problem URL: https://cses.fi/problemset/task/1678 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m;
cin >> n >> m;
vvi graph(n);
while (m--) {
int a, b;
cin >> a >> b;
a--, b--;
graph[a].push_back(b);
}
V<bool> vis(n);
V<bool> frame(n);
vi parent(n, -1);
function<pair<int, int>(int)> dfs = [&](int i) -> pair<int, int> {
vis[i] = true;
frame[i] = true;
for (auto j : graph[i]) {
if (frame[j]) {
return {i, j};
}
if (vis[j]) {
continue;
}
parent[j] = i;
auto res = dfs(j);
if (res != make_pair(-1, -1)) {
return res;
}
}
frame[i] = false;
return {-1, -1};
};
pair<int, int> res = {-1, -1};
rep(i, n) {
if (!vis[i]) {
res = dfs(i);
if (res != make_pair(-1, -1)) {
break;
}
res = {-1, -1};
}
}
if (res == make_pair(-1, -1)) {
cout << "IMPOSSIBLE\n";
return 0;
}
vi ans;
ans.push_back(res.first + 1);
int tmp = parent[res.first];
while (tmp != res.second) {
ans.push_back(tmp + 1);
tmp = parent[tmp];
}
ans.push_back(res.second + 1);
ans.push_back(res.first + 1);
reverse(all(ans));
cout << ans.size() << '\n' << ans;
}

142
CSES Problem Set/Round Trip.cpp Normal file
View File

@@ -0,0 +1,142 @@
/* Problem URL: https://cses.fi/problemset/task/1669 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m;
cin >> n >> m;
vvi graph(n);
while (m--) {
int a, b;
cin >> a >> b;
a--, b--;
graph[a].push_back(b);
graph[b].push_back(a);
}
vi parent(n, -1);
V<bool> frame(n, false);
vi ans;
function<bool(int, int)> dfs = [&](int i, int p) {
frame[i] = true;
repv(j, graph[i]) {
if (frame[j] && j != p) {
ans.push_back(i + 1);
int p = parent[i];
while (p != j) {
ans.push_back(p + 1);
p = parent[p];
}
ans.push_back(j + 1);
ans.push_back(i + 1);
return true;
}
if (parent[j] != -1) {
continue;
}
parent[j] = i;
if (dfs(j, i)) {
return true;
}
}
frame[i] = false;
return false;
};
rep(i, n) {
if (parent[i] != -1) {
continue;
}
if (dfs(i, i)) {
break;
}
}
if (ans.empty()) {
cout << "IMPOSSIBLE\n";
return 0;
}
cout << ans.size() << '\n' << ans;
}

118
CSES Problem Set/Salary Queries.cpp Normal file
View File

@@ -0,0 +1,118 @@
/* Problem URL: https://cses.fi/problemset/task/1144 */
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;
template <class T, class comp = less<>>
using ordered_set = tree<T, null_type , comp , rb_tree_tag , tree_order_statistics_node_update>;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, q;
cin >> n >> q;
ordered_set<pair<int, int>> ord;
vi sal(n);
rep(i, n) {
cin >> sal[i];
ord.insert({sal[i], i});
}
while (q--) {
char op;
cin >> op;
if (op == '?') {
int a, b;
cin >> a >> b;
int lower = ord.order_of_key({a, -1});
cout << ord.order_of_key({b + 1, -1}) - lower << '\n';
continue;
}
int i, x;
cin >> i >> x;
i--;
ord.erase({sal[i], i});
sal[i] = x;
ord.insert({sal[i], i});
}
}

198
CSES Problem Set/School Dance.cpp Normal file
View File

@@ -0,0 +1,198 @@
/* Problem URL: https://cses.fi/problemset/task/1696 */
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;
template <class T, class comp = less<>>
using ordered_set = tree<T, null_type , comp , rb_tree_tag , tree_order_statistics_node_update>;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
const int INF = INT32_MAX >> 1;
struct dinitz {
const bool scaling = false; // com scaling -> O(nm log(MAXCAP)),
int lim; // com constante alta
struct edge {
int to, cap, rev, flow;
bool res;
edge(int to_, int cap_, int rev_, bool res_)
: to(to_), cap(cap_), rev(rev_), flow(0), res(res_) {}
};
vector<vector<edge>> g;
vector<int> lev, beg;
ll F;
dinitz(int n) : g(n), F(0) {}
void add(int a, int b, int c) {
g[a].emplace_back(b, c, g[b].size(), false);
g[b].emplace_back(a, 0, g[a].size()-1, true);
}
bool bfs(int s, int t) {
lev = vector<int>(g.size(), -1); lev[s] = 0;
beg = vector<int>(g.size(), 0);
queue<int> q; q.push(s);
while (q.size()) {
int u = q.front(); q.pop();
for (auto& i : g[u]) {
if (lev[i.to] != -1 or (i.flow == i.cap)) continue;
if (scaling and i.cap - i.flow < lim) continue;
lev[i.to] = lev[u] + 1;
q.push(i.to);
}
}
return lev[t] != -1;
}
int dfs(int v, int s, int f = INF) {
if (!f or v == s) return f;
for (int& i = beg[v]; i < g[v].size(); i++) {
auto& e = g[v][i];
if (lev[e.to] != lev[v] + 1) continue;
int foi = dfs(e.to, s, min(f, e.cap - e.flow));
if (!foi) continue;
e.flow += foi, g[e.to][e.rev].flow -= foi;
return foi;
}
return 0;
}
ll max_flow(int s, int t) {
for (lim = scaling ? (1<<30) : 1; lim; lim /= 2)
while (bfs(s, t)) while (int ff = dfs(s, t)) F += ff;
return F;
}
};
// Recupera as arestas do corte s-t
vector<pair<int, int>> get_cut(dinitz& g, int s, int t) {
g.max_flow(s, t);
vector<pair<int, int>> cut;
vector<int> vis(g.g.size(), 0), st = {s};
vis[s] = 1;
while (st.size()) {
int u = st.back(); st.pop_back();
for (auto e : g.g[u]) if (!vis[e.to] and e.flow < e.cap)
vis[e.to] = 1, st.push_back(e.to);
}
for (int i = 0; i < g.g.size(); i++) for (auto e : g.g[i])
if (vis[i] and !vis[e.to] and !e.res) cut.emplace_back(i, e.to);
return cut;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m, k;
cin >> n >> m >> k;
dinitz din(n + m + 2);
int s = 0;
int t = n + m + 1;
rep(i, n) {
din.add(s, i + 1, 1);
}
rep(i, m) {
din.add(i + n + 1, t, 1);
}
while (k--) {
int a, b;
cin >> a >> b;
b += n;
din.add(a, b, 1);
}
int ans = din.max_flow(s, t);
cout << ans << '\n';
rep(i, n) {
ans = -1;
for (auto edge : din.g[i + 1]) {
if (edge.res || edge.flow == 0) {
continue;
}
ans = edge.to - n;
}
if (ans == -1) {
continue;
}
cout << i + 1 << ' ' << ans << '\n';
}
}

123
CSES Problem Set/Shortest Routes I.cpp Normal file
View File

@@ -0,0 +1,123 @@
/* Problem URL: https://cses.fi/problemset/task/1671 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, m;
cin >> n >> m;
V<V<pair<int, ll>>> graph(n);
while (m--) {
int a, b, c;
cin >> a >> b >> c;
a--, b--;
graph[a].emplace_back(b, c);
}
ll inf = INT64_MAX >> 1;
vl dis(n, inf);
auto dijkstra = [&]() {
dis[0] = 0;
priority_queue<pair<ll, int>, V<pair<ll, int>>, greater<>> pq;
pq.emplace(0, 0);
while (!pq.empty()) {
auto [c, i] = pq.top();
pq.pop();
if (c > dis[i]) {
continue;
}
for (auto [j, cc] : graph[i]) {
if (dis[j] > c + cc) {
dis[j] = c + cc;
pq.emplace(dis[j], j);
}
}
}
};
dijkstra();
rep(i, n) {
cout << dis[i] << ' ';
}
cout << '\n';
}

42
CSES Problem Set/Shortest Routes II.cpp
View File

@@ -9,8 +9,8 @@ using namespace std;
#define rmin(a, b) a = min(a, b) #define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b) #define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (size_t i = 0; i < (lim); i++) #define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (size_t i = s; i < (lim); i++) #define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v)) #define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; } #define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
@@ -77,40 +77,40 @@ int main()
int n, m, q; int n, m, q;
cin >> n >> m >> q; cin >> n >> m >> q;
vvl weight(n, vl(n, INT64_MAX >> 1)); ll inf = INT64_MAX >> 1;
vvl floyd(n, vl(n)); vvl graph(n, vl(n, inf));
for (size_t i = 0; i < n; i++) { rep(i, n) {
weight[i][i] = 0; graph[i][i] = 0;
} }
while (m--) { while (m--) {
ll a, b, c; ll a, b, c;
cin >> a >> b >> c; cin >> a >> b >> c;
a--, b--; a--, b--;
rmin(weight[a][b], c);
rmin(weight[b][a], c); rmin(graph[a][b], c);
rmin(graph[b][a], c);
} }
for (size_t i = 0; i < n; i++) { rep(k, n) {
for (size_t j = 0; j < n; j++) { rep(i, n) {
floyd[i][j] = weight[i][j]; rep(j, n) {
} rmin(graph[i][j], graph[i][k] + graph[k][j]);
}
for (size_t i = 0; i < n; i++) {
for (size_t j = 0; j < n; j++) {
for (size_t k = 0; k < n; k++) {
rmin(floyd[j][k], floyd[j][i] + floyd[i][k]);
} }
} }
} }
while (q--) { while (q--) {
ll a, b; int a, b;
cin >> a >> b; cin >> a >> b;
a--, b--; a--, b--;
cout << (floyd[a][b] != INT64_MAX >> 1 ? floyd[a][b] : -1) << '\n'; if (graph[a][b] == inf) {
cout << "-1\n";
continue;
}
cout << graph[a][b] << '\n';
} }
} }

104
CSES Problem Set/Sliding Window Sum.cpp Normal file
View File

@@ -0,0 +1,104 @@
/* Problem URL: https://cses.fi/problemset/task/3220 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, k;
cin >> n >> k;
ll x, a, b, c;
cin >> x >> a >> b >> c;
list<ll> fds;
fds.push_back(x);
ll total = x;
nrep(i, 1, k) {
fds.push_back((fds.back() * a + b) % c);
total += fds.back();
}
ll ans = total;
n -= k;
rep(i, n) {
fds.push_back((fds.back() * a + b) % c);
total += fds.back();
total -= fds.front();
fds.pop_front();
ans ^= total;
}
cout << ans << '\n';
}

100
CSES Problem Set/Stick Lengths.cpp Normal file
View File

@@ -0,0 +1,100 @@
/* Problem URL: https://cses.fi/problemset/task/1074 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (size_t i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (size_t i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
vl fds(n);
cin >> fds;
sortv(fds);
vl pref(n);
vl suf(n);
nrep(i, 1, n) {
pref[i] = pref[i - 1] + abs(fds[i] - fds[i - 1]) * i;
suf[n - i - 1] = suf[n - i] + abs(fds[n - i - 1] - fds[n - i]) * i;
}
ll ans = INT64_MAX >> 1;
rep(i, n) {
rmin(ans, pref[i] + suf[i]);
}
cout << ans << '\n';
}

125
CSES Problem Set/String Matching.cpp Normal file
View File

@@ -0,0 +1,125 @@
/* Problem URL: https://cses.fi/problemset/task/1753 */
#include <bits/stdc++.h>
#include <random>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (size_t i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (size_t i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
struct hash_string {
vl pow;
vl hash;
static ll base;
static ll mod;
hash_string() = delete;
hash_string(string &a) {
pow.resize(a.size() + 1);
hash.resize(a.size() + 1);
pow[0] = 1;
hash[0] = 0;
rep(i, a.size()) {
pow[i + 1] = ((__int128)pow[i] * base) % mod;
hash[i + 1] = ((__int128)hash[i] * base + a[i]) % mod;
}
}
ll gethash(int i, int f)
{
return ((hash[f + 1] - ((__int128)hash[i] * pow[f - i + 1])) % mod + mod) % mod;
}
};
ll hash_string::mod = (1LL << 61) - 1;
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
ll hash_string::base = uniform_int_distribution<ll>(0, (1LL << 61) - 1)(rng);
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
string a, b;
cin >> a >> b;
hash_string ahash(a);
hash_string bhash(b);
ll hash = bhash.gethash(0, b.size() - 1);
ll count = 0;
for (size_t i = 0; i + b.size() - 1 < a.size(); i++) {
if (hash == ahash.gethash(i, i + b.size() - 1)) {
count++;
}
}
cout << count << '\n';
}

120
CSES Problem Set/String Reorder.cpp Normal file
View File

@@ -0,0 +1,120 @@
/* Problem URL: https://cses.fi/problemset/task/1743 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
string fds;
cin >> fds;
vi count(26);
repv(i, fds) {
count[i - 'A']++;
}
string ans(fds.size(), ' ');
rep(i, ans.size()) {
int now = -1;
auto valid = [&](int act){
int maximal = 0;
int size = 0;
rep(i, 26) {
if (i == act) {
rmax(maximal, count[i] - 1);
size += count[i] - 1;
} else {
rmax(maximal, count[i]);
size += count[i];
}
}
return maximal <= size - maximal + 1;
};
rep(j, 26) {
if (count[j] != 0 && (i == 0 || j + 'A' != ans[i - 1]) && valid(j)) {
now = j;
count[j]--;
break;
}
}
if (now == -1) {
cout << "-1\n";
return 0;
}
ans[i] = now + 'A';
}
cout << ans << '\n';
}

126
CSES Problem Set/Subarray Distinct Values.cpp Normal file
View File

@@ -0,0 +1,126 @@
/* Problem URL: https://cses.fi/problemset/task/2428 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n, k;
cin >> n >> k;
map<ll, int> var;
vl fds(n);
repv(i, fds){
cin >> i;
var[i] = 0;
}
int v = 0;
repv(i, var) {
i.second = v;
v++;
}
repv(i, fds) {
i = var[i];
}
vi count(var.size(), 0);
int l = 0;
int r = 0;
ll ans = 0;
int diffs = 0;
while (r < n) {
if (count[fds[r]] == 0) {
diffs++;
}
count[fds[r]]++;
while (diffs > k) {
count[fds[l]]--;
if (count[fds[l]] == 0) {
diffs--;
}
l++;
}
ans += r - l + 1;
r++;
}
cout << ans << '\n';
}

95
CSES Problem Set/Subarray Divisibility.cpp Normal file
View File

@@ -0,0 +1,95 @@
/* Problem URL: https://cses.fi/problemset/task/1662 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
ll sum = 0;
ll ans = 0;
map<ll, ll> count;
count[0] = 1;
rep(i, n) {
ll now;
cin >> now;
sum = ((sum + now) % n + n) % n;
ans += count[sum];
count[sum]++;
}
cout << ans << '\n';
}

96
CSES Problem Set/Subarray Sums I.cpp Normal file
View File

@@ -0,0 +1,96 @@
/* Problem URL: https://cses.fi/problemset/task/1660 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
ll x;
cin >> n >> x;
map<ll, int> count;
ll sum = 0;
int ans = 0;
count[0] = 1;
rep(i, n) {
ll now;
cin >> now;
sum += now;
ans += count[sum - x];
count[sum]++;
}
cout << ans << '\n';
}

96
CSES Problem Set/Subarray Sums II.cpp Normal file
View File

@@ -0,0 +1,96 @@
/* Problem URL: https://cses.fi/problemset/task/1660 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (int i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (int i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
ll x;
cin >> n >> x;
map<ll, ll> count;
ll sum = 0;
ll ans = 0;
count[0] = 1;
rep(i, n) {
ll now;
cin >> now;
sum += now;
ans += count[sum - x];
count[sum]++;
}
cout << ans << '\n';
}

106
CSES Problem Set/Subordinates.cpp Normal file
View File

@@ -0,0 +1,106 @@
/* Problem URL: https://cses.fi/problemset/task/1674 */
#include <bits/stdc++.h>
using namespace std;
#define V vector
#define rmin(a, b) a = min(a, b)
#define rmax(a, b) a = max(a, b)
#define rep(i, lim) for (size_t i = 0; i < (lim); i++)
#define nrep(i, s, lim) for (size_t i = s; i < (lim); i++)
#define repv(i, v) for (auto &i : (v))
#define fillv(v) for (auto &itr_ : (v)) { cin >> itr_; }
#define sortv(v) sort(v.begin(), v.end())
#define all(v) (v).begin(), (v).end()
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using ll = long long;
using vl = vector<ll>;
using vvl = vector<vl>;
using vvvl = vector<vvl>;
using vvvvl = vector<vvvl>;
template<class v>
auto operator<<(ostream &os, const vector<v> &vec)->ostream& {
os << vec[0];
for (size_t i = 1; i < vec.size(); i++) {
os << ' ' << vec[i];
}
os << '\n';
return os;
}
template<class v>
auto operator>>(istream &is, vector<v> &vec)->istream& {
for (auto &i : vec) {
is >> i;
}
return is;
}
template<class v>
auto operator<<(ostream &os, const vector<vector<v>> &vec)->ostream& {
for (auto &i : vec) {
os << i[0];
for (size_t j = 1; j < i.size(); j++) {
os << ' ' << i[j];
}
os << '\n';
}
return os;
}
template<class v>
auto operator>>(istream &is, vector<vector<v>> &vec)->istream& {
for (auto &i : vec) {
for (auto &j : i) {
is >> j;
}
}
return is;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
vvi graph(n);
nrep(i, 1, n) {
int a;
cin >> a;
a--;
graph[a].push_back(i);
}
vi ans(n);
function<int(int)> dfs = [&](int i){
int tmp = 0;
for (auto j : graph[i]) {
tmp += dfs(j);
}
ans[i] = tmp;
return tmp + 1;
};
dfs(0);
cout << ans;
}

Some files were not shown because too many files have changed in this diff Show More