#include "simurgh.h"
#include <algorithm>
#include <vector>
using namespace std;
#define query count_common_roads
const int maxN = 500;
struct Jmp
{
int to, ind;
operator int&() { return to; }
operator const int&() const { return to; }
};
int in[maxN], out[maxN];
Jmp par[maxN];
int gold[maxN];
vector<int> dfsq;
vector<Jmp> nei[maxN];
vector<int> same[maxN];
pair<int, Jmp> fup[maxN];
inline bool Upper(int u, int v) { return in[u] <= in[v] && out[u] >= out[v]; }
bool used[maxN];
void DFS(int v, Jmp p = {-1, -1})
{
static int time = 0;
used[v] = true;
in[v] = ++time;
par[v] = p;
fup[v] = {-1, {-1, -1}};
for (Jmp to: nei[v])
if (!used[to])
{
dfsq.push_back(to.ind);
DFS(to, {v, to.ind});
if (fup[v].first == -1 || fup[to].first != -1 && in[fup[to].second] <= in[fup[v].second])
fup[v] = fup[to];
}
else if (to != p && (fup[v].first == -1 || in[to] <= in[fup[v].second]))
fup[v] = {v, to};
out[v] = time;
}
void DFS1(int v)
{
for (int to: same[v])
if (gold[to] == -1)
{
gold[to] = gold[v];
DFS1(to);
}
}
vector<int> find_roads(int n, vector<int> us, vector<int> vs)
{
for (int i = 0; i < n; ++i)
gold[i] = -1;
for (int i = 0; i < us.size(); ++i)
{
nei[us[i]].push_back({vs[i], i});
nei[vs[i]].push_back({us[i], i});
}
DFS(0, {-1, -1});
int dfsans = query(dfsq);
for (int i = 0; i < us.size(); ++i)
{
int u = us[i], v = vs[i];
if (in[u] > in[v]) swap(u, v);
if (fup[v].first == -1 || in[fup[v].second] > in[u])
{
gold[i] = 1;
continue;
}
int dfse, backe;
if (par[v] == u)
dfse = par[v].ind, backe = fup[v].second.ind;
else
{
for (backe = 0; ; ++backe)
if (us[backe] == u && vs[backe] == v)
break;
dfse = par[v].ind;
}
auto it = find(dfsq.begin(), dfsq.end(), dfse);
*it = backe;
int curans = query(dfsq);
if (curans == dfsans)
{
same[backe].push_back(dfse);
same[dfse].push_back(backe);
}
else if (curans > dfsans)
gold[backe] = 1, gold[dfse] = 0;
else
gold[backe] = 0, gold[dfse] = 1;
*it = dfse;
}
for (int i = 0; i < us.size(); ++i)
if (gold[i] != -1)
DFS1(i);
vector<int> ans;
for (int i = 0; i < us.size(); ++i)
if (gold[i])
ans.push_back(i);
return ans;
}
Compilation message
simurgh.cpp: In function 'void DFS(int, Jmp)':
simurgh.cpp:36:59: warning: suggest parentheses around '&&' within '||' [-Wparentheses]
if (fup[v].first == -1 || fup[to].first != -1 && in[fup[to].second] <= in[fup[v].second])
~~~~~~~~~~~~~~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
simurgh.cpp: In function 'std::vector<int> find_roads(int, std::vector<int>, std::vector<int>)':
simurgh.cpp:57:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for (int i = 0; i < us.size(); ++i)
~~^~~~~~~~~~~
simurgh.cpp:64:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for (int i = 0; i < us.size(); ++i)
~~^~~~~~~~~~~
simurgh.cpp:97:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for (int i = 0; i < us.size(); ++i)
~~^~~~~~~~~~~
simurgh.cpp:101:23: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for (int i = 0; i < us.size(); ++i)
~~^~~~~~~~~~~
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
4 ms |
384 KB |
WA in grader: NO |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
4 ms |
384 KB |
WA in grader: NO |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
4 ms |
384 KB |
WA in grader: NO |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
4 ms |
384 KB |
correct |
| 2 |
Incorrect |
4 ms |
384 KB |
WA in grader: NO |
| 3 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
4 ms |
384 KB |
WA in grader: NO |
| 2 |
Halted |
0 ms |
0 KB |
- |
