QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#783361#9810. Obliviate, Then ReincarnateKobicGendWA 380ms27904kbC++233.3kb2024-11-26 09:14:052024-11-26 09:14:06

Judging History

This is the latest submission verdict.

  • [2024-11-26 23:19:26]
  • hack成功,自动添加数据
  • (/hack/1260)
  • [2024-11-26 09:14:06]
  • Judged
  • Verdict: WA
  • Time: 380ms
  • Memory: 27904kb
  • [2024-11-26 09:14:05]
  • Submitted

answer

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
constexpr ll inf = 1e18;

struct SCC {
    int n, unix, tot;
    vector<vector<int>> E;
    vector<int> dfn, low, bel, stk;

    SCC(int n) : n(n), unix(0), tot(0), E(n), dfn(n, -1), low(n), bel(n, -1) {}

    void add(int u, int v) {
        E[u].push_back(v);
    }

    void dfs(int u) {
        dfn[u] = low[u] = unix++;    // 记录DFS序
        stk.push_back(u);

        // DP更新low[u]
        for (auto& v : E[u]) {
            if (dfn[v] == -1) dfs(v), low[u] = min(low[u], low[v]);
            else if (bel[v] == -1) low[u] = min(low[u], dfn[v]);
        }

        // u是SCC的根
        if (dfn[u] == low[u]) {
            int v;
            do {
                v = stk.back();
                stk.pop_back();   // 出栈
                bel[v] = tot;     // 记录u的子节点的根为u
            } while (v != u);     // 不断弹出直到根u弹出为止
            tot++;                // 记录SCC的数量
        }
    }

    struct Graph {
        int n;
        vector<vector<int>> E;
        vector<int> ecnt, pcnt, bel;
        Graph(int n) : n(n), E(n), ecnt(n), pcnt(n) {}
    };
    Graph work() {
        for (int u = 0; u < n; u++) {
            if (dfn[u] == -1) dfs(u);
        }
        Graph G(tot);
        G.bel = bel;
        for (int u = 0; u < n; u++) {
            for (auto& v : E[u]) {
                if (bel[u] != bel[v]) G.E[bel[u]].push_back(bel[v]);  // bel[u]总是小于p[v]
                else G.ecnt[bel[u]]++;
            }
            G.pcnt[bel[u]]++;
        }
        for (auto& vec : G.E) {
            sort(vec.begin(), vec.end());
            vec.erase(unique(vec.begin(), vec.end()), vec.end());
        }
        return G;
    }

    template<typename T>
    vector<T> compress(const vector<T>& w) {
        vector<T> res(tot);
        for (int u = 0; u < n; u++) {
            res[bel[u]] += w[u];
        }
        return res;
    }
};

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr), cout.tie(nullptr);

    int n, m, q;
    cin >> n >> m >> q;

    vector<vector<pair<int, int>>> E(n);
    SCC G(n);
    while (m--) {
        int u, w;
        cin >> u >> w;
        int v = u + w;
        u = (u % n + n) % n;
        v = (v % n + n) % n;
        G.add(u, v);
        E[u].push_back({ w, v });
        E[v].push_back({-w, u });
    }

    vector<ll> val(n, inf);
    auto dfs = [&](auto&& self, int u) -> void {
        for (auto [w, v] : E[u]) {
            if (val[v] == inf) { 
                val[v] = val[u] + w;
                self(self, v);
            }
        }
    };
    for (int u = 0; u < n; u++) {
        if (val[u] == inf) {
            val[u] = 0;
            dfs(dfs, u);
        }
    }

    auto g = G.work();
    vector<int> dp(g.n);
    for (int u = 0; u < n; u++) {
        for (auto [w, v] : E[u]) {
            if (g.bel[u] == g.bel[v] && val[u] + w != val[v]) {
                dp[g.bel[u]] = 1;
            }
        }
    }

    for (int u = 0; u < g.n; u++) {
        for (auto v : g.E[u]) {
            dp[u] |= dp[v];
        }
    }

    while (q--) {
        int u;
        cin >> u;
        u = (u % n + n) % n;
        cout << (dp[g.bel[u]] ? "Yes" : "No") << endl;
    }

    return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 0ms
memory: 3836kb

input:

3 2 3
1 1
-1 3
1
2
3

output:

Yes
Yes
No

result:

ok 3 tokens

Test #2:

score: 0
Accepted
time: 0ms
memory: 3628kb

input:

3 2 3
1 1
-1 0
1
2
3

output:

No
No
No

result:

ok 3 tokens

Test #3:

score: 0
Accepted
time: 0ms
memory: 3752kb

input:

1 1 1
0 1000000000
-1000000000

output:

Yes

result:

ok "Yes"

Test #4:

score: 0
Accepted
time: 0ms
memory: 3628kb

input:

3 2 3
0 1000000000
1 -1000000000
-1000000000
0
-1000000000

output:

No
No
No

result:

ok 3 tokens

Test #5:

score: -100
Wrong Answer
time: 380ms
memory: 27904kb

input:

50134 500000 500000
-154428638 -283522863
-186373509 -327130969
154999046 46750274
-933523447 349415487
-437683609 140099255
864996699 -262318199
811293034 -264299324
120273173 52410685
874944410 -52048424
445049930 -803690605
-138111276 -104634331
720288580 126597671
471164416 -348777147
-356502322...

output:

Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
Yes
...

result:

wrong answer 1st words differ - expected: 'No', found: 'Yes'