oj.uz

답안 #924537

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
924537 2024-02-09T07:25:27 Z UmairAhmadMirza Railway (BOI17_railway) C++17
0 / 100
 10 ms 4280 KB 
#include <bits/stdc++.h>
using namespace std;
int const N=10005;
int const M=2005;
vector<int> adj[N];
int cnt[N][M];
int sz[M];
vector<int> ans;
int n,m,k;
map<pair<int,int>,int> mp;
void dfs(int node,int par){
  for(auto i:adj[node])
    if(i!=par){
      dfs(i,node);
      for(int j=0;j<m;j++)
        cnt[node][j]+=cnt[i][j];
    }
  int p=0;
  for(int j=0;j<m;j++)
    if(cnt[node][j]<sz[j] && cnt[node][j]>0)
      p++;
  if(p>=k && par!=-1)
    ans.push_back(mp[{par,node}]);
}
int main(){
  cin>>n>>m>>k;
  for(int i=0;i<n-1;i++){
    int a,b;
    cin>>a>>b;
    mp[{a,b}]=i+1;
    mp[{b,a}]=i+1;
    adj[a].push_back(b);
    adj[b].push_back(a);
  }
  for(int i=0;i<m;i++){
    cin>>sz[i];
    for(int j=0;j<sz[i];j++){
      int a;
      cin>>a;
      cnt[a][i]++;
    }
  }
  dfs(1,-1);
  cout<<ans.size()<<endl;
  for(auto i:ans)
    cout<<i<<' ';
  cout<<endl;
  return 0;
}

Subtask #1
0 / 8.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 1 ms 604 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #2
0 / 15.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 1 ms 604 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #3
0 / 7.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 10 ms 4280 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -

Subtask #4
0 / 29.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 3 ms 948 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -

Subtask #5
0 / 16.0

# 결과 실행 시간 메모리 Grader output
1 Runtime error 3 ms 948 KB Execution killed with signal 11
2 Halted 0 ms 0 KB -

Subtask #6
0 / 25.0

# 결과 실행 시간 메모리 Grader output
1 Incorrect 1 ms 604 KB Output isn't correct
2 Halted 0 ms 0 KB -