#include <bits/stdc++.h>
#define all(v) v.begin(), v.end()
#define chmax(x, v) x = max(x, v)
#define chmin(x, v) x = min(x, v)
#define pb push_back
#define pii pair<int, int>
#define sz(x) (int)x.size()
#define x first
#define y second
//#define int long long
using namespace std;
const int N = 3e5;
int sousArbre[N];
vector<int> enfants[N];
vector<int> winner[N];
int vals[N];
void dfs(int noeud){
sousArbre[noeud] = vals[noeud];
for (int enfant : enfants[noeud]){
dfs(enfant);
sousArbre[noeud] += sousArbre[enfant];
if (sousArbre[enfant] >= vals[noeud])
winner[noeud].pb(enfant);
}
}
bool vainqueur[N];
void complete(int noeud){
vainqueur[noeud] = true;
for (int enf : winner[noeud])
complete(enf);
}
struct Noeud {
int noeud, val;
bool operator < (const Noeud &autre) const {
if (val != autre.val)
return val < autre.val;
return noeud < autre.noeud;
}
};
vector<int> voisins[N];
int ens[N], containerVoisins[N];
int getEns(int noeud){
if (ens[noeud] != noeud)
ens[noeud] = getEns(ens[noeud]);
return ens[noeud];
}
int fusion(int a, int b){
if (sz(voisins[a]) < sz(voisins[b]))
return fusion(b, a);
for (int voisin : voisins[b])
voisins[a].pb(voisin);
return a;
}
signed main(){
ios::sync_with_stdio(false);
cin.tie(nullptr);
cout.tie(nullptr);
int nNoeuds, nAretes; cin >> nNoeuds >> nAretes;
for (int noeud = 0; noeud < nNoeuds; ++noeud)
ens[noeud] = containerVoisins[noeud] = noeud;
for (int i = 0; i < nNoeuds; ++i)
cin >> vals[i];
for (int i = 0; i < nAretes; ++i){
int u, v; cin >> u >> v;
u--; v--;
voisins[u].pb(v);
voisins[v].pb(u);
}
vector<Noeud> sac;
for (int noeud = 0; noeud < nNoeuds; ++noeud)
sac.pb({noeud, vals[noeud]});
sort(all(sac));
int root = -1;
for (Noeud nod : sac){
int noeud = nod.noeud;
int pere = -1;
vector<int> vrais;
for (int& voisin : voisins[containerVoisins[noeud]]){
voisin = getEns(voisin);
if (voisin == noeud) continue;
if (vals[voisin] >= vals[noeud]){
vrais.pb(voisin);
if (pere == -1 || vals[pere] > vals[voisin])
pere = voisin;
}
}
// voisins[containerVoisins[noeud]] = vrais;
if (pere == -1) {
root = noeud;
continue;
}
enfants[pere].pb(noeud);
ens[noeud] = pere;
containerVoisins[pere] = fusion(containerVoisins[pere], containerVoisins[noeud]);
}
/*
for (int noeud = 0; noeud < nNoeuds; ++noeud){
cout << "Noeud " << noeud << endl;
for (int enf : enfants[noeud])
cout << enf << " ";
cout << endl;
}
*/
dfs(root);
complete(root);
for(int i = 0; i < nNoeuds; ++i)
cout << vainqueur[i];
cout << endl;
return 0;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
11 ms |
21460 KB |
Output is correct |
2 |
Correct |
12 ms |
21404 KB |
Output is correct |
3 |
Correct |
10 ms |
21424 KB |
Output is correct |
4 |
Incorrect |
18 ms |
21688 KB |
Output isn't correct |
5 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
10 ms |
21460 KB |
Output is correct |
2 |
Correct |
11 ms |
21460 KB |
Output is correct |
3 |
Execution timed out |
1086 ms |
32360 KB |
Time limit exceeded |
4 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
13 ms |
21460 KB |
Output is correct |
2 |
Incorrect |
351 ms |
52512 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
10 ms |
21460 KB |
Output is correct |
2 |
Incorrect |
397 ms |
46160 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
11 ms |
21460 KB |
Output is correct |
2 |
Correct |
12 ms |
21404 KB |
Output is correct |
3 |
Correct |
10 ms |
21424 KB |
Output is correct |
4 |
Incorrect |
18 ms |
21688 KB |
Output isn't correct |
5 |
Halted |
0 ms |
0 KB |
- |