#include "simurgh.h"
#include <iostream>
#include <queue>
using namespace std;
const int MAX_NOEUD = 500;
int nbNoeud, nbArc;
int groupe[MAX_NOEUD];
bool dv[MAX_NOEUD];
bool estImpossible[MAX_NOEUD];
int valide[MAX_NOEUD];
vector<int> result;
vector<int> curArbre;
vector<int> arcParType[MAX_NOEUD];
vector<int> U, V;
vector<pair<int, int> > graphe[MAX_NOEUD];
int retourne_Groupe(int a)
{
if(groupe[a] == a)
{
return a;
}
groupe[a] = retourne_Groupe(groupe[a]);
return groupe[a];
}
void fusionne(int a, int b)
{
int gpA = retourne_Groupe(a), gpB = retourne_Groupe(b);
groupe[gpA] = gpB;
}
void construire_Arbre(int noeudSuppr)
{
for(int iNoeud = 0; iNoeud < nbNoeud; iNoeud++)
{
groupe[iNoeud] = iNoeud;
dv[iNoeud] = false;
}
curArbre.clear();
queue<int> noeudsEnCours;
for(int iNoeud = 0; iNoeud < nbNoeud; iNoeud++)
{
if(iNoeud == noeudSuppr || dv[iNoeud])
{
continue;
}
noeudsEnCours.push(iNoeud);
while(!noeudsEnCours.empty())
{
int curNoeud = noeudsEnCours.front();
noeudsEnCours.pop();
dv[curNoeud] = true;
for(pair<int, int> voisin : graphe[curNoeud])
{
if(voisin.first == noeudSuppr)
{
continue;
}
if(retourne_Groupe(voisin.first) != retourne_Groupe(curNoeud))
{
fusionne(voisin.first, curNoeud);
noeudsEnCours.push(voisin.first);
curArbre.push_back(voisin.second);
}
}
}
}
}
void affiche_Arbre()
{
for(int i : curArbre)
{
cout << i << endl;
}
cout << endl;
}
void test(int curNoeud)
{
vector<int> typeDispo;
for(pair<int, int> voisin : graphe[curNoeud])
{
if(arcParType[groupe[voisin.first]].size() == 0)
{
typeDispo.push_back(groupe[voisin.first]);
}
arcParType[groupe[voisin.first]].push_back(voisin.second);
}
for(size_t curType = 0; curType < typeDispo.size(); curType++)
{
for(size_t otherType = 0; otherType < typeDispo.size(); otherType++)
{
if(otherType != curType)
{
curArbre.push_back(arcParType[typeDispo[otherType]][0]);
}
}
int maxi = 0;
vector<int> arcBon;
if(valide[curNoeud] != -1)
{
curArbre.push_back(valide[curNoeud]);
maxi = count_common_roads(curArbre);
curArbre.pop_back();
}
for(int voisin : arcParType[typeDispo[curType]])
{
int dest = U[voisin];
if(dest == curNoeud)
{
dest = V[voisin];
}
if(valide[curNoeud] != -1 && dest > curNoeud)
{
continue;
}
curArbre.push_back(voisin);
int val = count_common_roads(curArbre);
if(val > maxi)
{
arcBon.clear();
maxi = val;
}
if(val == maxi)
{
arcBon.push_back(voisin);
}
curArbre.pop_back();
}
for(int i : arcBon)
{
int dest = U[i];
if(dest == curNoeud)
{
dest = V[i];
}
valide[dest] = i;
if(dest > curNoeud)
{
continue;
}
result.push_back(i);
}
for(int i = 0; i < (int)typeDispo.size() - 1; i++)
{
curArbre.pop_back();
}
}
for(int i : typeDispo)
{
arcParType[i].clear();
}
}
vector<int> find_roads(int n, vector<int> u, vector<int> v)
{
nbNoeud = n;
nbArc = (int)u.size();
for(int iArc = 0; iArc < nbArc; iArc++)
{
graphe[u[iArc]].push_back({v[iArc], iArc});
graphe[v[iArc]].push_back({u[iArc], iArc});
U.push_back(u[iArc]);
V.push_back(v[iArc]);
}
for(int iNoeud = 0; iNoeud < nbNoeud; iNoeud++)
{
valide[iNoeud] = -1;
}
for(int iNoeud = 0; iNoeud < nbNoeud; iNoeud++)
{
construire_Arbre(iNoeud);
test(iNoeud);
}
return result;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
384 KB |
correct |
2 |
Correct |
1 ms |
384 KB |
correct |
3 |
Correct |
0 ms |
384 KB |
correct |
4 |
Correct |
0 ms |
384 KB |
correct |
5 |
Correct |
0 ms |
384 KB |
correct |
6 |
Correct |
0 ms |
384 KB |
correct |
7 |
Correct |
0 ms |
384 KB |
correct |
8 |
Correct |
1 ms |
384 KB |
correct |
9 |
Incorrect |
0 ms |
384 KB |
WA in grader: NO |
10 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
384 KB |
correct |
2 |
Correct |
1 ms |
384 KB |
correct |
3 |
Correct |
0 ms |
384 KB |
correct |
4 |
Correct |
0 ms |
384 KB |
correct |
5 |
Correct |
0 ms |
384 KB |
correct |
6 |
Correct |
0 ms |
384 KB |
correct |
7 |
Correct |
0 ms |
384 KB |
correct |
8 |
Correct |
1 ms |
384 KB |
correct |
9 |
Incorrect |
0 ms |
384 KB |
WA in grader: NO |
10 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
384 KB |
correct |
2 |
Correct |
1 ms |
384 KB |
correct |
3 |
Correct |
0 ms |
384 KB |
correct |
4 |
Correct |
0 ms |
384 KB |
correct |
5 |
Correct |
0 ms |
384 KB |
correct |
6 |
Correct |
0 ms |
384 KB |
correct |
7 |
Correct |
0 ms |
384 KB |
correct |
8 |
Correct |
1 ms |
384 KB |
correct |
9 |
Incorrect |
0 ms |
384 KB |
WA in grader: NO |
10 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
384 KB |
correct |
2 |
Correct |
1 ms |
384 KB |
correct |
3 |
Incorrect |
195 ms |
4848 KB |
WA in grader: NO |
4 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
384 KB |
correct |
2 |
Correct |
1 ms |
384 KB |
correct |
3 |
Correct |
0 ms |
384 KB |
correct |
4 |
Correct |
0 ms |
384 KB |
correct |
5 |
Correct |
0 ms |
384 KB |
correct |
6 |
Correct |
0 ms |
384 KB |
correct |
7 |
Correct |
0 ms |
384 KB |
correct |
8 |
Correct |
1 ms |
384 KB |
correct |
9 |
Incorrect |
0 ms |
384 KB |
WA in grader: NO |
10 |
Halted |
0 ms |
0 KB |
- |