#include <iostream>
#include <string>
#include <queue>
#include <stack>
#include <algorithm>
#include <math.h>
#include <vector>
#include <cstring>
#include <ctime>
#include <set>
#include <map>
#include <unordered_map>
#include <unordered_set>
#include <iomanip>
#include <ctime>
#define fr first
#define sc second
#define MP make_pair
#define PB push_back
#define fastio ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0);
#define lli long long int
#define y1 arayikhalatyan
using namespace std;
lli gcd(lli a, lli b) {
if (b == 0)
return a;
return gcd(b, a % b);
}
lli cg(lli n) {
return n ^ (n >> 1);
}
lli SUM(lli a)
{
return (a * (a + 1) / 2);
}
bool CAN(int x, int y, int n, int m)
{
if (x >= 0 && y >= 0 && x < n && y < m)
{
return true;
}
return false;
}
double her(double x1, double y1, double x2, double y2)
{
return sqrt((x1 - x2) * (x1 - x2) + (y1 - y2) * (y1 - y2));
}
string strsum(string a, string b)
{
int p = 0;
string c;
reverse(a.begin(), a.end());
reverse(b.begin(), b.end());
if (b.length() < a.length())
{
for (int i = b.length(); i < a.length(); i++)
{
b += "0";
}
}
else
{
for (int i = a.length(); i < b.length(); i++)
{
a += "0";
}
}
a += "0", b += "0";
for (int i = 0; i < a.length(); i++)
{
c += (a[i] - '0' + b[i] - '0' + p) % 10 + '0';
p = (a[i] + b[i] - '0' - '0' + p) / 10;
}
if (c[c.length() - 1] == '0') c.erase(c.length() - 1, 1);
reverse(c.begin(), c.end());
return c;
}
string strmin(string a, string b)
{
if (a.length() > b.length()) return b;
if (b.length() > a.length()) return a;
for (int i = 0; i < a.length(); i++)
{
if (a[i] > b[i]) return b;
if (b[i] > a[i]) return a;
}
return a;
}
char vow[] = { 'a', 'e', 'i', 'o', 'u' };
int dx[] = { 1, -1, 0, 0 };
int dy[] = { 0, 0, 1, -1 };
const int N = 1e6 + 30;
const lli mod = 1000000007;
int mn(int a, int b)
{
if (a < b) return a;
return b;
}
int n, sm;
string s;
vector <int> g[N];
bool col[N];
int qan[N];
queue <int> h;
int nerq[2 * N], ver[2 * N];
lli pat;
void dfsqan(int v, int par)
{
qan[v] = 1;
for (auto p : g[v])
{
if (p != par && !col[p])
{
dfsqan(p, v);
qan[v] += qan[p];
}
}
}
void dfshashv(int v, int par, int min, int val)
{
for (auto p : g[v])
{
if (p != par)
{
if (s[v - 1] == '(') dfshashv(p, v, min, val + 1);
else
{
val--;
if (val <= min) nerq[val + N]++;
dfshashv(p, v, mn(min, val), val);
val++;
}
if (s[v - 1] == '(')
{
val++;
dfshashv(p, v, mn(1, min + 1), val);
if (mn(1, min + 1) >= 0) ver[val + N]++;
val--;
}
else
{
val--;
dfshashv(p, v, mn(-1, min - 1), val);
if (mn(-1, min - 1) >= 0) ver[val + N]++;
val++;
}
}
}
}
void dfspat(int v, int par, int min, int val)
{
for (auto p : g[v])
{
if (p != par)
{
if (s[v - 1] == '(') dfspat(p, v, min, val + 1);
else
{
val--;
if (val <= min) pat += ver[N - val];
dfspat(p, v, mn(min, val), val);
val++;
}
if (s[v - 1] == '(' && pat != -1)
{
val++;
dfspat(p, v, mn(1, min + 1), val);
if (mn(1, min + 1) >= 0) pat += nerq[N - val];
val--;
}
else if (pat != -1)
{
val--;
dfspat(p, v, mn(-1, min - 1), val);
if (mn(-1, min - 1) >= 0) ver[val + N]++;
val++;
}
}
}
}
void dfsc(int v, int par)
{
bool flag = true;
for (auto p : g[v])
{
if (p != par && !col[p])
{
if (qan[p] > sm)
{
dfsc(p, v);
flag = false;
}
}
}
if (flag)
{
cout << v << endl;
col[v] = true;
for (auto p : g[v])
{
if (!col[p])
{
h.push(p);
}
}
dfspat(v, -1, 0, 0);
dfshashv(v, -1, 0, 0);
}
}
int main()
{
fastio;
cin >> n >> s;
for (int i = 0; i < n - 1; i++)
{
int a, b;
cin >> a >> b;
g[a].PB(b);
g[b].PB(a);
}
h.push(1);
while (!h.empty())
{
int k = h.front();
h.pop();
dfsqan(k, -1);
sm = qan[k] / 2;
dfsc(k, -1);
}
cout << pat;
return 0;
}
Compilation message
zagrade.cpp: In function 'std::__cxx11::string strsum(std::__cxx11::string, std::__cxx11::string)':
zagrade.cpp:57:30: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for (int i = b.length(); i < a.length(); i++)
~~^~~~~~~~~~~~
zagrade.cpp:64:30: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for (int i = a.length(); i < b.length(); i++)
~~^~~~~~~~~~~~
zagrade.cpp:71:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for (int i = 0; i < a.length(); i++)
~~^~~~~~~~~~~~
zagrade.cpp: In function 'std::__cxx11::string strmin(std::__cxx11::string, std::__cxx11::string)':
zagrade.cpp:84:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for (int i = 0; i < a.length(); i++)
~~^~~~~~~~~~~~
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Execution timed out |
3015 ms |
24056 KB |
Time limit exceeded |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Execution timed out |
3031 ms |
64780 KB |
Time limit exceeded |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Execution timed out |
3015 ms |
24056 KB |
Time limit exceeded |
| 2 |
Halted |
0 ms |
0 KB |
- |
