💻 오늘의 코드 팁 — 처음부터 만드는 Partition
문제 상황
배열을 특정 조건에 따라 두 그룹으로 나누고 싶은데, 반복문을 여러 번 돌아야 한다면?
```typescript
// ❌ 비효율적 — 배열을 두 번 순회
const numbers = [1, 2, 3, 4, 5, 6];
const evens = numbers.filter(n => n % 2 === 0);
const odds = numbers.filter(n => n % 2 !== 0);
```
해결 코드
한 번의 순회로 두 그룹을 동시에 만드는 `partition` 함수:
```typescript
function partition
array: T[],
predicate: (item: T, index: number) => boolean
): [T[], T[]] {
const trueGroup: T[] = [];
const falseGroup: T[] = [];
for (let i = 0; i < array.length; i++) {
const item = array[i];
if (predicate(item, i)) {
trueGroup.push(item);
} else {
falseGroup.push(item);
}
}
return [trueGroup, falseGroup];
}
// ✅ 사용 예
const numbers = [1, 2, 3, 4, 5, 6];
const [evens, odds] = partition(numbers, n => n % 2 === 0);
console.log(evens); // [2, 4, 6]
console.log(odds); // [1, 3, 5]
// 다른 예: 사용자를 활성/비활성으로 분류
interface User { name: string; active: boolean }
const users: User[] = [
{ name: 'Alice', active: true },
{ name: 'Bob', active: false },
{ name: 'Charlie', active: true },
];
const [activeUsers, inactiveUsers] = partition(users, u => u.active);
console.log(activeUsers); // [{name: 'Alice', active: true}, {name: 'Charlie', active: true}]
console.log(inactiveUsers); // [{name: 'Bob', active: false}]
```
설명
`partition`은 배열을 한 번만 순회하면서 predicate 함수 결과에 따라 요소를 두 그룹으로 분류합니다:
reduce 버전 (함수형):
```typescript
function partition
array: T[],
predicate: (item: T) => boolean
): [T[], T[]] {
return array.reduce(
([trueGroup, falseGroup], item) => {
return predicate(item)
? [[...trueGroup, item], falseGroup]
: [trueGroup, [...falseGroup, item]];
},
[[], []] as [T[], T[]]
);
}
```
📚 관련 유틸리티 라이브러리:
👁 0 views
Comments (0)
💬
No comments yet.
Be the first to comment!