숫자 N개를 입력받아, 숫자끼리 비교하며 0~9 중 같은 숫자가 쓰이는 경우가 있다면 ans++ 해주면 된다.
앗... 자리수가 0~9까지로 단 10자리?!
둘 다 있는지 비교?!
이것은...
비트마스킹.
과거의 나는 비트마스킹을 정말 싫어하였지만
이제는 그것도 다 옛말.
나는 비트의 몸을 맡기는 비트마스커 뤂시우가 되었다.
N개의 수를 입력받을 때, 각각의 자리수로 for문을 한 번 더 돌려, 해당하는 수의 위치의 비트를 1로 바꿔준다.
예를 들어, 비트를 바꾸기 전
int number = 0;
으로 설정해준다.
해당 number을 이진수로 표현하자면
number
9
8
7
6
5
4
3
2
1
0
여부
0
0
0
0
0
0
0
0
0
0
일 것이다.
만약 입력받은 수가 32라면
String temp = "32";
temp.length()로 for문을 돌렸을 때, 처음 temp.charAt(POINT) 은 '3'일 것이다.
그렇다면, 우리가 number에 적용해야 하는 숫자는
temp
9
8
7
6
5
4
3
2
1
0
여부
0
0
0
0
0
0
1
0
0
0
일 것이다.
따라서 1을 왼쪽으로 3번 shift연산 후,
number |= temp.charAt(j) - '0';
number와 이를 합치면 된다!
2도 이와같이 진행하면, 최종적으로 number은
number
9
8
7
6
5
4
3
2
1
0
여부
0
0
0
0
0
0
1
1
0
0
1100 이렇게 될 것이다.
마찬가지로, 282도 이와같이 나타낼경우
number
9
8
7
6
5
4
3
2
1
0
여부
0
1
0
0
0
0
0
1
0
0
100000100 이렇게 될 것이다. 마지막 2를 적용할 때, 이미 2가 1이 있지만 합집합 연산하면 그대로 1이 적용된다.
이 숫자들을 HashMap nums에 갯수와 함께 넣는다!
그 후, HashMap을 돌면서 같은 nums에 대해 1개 이상일 경우 ans를 업데이트 해준다.
for (Map.Entry<Integer, Integer> entry:nums.entrySet()) {
int num = entry.getKey();
int cnt = entry.getValue();
ans += (long)cnt*(cnt-1)/2; //cnt 중 순서 없이 2개 뽑기
이 num을 다른 HashMap내 num 들과 비교해아하는데, 중복 방지를 위해 현재의 num보다 큰 애들과 비교해준다.
비교하는 수를 num = 32, num2 = 282라고 가정해보자.
두 수는 "2" 라는 공통된 수를 갖고있다. 이는 비트마스킹 된 값으로 쉽게 찾을 수 있는데
num
9
8
7
6
5
4
3
2
1
0
여부
0
0
0
0
0
0
1
1
0
0
이것은 32의 nums 값이고,
num2
9
8
7
6
5
4
3
2
1
0
여부
0
1
0
0
0
0
0
1
0
0
이것은 282의 nums 값이다.
이 둘을 교집합연산 (&) 하면
num&num2
9
8
7
6
5
4
3
2
1
0
여부
0
0
0
0
0
0
0
1
0
0
공통으로 갖는 2만 1이되고, 나머지는 0이 됨을 알 수 있다.
따라서 num&num2 연산을 했을 시 0이 아니라면, 공통된 수를 갖고 있는 것이다!
이때는 num이 가지고있는 갯수 중 하나와, num2가 가지고 있는 갯수 중 하나를 뽑아야되기 때문에 nums.get(num) * nums.get(num2) 로 ans를 업데이트한다.
for (int num2 : nums.keySet()) {
if(num >= num2) continue; //num보다 작은 경우 continue처리
if((num & num2) != 0) {
ans += (long) cnt*nums.get(num2);
}
}
이 과정을 모두 끝낸다면
맞으면 된다!
전체 코드
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;
public class 수업시간에교수님몰래교실을나간상근이_2825 {
public static void main(String[] args) throws NumberFormatException, IOException {
BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
int N = Integer.parseInt(bf.readLine());
Map<Integer, Integer> nums = new HashMap<>();
for(int i = 0; i < N; i++) {
String temp = bf.readLine();
int number = 0;
for(int j = 0; j < temp.length(); j++) {
number |= 1 << (temp.charAt(j) - '0');
}
nums.put(number, nums.getOrDefault(number, 0) + 1);
}
long ans = 0;
for (Map.Entry<Integer, Integer> entry:nums.entrySet()) {
int num = entry.getKey();
int cnt = entry.getValue();
ans += (long)cnt*(cnt-1)/2;
for (int num2 : nums.keySet()) {
if(num >= num2) continue;
if((num & num2) != 0) {
ans += (long) cnt*nums.get(num2);
}
}
}
System.out.println(ans);
}
}
제 코드 및 다른 팀원들의 코드를 내맘대로 리뷰하는 중 여러 Handler를 사용하면 코드가 더 깔끔하고 유지보수가 편해질 것 같아 상의도 없이 만들고 PR날린 그것,
ErrorHandler, Validator
기존 코드를 봅시다
@Service
@RequiredArgsConstructor
public class BattleServiceImpl implements BattleService {
//생략
@Override
public void registBattle(BattleInviteRequest battleInviteRequest, User user) {
//생략
if (battleInviteRequest.getOppositeUserId() == user.getId()) {
throw new IllegalArgumentException("Not valid user");
}
if (battleInviteRequest.getTime() <= 0) {
throw new RuntimeException("Wrong time setting");
}
아...
진짜 아... 싶죠? 분명 던지는 에러들이 서버 상 오류가아닌 비지니스로직 적 이루어질 수 있는 에러인데, 저렇게 서비스단에 2~3줄 심지어 if문으로 메서드 안에 넣으면...... 나중에 혹시라도 수정하게되거나 로직을 변경해야 되는 일이 생겼을 시 굉장히 귀찮아지겠죠?ㅎㅎ....
여기서 제가 생각해낸 방법은
1. 체크해야되는 조건들을 하나의 파일로 묶는다.
2. 오류 결과들을 하나의 파일에 정리한다
3. 던진다!
의 과정입니다.
먼저, 저 말도안되는 Not a valid user과 Wrong time setting을 과감히 지우고, "ErrorCode"의 Enum을 만들었습니다.
@Getter
@AllArgsConstructor
public enum ErrorCode {
INVALID_BATTLE_TIME("최소 10분부터 최대 60분까지 설정 가능합니다."),
INVALID_USER("유저 설정이 유효하지 않습니다."),
INVALID_BATTLE_STARTTIME("시작 시간은 최소 60분 후부터 가능합니다."),
INVALID_OPINION_LENGTH("선택지는 최대 16자까지 작성 가능합니다."),
DUPLICATED_BATTLE("이미 해당시간에 예정된 배틀이 존재합니다."),
ENDED_BATTLE("응답 기간이 종료된 배틀입니다.");
private final String message;
}
public class CustomException extends RuntimeException {
private final ErrorCode errorCode;
public CustomException(ErrorCode errorCode) {
super(errorCode.name());
this.errorCode = errorCode;
}
public ErrorCode getErrorCode() {
return errorCode;
}
}
그리고, 저 errorcode를 throw할 수 있는 custom exception을 정의해줍니다.
여기서 잠깐! RuntimeException을 extends하는 이유! 왜 하필 많은 오류 중 Runtime exception일까! 저는 굉장히 궁금했습니다. 사실 이름만 보면 그냥 Exception이 더 일반적으로 보이는데 쓰면 계속 throw를 하래잖아요! 그래서 패트병 대신 텀블러로 커피를 1회 마시고 GPT선생님께 여쭈어봤죠.
역시 우리으 친절하신 즤선생님,,
RuntimeException과 IllegalArgumentException과 같이 평소에 우리가 사용해도 try-catch 등을 쓰지 않아도 되는 Exception들은 Unchecked Exception이라고 합니다! 그에 반해, 일반 Exception은 Checked Exception이라고 try-catch나 throws를 해야된다고 하네요.
그래서 아무튼 Exception이 아닌 것들 중 가장 무난한 RuntimeException을 채택하였습니다.
그 후 원래 제가 만들어놨던 GlobalExceptionHandler 안에 제 작고 소중한 CustomException도 추가해줬어요...><
@RestControllerAdvice
public class GlobalExceptionHandler {
@ExceptionHandler(RuntimeException.class)
public ResponseEntity<ApiResponseDto> handleRuntimeException(RuntimeException error) {
ApiResponseDto res = new ApiResponseDto("fail", error.getMessage(), null);
return new ResponseEntity<>(res, HttpStatus.BAD_REQUEST);
}
//생략
@ExceptionHandler(CustomException.class)
public ResponseEntity<ApiResponseDto<String>> handleCustomException(CustomException error) {
ApiResponseDto<String> response = new ApiResponseDto<>("fail", error.getErrorCode().getMessage(), null);
return new ResponseEntity<>(response, HttpStatus.BAD_REQUEST);
}
}
첫번째 메서드가 제가 직접 사용하던 RuntimeException 이고요, 밑이 제가 이제 사용하는 CustomException입니다!!
사실 기능상의 차이는 없어요~! RuntimeException도 message를 반환하고, 제 custom handler도 완전 같은 리턴타입이기에..
하지만, 귀엽잖아요.
다꾸, 폰꾸에 이어 에꾸(에러 꾸미기)까지... 즐겁지않나요?><
그리고 이제 Error을 Throw해주는데 사용될 Validator을 만들어줍니다~
@Component
@AllArgsConstructor
public class BattleValidator {
static Date now = new Date();
static Calendar calendar = Calendar.getInstance();
private final BattleRepository battleRepository;
public void validateOppositeUser(BattleInviteRequest battleInviteRequest, User user) {
if (battleInviteRequest.getOppositeUserId() == user.getId()) {
throw new CustomException(ErrorCode.INVALID_USER);
}
}
public void validateTime(int minutes) {
if (minutes < 10) {
throw new CustomException(ErrorCode.INVALID_BATTLE_TIME);
}
}
아까랑 똑같은 코드에 똑같은 결과를 반환하는데, 이렇게 설정하는것이 훨씬 간지작살나지 않나요?
이제 서비스단에 이 Validator 내 메서드를 적용하면...!!
@Transactional
@Override
public void registBattle(BattleInviteRequest battleInviteRequest, User user) {
//생략...
battleValidator.validateOppositeUser(battleInviteRequest, user);
battleValidator.validateTime(battleInviteRequest.getTime());
battleValidator.validateStartTime(battleInviteRequest.getStartDate());
battleValidator.checkOtherBattles(user, battleInviteRequest.getStartDate(), endDate);
}
TADA!!!!!
나만의 귀여운 코꾸(코드꾸미기...), 완성!!!
아까 말도안되는 if문과 비교해보면,, 훨씬 귀엽고 깜찍해졌죠?
이렇게 작성한 코드는 후에 다른사람이 보았을 때도 이해하기 쉽고, 유지보수도 용이해집니다~
사실 저는 1년전 이맘때쯤, 굉장한 홍머병에 걸렸었는데요,
그때는 플젝때도 남들이 내 코드를 이해하기 어렵고, 건드리지 못하는 것이 뭔가 고수같고 개짱짱맨 코드인 줄 알았답니다...
하하 코드는 무조건 간결!!해야하고 모듈화가 잘 되어있어야합니다~
이거는 알림 기능에서도 적용할 수 있는데요!
어떠한 알림을 보낼 지 프론트님님님들과 상의 후
public enum NotificationType {
BATTLE_REQUEST(0, "%s님이 배틀을 신청했어요! 지금 바로 확인해보세요."),
LIVE_NOTICE(1, "[%s] 라이브 시작 5분 전입니다!"),
BATTLE_ACCEPT(2, "[%s] 배틀이 승낙되어 대기중입니다."),
BATTLE_DECLINE(3, "[%s] 배틀이 거절되었습니다."),
BATTLE_UNSATISFIED(4, "[%s] 배틀이 인원수를 충족하지 못해 모닥불로 이동하였습니다."),
BATTLE_SATISFIED(5, "[%s] 배틀이 인원수를 충족하여 불구경 대기중입니다.");
//todo 인원 수 미달 -> 모닥불 (참여자, 개최자)
//todo 인원 수 충족 -> 라이브 개최 (참여자, 개최자)
private final int code;
private final String messageTemplate;
NotificationType(int code, String messageTemplate) {
this.code = code;
this.messageTemplate = messageTemplate;
}
public int getCode() {
return code;
}
public String getMessageTemplate() {
return messageTemplate;
}
}
이렇게 enum으로 정리해놓으며, 추가/수정하면 편하답니다~
그럼 여기서 퀴즈! 알림 메서드들의 문구들을 어떻게하면 더 그럴싸하게 작성할 수 있을까요?(제발도와주세요 저거 넘 구린거같아)
