[SWEA 5521] 상원이의 생일파티

SW Expert Academy :: SWEA :: 5521 :: 상원이의 생일파티

출처 : https://www.swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AWWO3kT6F2oDFAV4&categoryId=AWWO3kT6F2oDFAV4&categoryType=CODE

#include < stdio.h >

#include < vector >

using namespace std;

int main() {

int T; scanf("%d", &T);

for (int tc = 1; tc <= T; tc++) {

int N, M, a, b; scanf("%d %d", &N, &M);

vector< vector< int >> v(501);

bool c[501] = { 0, }; c[1] = 1;

while (M--) {

scanf("%d %d", &a, &b); v[a].push_back(b); v[b].push_back(a);

if (a == 1) c[b] = 1;

}

int ans = v[1].size();

for (a = 0; a < v[1].size();a++) {

for (b = 0; b < v[v[1][a]].size(); b++) {

if (!c[v[v[1][a]][b]]) {

ans++; c[v[v[1][a]][b]] = 1;

}

}

}

printf("#%d %d\n", tc, ans);

}

return 0;

}