제출 #1336851

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1336851		hauserl	Railway (BOI17_railway)	C++20	23 / 100	1096 ms	25432 KiB

railway

#include <bits/stdc++.h>
#define ll long long
using namespace std;

int n,m,k; 
vector<vector<pair<int, int>>> adj;
vector<int> countS;
vector<unordered_set<int>> chosenC;

bool dfs(int u, int p, int pos, int mPos) {

    bool isChosen = false;

    for (const auto& t : adj[u]) {
        int v = t.first;
        if (v == p) continue;

        isChosen |= dfs(v,u, pos, t.second);
    }

    if (chosenC[pos].count(u) > 0) isChosen = true;

    if (isChosen && mPos != -1) countS[mPos]++;

    return isChosen;
}

int main() {

    scanf("%d %d %d",&n,&m,&k);
    adj.resize(n+1, {});
    countS.resize(n, 0);
    chosenC.resize(m+1, {});

    for (int i = 1; i < n; i++) {
        int a,b; scanf("%d %d",&a,&b);
        adj[a].push_back({b, i});
        adj[b].push_back({a,i});
    }

    for (int i = 1; i <= m; i++) {
        int s; scanf("%d",&s);
        for (int j = 0; j < s; j++) {
            int z; scanf("%d",&z);
            chosenC[i].insert(z);
        }
    }


    for (int i = 1; i <= m; i++) {
        
        dfs(*chosenC[i].begin(), -1, i, -1);
    }


    set<int> roads;
    
    for (int i = 1; i < n; i++) {
        if (countS[i] >= k) roads.insert(i);
    }

    printf("%d\n", roads.size());
    
    for (const auto& r : roads) {
        printf("%d ", r);
    }


    return 0;
}

컴파일 시 표준 에러 (stderr) 메시지

railway.cpp: In function 'int main()':
railway.cpp:62:14: warning: format '%d' expects argument of type 'int', but argument 2 has type 'std::set<int>::size_type' {aka 'long unsigned int'} [-Wformat=]
   62 |     printf("%d\n", roads.size());
      |             ~^     ~~~~~~~~~~~~
      |              |               |
      |              int             std::set<int>::size_type {aka long unsigned int}
      |             %ld
railway.cpp:30:10: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   30 |     scanf("%d %d %d",&n,&m,&k);
      |     ~~~~~^~~~~~~~~~~~~~~~~~~~~
railway.cpp:36:23: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   36 |         int a,b; scanf("%d %d",&a,&b);
      |                  ~~~~~^~~~~~~~~~~~~~~
railway.cpp:42:21: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   42 |         int s; scanf("%d",&s);
      |                ~~~~~^~~~~~~~~
railway.cpp:44:25: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   44 |             int z; scanf("%d",&z);
      |                    ~~~~~^~~~~~~~~

• Subtask 1: 8 / 8
• Subtask 2: 15 / 15
• Subtask 3: 0 / 7
• Subtask 4: 0 / 29
• Subtask 5: 0 / 16
• Subtask 6: 0 / 25