| # | Time | Username | Problem | Language | Result | Execution time | Memory |
|---|---|---|---|---|---|---|---|
| 629547 | hy_1 | Keys (IOI21_keys) | C++17 | 0 ms | 0 KiB |
This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <algorithm>
#include <bitset>
#include <complex>
#include <deque>
#include <exception>
#include <fstream>
#include <functional>
#include <iomanip>
#include <ios>
#include <iosfwd>
#include <iostream>
#include <istream>
#include <iterator>
#include <limits>
#include <list>
#include <cmath>
#include <locale>
#include <map>
#include <memory>
#include <new>
#include <numeric>
#include <ostream>
#include <queue>
#include <set>
#include <sstream>
#include <stack>
#include <stdio.h>
#include <stdexcept>
#include <streambuf>
#include <string>
#include <typeinfo>
#include <utility>
#include <valarray>
#include <vector>
#include <unordered_map>
using namespace std;
#define all(x) x.begin(),x.end()
const int mx = 300001;
pair<int , vector<int > > dfs(vector<int> &r , int source , vector<vector<pair<int , int > > > &adj , vector<int> &stock){
stack<int> st;
st.push(source);
pair<int , vector<int> > ans;
vector<bool> visited(r.size() , false);
while(!st.empty()){
int cur = st.top();
st.pop();
visited[cur] = true;
for(pair<int ,int > : adj[cur]){
if(count(all(stock) , i.second) != 0){
ans.second.push_back(r[i.first]);
stock.push_back(r[i.first]);
visited[i.first] = true;
st.push(i.first);
}
}
}
ans.first = ans.second.size();
return ans;
}
vector<int> find_reachable(vector<int> r, vector<int> u, vector<int> v, vector<int> c) {
//Graph implementatiom
vector<vector<pair<int , int > > > adj(mx);
for(int i=0; i <u.size(); i++){
pair<int, int> vv;
vv.first = v[i];
vv.second = c[i];
adj[u[i]].push_back(vv);
vv.first = u[i];
vv.second = c[i];
adj[v[i]].push_back(vv);
}
//BFS
vector<int> dp(r.size() , 1);
vector<int> visited(r.size() , false);
for(int i =0; i <r.size(); i++){
int source = i;
queue<int> qu;
if(visited[source] == false){
qu.push(source);
vector<int> stock();
visited[source] = false;
while(!qu.empty()){
int cur = qu.front();
qu.pop();
for(pair<int , int> i : adj[cur]){
if(dp[i.first] != -1 && count(all(stock) , i.second) != 0){
dp[cur] += dp[i.first] + 1;
for(int h : key[i.first])
stock.push_back(h);
}else{
pair<int, int> yyy = dfs(r , i.first , adj , stock);
dp[i.first] = yyy.first;
key[i.first] = yy.second;
dp[cur] += dp[i.first]+1;
key[i.first] = yyy.second;
}
}
}
}
}
vector<int> solution(r.size() , 0);
int y = *min_element(all(dp));
for(int i =0; i <dp.size(); i++){
if(dp[i] == y){
solution[i] = 1;
}else{
solution[i] = 0;
}
}
return solution;
}
/*
#1
4
0 1 1 2
5
0 0 1 1 3
1 2 2 3 1
0 0 1 0 2
#2
3
0 0 0
1
0
1
0
#3
7
0 1 1 2 2 1 2
10
0 0 1 1 2 3 3 4 4 5
1 2 2 3 3 4 5 5 6 6
0 0 1 0 0 1 2 0 2 1
*/
// int main(){
// int n; cin >> n;
// vector<int> r(n);
// for(int &i : r) cin >> i;
// int m; cin >> m;
// vector<int> u(m);
// vector<int> v(m);
// for(int &i : u) cin >> i;
// for(int &i : v) cin >> i;
// vector<int> c(m);
// for(int &i : c) cin >> i;
// for(int i =0; i <n; i++){
// vector<int> ans = find_reachable(r , u , v, c);
// for(int i :ans){
// cout << i << " ";
// }
// cout << endl;
// }
// return 0;
// }