oj.uz

Submission #1056962

# Submission time Handle Problem Language Result Execution time Memory
1056962 2024-08-13T12:44:13 Z Boas Jail (JOI22_jail) C++17
0 / 100
 5000 ms 14420 KB 
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
using namespace std;
using namespace __gnu_pbds;

template <typename T1, typename T2>
using indexed_map = tree<T1, T2, less<T1>, rb_tree_tag, tree_order_statistics_node_update>;
template <typename T>
using indexed_set = indexed_map<T, null_type>;

#define loop(x, i) for (int i = 0; i < (x); i++)
#define loop1(x, i) for (int i = 1; i <= (x); i++)
#define rev(x, i) for (int i = (int)(x) - 1; i >= 0; i--)
#define itloop(x) for (auto it = begin(x); x != end(x); it++)
#define itrev(x) for (auto it = rbegin(x); x != rend(x); it++)
#define int long long
#define INF ((int64_t)(4e18 + 1))
#define INF32 ((int32_t)(2e9 + 1))
#define ALL(x) begin(x), end(x)
#define RALL(x) rbegin(x), rend(x)
#define existsIn(x, l) (count(ALL(l), x) > 0)
#define removeIn(x, l) l.erase(find(ALL(l), x))
#define pb push_back
#define sz(x) (int)(x).size()
#define F first
#define S second
#define var const auto &
#define foreach(l) for (var e : l)

typedef pair<int, int> ii;
typedef tuple<int, int, int> iii;
typedef vector<int> vi;
typedef vector<vi> vvi;
typedef vector<ii> vii;
typedef vector<iii> viii;
typedef vector<vii> vvii;
typedef vector<viii> vviii;
typedef set<int> si;
typedef set<ii> sii;
typedef set<iii> siii;
typedef vector<si> vsi;
typedef vector<sii> vsii;
typedef vector<vsi> vvsi;
typedef vector<string> vstr;
typedef vector<vector<string>> vvstr;
typedef vector<bool> vb;
typedef vector<vb> vvb;
typedef vi::iterator viit;
typedef si::iterator siit;
typedef int8_t in8;
typedef int16_t in16;
typedef int32_t in32;
typedef int64_t in64;

typedef bitset<500> bs;
typedef vector<bs> vbs;
typedef vector<vbs> vvbs;

void solve()
{
    int n, m;
    cin >> n;
    vvi adj(n);
    loop(n - 1, i)
    {
        int a, b;
        cin >> a >> b;
        a--;
        b--;
        adj[a].pb(b);
        adj[b].pb(a);
    }
    cin >> m;
    if (m > 500)
        throw;
    // vi mIx(n);
    vvbs bedRooms(20, vbs(n));
    vvbs workRooms(20, vbs(n));
    vvi p(20, vi(n));
    vi s(m), t(m), d(n);
    loop(m, i)
    {
        cin >> s[i] >> t[i];
        s[i]--;
        t[i]--;
        // mIx[s[i]] = i;
        bedRooms[0][s[i]][i] = 1;
        workRooms[0][t[i]][i] = 1;
    }
    auto dfs = [&](auto &&self, int i, int prev) -> void
    {
        for (int j : adj[i])
        {
            if (j == prev)
                continue;
            d[j] = d[i] + 1;
            p[0][j] = i;
            self(self, j, i);
        }
    };
    dfs(dfs, 0, 0);
    loop1(19, x)
    {
        loop(n, i)
        {
            int half = p[x - 1][i];
            p[x][i] = p[x - 1][half];
            bedRooms[x][i] = bedRooms[x - 1][i] | bedRooms[x - 1][half];
            workRooms[x][i] = workRooms[x - 1][i] | workRooms[x - 1][half];
        }
    }
    typedef tuple<int, bs, bs> ibs;
    auto getParent = [&](int i, int k) -> ibs
    {
        bs bedrms;
        bs workrms;
        loop(20, x)
        {
            if ((1 << x) & k)
            {
                bedrms |= bedRooms[x][i];
                workrms |= workRooms[x][i];
                i = p[x][i];
            }
        }
        // bedrms |= bedRooms[0][i];
        return {i, bedrms, workrms};
    };
    auto LCA = [&](int A, int b) -> ibs
    {
        if (d[A] - d[b] < 0)
            swap(A, b);
        auto [a, bedrms, workrms] = getParent(A, d[A] - d[b]);
        if (a == b)
        {
            bedrms |= bedRooms[0][a];
            workrms |= workRooms[0][a];
            return {a, bedrms, workrms};
        }
        rev(20, x)
        {
            if (p[x][a] != p[x][b])
            {
                bedrms |= bedRooms[x][a];
                bedrms |= bedRooms[x][b];
                workrms |= workRooms[x][a];
                workrms |= workRooms[x][b];
                a = p[x][a];
                b = p[x][b];
            }
        }
        bedrms |= bedRooms[1][a];   // voor de laatste erbij
        workrms |= workRooms[1][a]; // voor de laatste erbij
        bedrms |= bedRooms[0][b];
        workrms |= workRooms[0][b];
        return {p[0][a], bedrms, workrms};
    };
    int done = 0;
    vvi inPad(m);
    vi ervoor(m);
    loop(m, i)
    {
        auto [lca, bedrms, workrms] = LCA(s[i], t[i]);
        for (int j = bedrms._Find_first(); j < 500; j = bedrms._Find_next(j))
        {
            if (j == i)
                continue;
            if (workrms[j])
            {
                cout << "No" << endl;
                return;
            }
            ervoor[i]++;
            inPad[j].pb(i);
        }
    }
    stack<int> q;
    loop(m, i)
    {
        if (ervoor[i] == 0)
            q.push(i);
    }
    while (!q.empty())
    {
        int i = q.top();
        q.pop();
        done++;
        for (int j : inPad[i])
        {
            ervoor[j]--;
            if (ervoor[j] == 0)
            {
                q.push(j);
            }
        }
    }
    if (done == m)
        cout << "Yes" << endl;
    else
        cout << "No" << endl;
}

void solve2()
{
    int n, m;
    cin >> n;
    vvi adj(n);
    loop(n - 1, i)
    {
        int a, b;
        cin >> a >> b;
        a--;
        b--;
        adj[a].pb(b);
        adj[b].pb(a);
    }
    cin >> m;
    vi s(m), t(m), d(n);
    vi start(n), end(n);
    loop(m, i)
    {
        cin >> s[i] >> t[i];
        s[i]--;
        t[i]--;
        start[s[i]] = i;
        end[t[i]] = i;
    }
    loop(m, i)
    {
        loop(m, j)
        {
            if (s[i] < s[j] && t[i] > t[j])
            {
                cout << "No" << endl;
                return;
            }
            else if (s[i] < t[j] && t[j] < t[i] && t[j] < s[i] && s[i] < t[j])
            {
                cout << "No" << endl;
                return;
            }
            else if (t[i] < t[j] && t[j] < s[i] && s[j] < s[i] && s[i] < t[j])
            {
                cout << "No" << endl;
                return;
            }
        }
    }
    cout << "Yes" << endl;
}

signed main()
{
    cin.tie(0)->sync_with_stdio(0);
    int t = 1;
    cin >> t;
    while (t--)
    {
        solve2();
    }
}

Subtask #1
0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 12 ms 600 KB Output is correct
5 Correct 14 ms 348 KB Output is correct
6 Correct 2 ms 348 KB Output is correct
7 Correct 1 ms 348 KB Output is correct
8 Correct 1 ms 348 KB Output is correct
9 Correct 16 ms 944 KB Output is correct
10 Correct 17 ms 9752 KB Output is correct
11 Correct 3 ms 348 KB Output is correct
12 Correct 27 ms 520 KB Output is correct
13 Correct 1400 ms 11008 KB Output is correct
14 Correct 87 ms 12636 KB Output is correct
15 Correct 22 ms 12636 KB Output is correct
16 Correct 28 ms 14420 KB Output is correct
17 Correct 2156 ms 12940 KB Output is correct
18 Execution timed out 5028 ms 14356 KB Time limit exceeded
19 Halted 0 ms 0 KB -

Subtask #2
0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 1 ms 348 KB Output is correct
4 Incorrect 1 ms 348 KB Output isn't correct
5 Halted 0 ms 0 KB -

Subtask #3
0 / 16.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 1 ms 348 KB Output is correct
4 Incorrect 1 ms 348 KB Output isn't correct
5 Halted 0 ms 0 KB -

Subtask #4
0 / 28.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 1 ms 348 KB Output is correct
4 Incorrect 1 ms 348 KB Output isn't correct
5 Halted 0 ms 0 KB -

Subtask #5
0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 1 ms 348 KB Output is correct
4 Incorrect 1 ms 348 KB Output isn't correct
5 Halted 0 ms 0 KB -

Subtask #6
0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 0 ms 348 KB Output is correct
5 Correct 7 ms 348 KB Output is correct
6 Incorrect 1 ms 344 KB Output isn't correct
7 Halted 0 ms 0 KB -

Subtask #7
0 / 23.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 12 ms 600 KB Output is correct
5 Correct 14 ms 348 KB Output is correct
6 Correct 2 ms 348 KB Output is correct
7 Correct 1 ms 348 KB Output is correct
8 Correct 1 ms 348 KB Output is correct
9 Correct 16 ms 944 KB Output is correct
10 Correct 17 ms 9752 KB Output is correct
11 Correct 3 ms 348 KB Output is correct
12 Correct 27 ms 520 KB Output is correct
13 Correct 1400 ms 11008 KB Output is correct
14 Correct 87 ms 12636 KB Output is correct
15 Correct 22 ms 12636 KB Output is correct
16 Correct 28 ms 14420 KB Output is correct
17 Correct 2156 ms 12940 KB Output is correct
18 Execution timed out 5028 ms 14356 KB Time limit exceeded
19 Halted 0 ms 0 KB -