QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#798611#9810. Obliviate, Then Reincarnateucup-team4975WA 157ms17652kbC++232.7kb2024-12-04 15:13:312024-12-04 15:13:32

Judging History

This is the latest submission verdict.

  • [2024-12-04 15:13:32]
  • Judged
  • Verdict: WA
  • Time: 157ms
  • Memory: 17652kb
  • [2024-12-04 15:13:31]
  • Submitted

answer

#define LOCAL
#include <bits/stdc++.h>
#define fir first
#define sec second
#define el '\n'

#ifdef LOCAL
#define FINISH cerr << "FINISH" << endl;
#else
#define FINISH ;
#endif

#ifdef LOCAL
#define debug(x) cerr << setw(4) << #x << " == " << x << endl
#else
#define debug(x)
#endif

#ifdef LOCAL
#define debugv(x)                   \
    cerr << setw(4) << #x << ":: "; \
    for (auto i : x)                \
        cerr << i << " ";           \
    cerr << endl
#else
#define debugv(x)
#endif

using namespace std;
typedef long long ll;
typedef pair<int, int> PII;
ostream& operator<<(ostream& out, PII& x)
{
    out << x.fir << " " << x.sec << endl;
    return out;
}

const int mod = 998244353;
const int inf = 0x3f3f3f3f;
const int N = 200020;
int id(int x, int n) {
	return (x % n +n) % n + 1;
}
void solve()
{
    int n, m, q;
    cin >> n >> m >> q;
    vector<vector<PII>> edge(n + 1);
    for (int i = 1; i <= m; i++) {
    	int x, y;
    	cin >> x >> y;
    	edge[id(x, n)].push_back(PII(id(y + x, n), y));
        // cout << id(x, n) << " " << id(y, n) << " " << y << el;
    }

    vector<int> dfn(n + 1, 0), low(n + 1, 0), col(n + 1, 0), val(n + 1, 0), stk;

    int cnt = 0, tot = 0;
    auto dfs = [&] (auto &self, int x) -> void {
    	dfn[x] = low[x] = ++tot;
    	stk.push_back(x);
    	for (auto [to, w] : edge[x]) {
    		val[to] = val[x] + w;
    		if (!dfn[to]) {
    			self(self, to);
    			low[x] = min(low[x], low[to]);
    		}
    		else if (!col[to]) {
    			low[x] = min(low[x], dfn[to]);
    		}
    	}
    	if (dfn[x] == low[x]) {
    		int y;
    		cnt++;
    		do {
    			y = stk.back();
    			col[y] = cnt;
    			stk.pop_back();
    		} while (y != x);
    	}
    };

    for (int i = 1; i <= n; i++) {
    	if (!dfn[i]) {
    		dfs(dfs, i);
    	}
    }

    vector<int> ans(n + 1, 0);
    vector<vector<int>> g(n + 1);
    for (int i = 1; i <= n; i++) {
    	for (auto [to, w] : edge[i]) {
    		if (col[i] != col[to]) {
    			g[col[to]].push_back(col[i]);
    		}
    		if (col[i] == col[to] && val[i] + w != val[to]) {
    			ans[col[i]] = 1;
    		}
    	}
    }
    for (int i = 1; i <= cnt; i++) {
    	if (ans[i])
	    	for (auto to : g[i]) {
	    		ans[to] = 1;
	    	}
    }

    for (int i = 1; i <= q; i++) {
    	int x;
    	cin >> x;
    	x = id(x, n);
    	if (ans[col[x]] == 1)
    		cout <<"Yes" <<el;
    	else 
    		cout << "No" << el;
    }
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);
    int T = 1;
    // cin >> T;
    while (T--) {
        solve();
    }
    return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

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

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: 3580kb

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: 1ms
memory: 3640kb

input:

1 1 1
0 1000000000
-1000000000

output:

Yes

result:

ok "Yes"

Test #4:

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

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: 157ms
memory: 17652kb

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'