728x90
SW Expert Academy
SW 프로그래밍 역량 강화에 도움이 되는 다양한 학습 컨텐츠를 확인하세요!
swexpertacademy.com

인접 리스트
import java.util.ArrayList;
import java.util.Scanner;
class Solution {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int t = sc.nextInt();
for(int tc=1;tc<=t;tc++) {
int n = sc.nextInt();
int m = sc.nextInt();
ArrayList<Integer>list[]=new ArrayList[n+1];
for(int i=1;i<n+1;i++) {
list[i] = new ArrayList<>();
}
for(int i=0;i<m;i++) {
int x = sc.nextInt();
int y = sc.nextInt();
list[x].add(y);
list[y].add(x);
}
int cnt=0;
for(int i=1;i<n+1;i++) {
for(int j:list[i]) {
for(int k:list[j]) {
if(i<j&&j<k&&list[k].contains(i)) {
cnt++;
}
}
}
}
System.out.println("#"+tc+" "+cnt);
}
}
}
인접 행렬
import java.util.Scanner;
class Solution {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int t = sc.nextInt();
for (int tc = 1; tc <= t; tc++) {
int n = sc.nextInt();
int m = sc.nextInt();
int arr[][] = new int[n + 1][n + 1];
for (int i = 0; i < m; i++) {
int x = sc.nextInt();
int y = sc.nextInt();
arr[x][y] = arr[y][x] = 1;
}
int ans = 0;
for (int i = 1; i < n + 1 - 2; i++) {
for (int j = i + 1; j < n + 1 - 1; j++) {
if(arr[i][j]==0) continue;
for (int k = j + 1; k < n + 1; k++) {
if(arr[j][k]==0) continue;
if (i<j && arr[k][i] == 1) {
ans++;
}
}
}
}
System.out.println("#" + tc + " " + ans);
}
}
}728x90
'SW Expert Academy > SWEA D3' 카테고리의 다른 글
| [SW Expert Academy] Digit sum (D3) (1) | 2024.10.12 |
|---|---|
| [SW Expert Academy] 등차수열 만들기 (D3) (0) | 2024.10.12 |
| [SW Expert Academy] 식료품 가게 (D3) (0) | 2024.10.08 |
| [SW Expert Academy] 공평한 분배 2 (D3) (0) | 2024.10.08 |
| [SW Expert Academy] 세제곱근을 찾아라 (D3) (0) | 2024.10.08 |