#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using ld = long double;
using uint = unsigned int;
using ull = unsigned long long;
template<typename T>
using pair2 = pair<T, T>;
using pii = pair<int, int>;
using pli = pair<ll, int>;
using pll = pair<ll, ll>;
#define pb push_back
#define pf push_front
#define mp make_pair
#define all(x) (x).begin(),(x).end()
#define fi first
#define se second
#define endl "\n"
#define in(x) cin >> x
#define in_ll(x) ll x; in(x)
#define in_i(x) int x; in(x)
#define in_str(x) string x; in(x)
//https://codeforces.com/blog/entry/62393
struct custom_hash {
static uint64_t splitmix64(uint64_t x) {
// http://xorshift.di.unimi.it/splitmix64.c
x += 0x9e3779b97f4a7c15;
x = (x ^ (x >> 30)) * 0xbf58476d1ce4e5b9;
x = (x ^ (x >> 27)) * 0x94d049bb133111eb;
return x ^ (x >> 31);
}
size_t operator()(uint64_t x) const {
static const uint64_t FIXED_RANDOM = chrono::steady_clock::now().time_since_epoch().count();
return splitmix64(x + FIXED_RANDOM);
}
};
//Można zauważyć, że poniższy warunek musi zostać spełniony, aby liny się przecięły:
//x1 - pozycja x na pierwszej wysokości
//x2 - pozycja x na drugiej wysokości
//x3 - pozycja x drugiej liny na pierwszej wysokości
//x4 - pozycja x drugiej liny na drugiej wysokości
//(x1 < x3 && x2 > x4) || (x1 > x3 && x2 < x4)
//Wiedząc to jesteśmy w stanie skonstruować graf w czasie O(n^2) poprzez
//iterację przez wszystkie pary lin i sprawdzenie powyższego warunku.
//Następnie stosując algorytm Floyda-Warshalla oraz iterację po rzędach
//otrzymanej tabelki znajdujemy odpowiedź w czasie O(n^3).
//Głównym problemem jest stosowanie algorytmu Floyda-Marshalla w celu znalezienia
//odpowiedzi. Wiadomo, że wszystkie krawędzie mają długość 1. W takim razie najkrótsza trasa to taka, która przechodzi
//przez najmniej wierzchołków. W związku z tym przy pomocy algorytmu BFS dla każdego
//wierzchołka możemy znaleźć długość najkrótszej ścieżki do każdego innego w czasie liniowym.
//Skoro szukamy sumy, to wystarczy przechowywać zmienną o nazwie sum podczas iteracji przez wierzchołki
//i dla każdego wierzchołka dodawać jego odległość do sum. Otrzymamy w ten sposób odpowiedź.
//Złożoność czasowa tego rozwiązania to O(n^2).
//Dla każdego x-a na pierwszej wysokości istnieje dokładnie jedna lina. Dokładnie to samo dla x-a na drugiej
//wysokości. Oznacza to, że liny nigdy nie będą przyczepione do tego samego punktu.
int bfs(int at, vector<vector<int>> &graph) {
vector<bool> vis(graph.size());
queue<pii> q;
q.push(mp(at,0));
int sum = 0;
while (!q.empty()) {
int next = q.front().first;
vis[next] = true;
int dist = q.front().second;
sum += dist;
q.pop();
for (int i : graph[next]) {
if (!vis[i]) {
vis[i] = true;
q.push(mp(i,dist + 1));
}
}
}
return sum;
}
int main() {
ios_base::sync_with_stdio(false);
cin.tie(0);
in_i(size);
vector<int> vec;
for (int i = 0; i < size; i++) {
in_i(a);
vec.pb(a);
}
vector<vector<int>> graph(size);
for (int x1 = 0; x1 < vec.size(); x1++) {
int x2 = vec[x1];
for (int x3 = x1 + 1; x3 < vec.size(); x3++) {
int x4 = vec[x3];
if (x2 > x4) {
graph[x1].pb(x3);
graph[x3].pb(x1);
}
}
}
for (int i = 0; i < size; i++) {
cout << bfs(i, graph) << " ";
}
cout << endl;
}
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 | #include <bits/stdc++.h> using namespace std; using ll = long long; using ld = long double; using uint = unsigned int; using ull = unsigned long long; template<typename T> using pair2 = pair<T, T>; using pii = pair<int, int>; using pli = pair<ll, int>; using pll = pair<ll, ll>; #define pb push_back #define pf push_front #define mp make_pair #define all(x) (x).begin(),(x).end() #define fi first #define se second #define endl "\n" #define in(x) cin >> x #define in_ll(x) ll x; in(x) #define in_i(x) int x; in(x) #define in_str(x) string x; in(x) //https://codeforces.com/blog/entry/62393 struct custom_hash { static uint64_t splitmix64(uint64_t x) { // http://xorshift.di.unimi.it/splitmix64.c x += 0x9e3779b97f4a7c15; x = (x ^ (x >> 30)) * 0xbf58476d1ce4e5b9; x = (x ^ (x >> 27)) * 0x94d049bb133111eb; return x ^ (x >> 31); } size_t operator()(uint64_t x) const { static const uint64_t FIXED_RANDOM = chrono::steady_clock::now().time_since_epoch().count(); return splitmix64(x + FIXED_RANDOM); } }; //Można zauważyć, że poniższy warunek musi zostać spełniony, aby liny się przecięły: //x1 - pozycja x na pierwszej wysokości //x2 - pozycja x na drugiej wysokości //x3 - pozycja x drugiej liny na pierwszej wysokości //x4 - pozycja x drugiej liny na drugiej wysokości //(x1 < x3 && x2 > x4) || (x1 > x3 && x2 < x4) //Wiedząc to jesteśmy w stanie skonstruować graf w czasie O(n^2) poprzez //iterację przez wszystkie pary lin i sprawdzenie powyższego warunku. //Następnie stosując algorytm Floyda-Warshalla oraz iterację po rzędach //otrzymanej tabelki znajdujemy odpowiedź w czasie O(n^3). //Głównym problemem jest stosowanie algorytmu Floyda-Marshalla w celu znalezienia //odpowiedzi. Wiadomo, że wszystkie krawędzie mają długość 1. W takim razie najkrótsza trasa to taka, która przechodzi //przez najmniej wierzchołków. W związku z tym przy pomocy algorytmu BFS dla każdego //wierzchołka możemy znaleźć długość najkrótszej ścieżki do każdego innego w czasie liniowym. //Skoro szukamy sumy, to wystarczy przechowywać zmienną o nazwie sum podczas iteracji przez wierzchołki //i dla każdego wierzchołka dodawać jego odległość do sum. Otrzymamy w ten sposób odpowiedź. //Złożoność czasowa tego rozwiązania to O(n^2). //Dla każdego x-a na pierwszej wysokości istnieje dokładnie jedna lina. Dokładnie to samo dla x-a na drugiej //wysokości. Oznacza to, że liny nigdy nie będą przyczepione do tego samego punktu. int bfs(int at, vector<vector<int>> &graph) { vector<bool> vis(graph.size()); queue<pii> q; q.push(mp(at,0)); int sum = 0; while (!q.empty()) { int next = q.front().first; vis[next] = true; int dist = q.front().second; sum += dist; q.pop(); for (int i : graph[next]) { if (!vis[i]) { vis[i] = true; q.push(mp(i,dist + 1)); } } } return sum; } int main() { ios_base::sync_with_stdio(false); cin.tie(0); in_i(size); vector<int> vec; for (int i = 0; i < size; i++) { in_i(a); vec.pb(a); } vector<vector<int>> graph(size); for (int x1 = 0; x1 < vec.size(); x1++) { int x2 = vec[x1]; for (int x3 = x1 + 1; x3 < vec.size(); x3++) { int x4 = vec[x3]; if (x2 > x4) { graph[x1].pb(x3); graph[x3].pb(x1); } } } for (int i = 0; i < size; i++) { cout << bfs(i, graph) << " "; } cout << endl; } |
English