자바로 배우는 핵심 자료구조와 알고리즘: 기술 면접에 필요한 실용주의 자료구조와 알고리즘

130

자바로 배우는 핵심 자료구조와 알고리즘

13.3

put

메서드 구현하기

put

메서드는

get

메서드보다 조금 더 복잡합니다. 이는

put

메서드가 다음 두 가지 경우를 모

두 처리해야 하기 때문입니다.

●

주어진 키가 트리에 이미 있으면 값을 대체하고 기존 값을 반환합니다.

●

주어진 키가 트리에 없으면 올바른 위치에 새로운 노드를 추가해야 합니다.

실습

10

에서 다음 시작 코드를 제공하고 여러분에게

putHelper

메서드를 채우게 하였습니다

(파일명:

MyTreeMap.java

)

.

public V put

(

K key

,

V value

)

{

if

(

key

==

null

)

{

throw new IllegalArgumentException

();

}

if

(

root

==

null

)

{

root

=

new Node

(

key

,

value

);

size

++;

return null

;

}

return putHelper

(

root

,

key

,

value

);

}

필자의 해답은 다음과 같습니다.

(파일명:

MyTreeMap.java

)

private V putHelper

(

Node node

,

K key

,

V value

)

{

Comparable

<?

super K

>

k

=

(

Comparable

<?

super K

>)

key

;

int cmp

=

k

.

compareTo

(

node

.

key

);