Peli

input/code.cpp: In function 'void search(int, int, int, int)':
input/code.cpp:12:22: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   12 |     for(int i = 0; i < v[k].size(); i++) {
      |                    ~~^~~~~~~~~~~~~

#include <bits/stdc++.h>

using namespace std;

vector<int> v[2501];
int seven;
int sodd;
set<int> vis;

void search(int k,int w,int y,int z) {
    vis.insert(k);
    for(int i = 0; i < v[k].size(); i++) {
        if (v[k][i] == w) {
                if (y%2 == 0) {
                    seven = min(seven,y);
                }
                else {
                    sodd = min(sodd,y);
                }
            }
        if (v[k][i] != z) {
            if (vis.count(v[k][i]) != 1) {
                search(v[k][i],w,y+1,k);
            }
        }
    }
    vis.erase(k);
}

int main() {
    int n,m,q;
    int a,b,c;
    cin >> n >> m >> q;
    for(int i = 0; i<m; i++) {
        cin >> a >> b;
        v[a].push_back(b);
        v[b].push_back(a);
    }
    cout << "\n";
    for(int i = 0; i<q; i++) {
        vis = {};
        cin >> a >> b >> c;
        seven = 1000000;
        sodd = 1000000;
        if (c == 0) {
            if (a == b) {
                cout << "YES" << "\n";
            }
            else {
                cout << "NO" << "\n";
            }
            continue;
        }
        search(a,b,1,0);
        if (c%2 == 0) {
            if (seven <= c) {
                cout << "YES" << "\n";
            }
            else {
                cout << "NO" << "\n";
            }
        }
        else {
            if (sodd <= c) {
                cout << "YES" << "\n";
            }
            else {
                cout << "NO" << "\n";
            }
        }
    }
}

Test 1

Group: 1, 2

Verdict: ACCEPTED

Test 2

Group: 1, 2

Verdict: ACCEPTED

Test 3

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 4

Group: 1, 2

Verdict: ACCEPTED

Test 5

Group: 1, 2

Verdict: ACCEPTED

Test 6

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 7

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 8

Group: 1, 2

Verdict: TIME LIMIT EXCEEDED

Test 9

Group: 2

Verdict: TIME LIMIT EXCEEDED

Test 10

Group: 2

Verdict: TIME LIMIT EXCEEDED