https://school.programmers.co.kr/learn/courses/30/lessons/118667?language=java
프로그래머스
코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.
programmers.co.kr
내 답안 :
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
|
import java.util.*;
class Solution {
public int solution(int[] queue1, int[] queue2) {
Queue<Integer> que1 = new LinkedList<>();
Queue<Integer> que2 = new LinkedList<>();
long total = 0;
long sumOfOne = 0;
for(int i=0; i<queue1.length; i++){
total += queue1[i];
total += queue2[i];
sumOfOne += queue1[i];
que1.add(queue1[i]);
que2.add(queue2[i]);
}
int maxCount = queue1.length * 3;
total/=2;
// 큐 하나의 합이 전체 합의 절반이 아니면
// maxCount가 0이 되면 -1 리턴
while(sumOfOne!=total){
if(maxCount==0){
return -1;
}
if(sumOfOne>total){
int temp1=que1.poll();
sumOfOne-=temp1;
que2.add(temp1);
}
else{
int temp2=que2.poll();
sumOfOne+=temp2;
que1.add(temp2);
}
maxCount--;
}
return queue1.length*3-maxCount;
}
}
|
cs |
배열, ArrayList가 아닌 LinkedList를 사용한 이유?
LinkedList는 노드의 주소를 가지고 서로 포인터로 가리키는 방식으로 연결되어있기 때문에 데이터의 삭제와 추가가 매우 간편
개선안 :
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
|
import java.util.*;
class Solution {
static int func(Deque<Integer> q1, Deque<Integer> q2, long sum1, long sum2){
int cnt =0;
long targ = (sum1+sum2)/2;
int len = q1.size()*3;
int idx= 0;
while(idx<len){
idx++;
if(sum1>sum2){
int val = q1.poll();
q2.add(val);
sum1-=val;
sum2+=val;
cnt++;
}else if(sum1<sum2){
int val = q2.poll();
q1.add(val);
sum1+=val;
sum2-=val;
cnt++;
}
else return cnt;
}
return -1;
}
public int solution(int[] queue1, int[] queue2) {
int answer = -2;
long sum1 =0;
long sum2 =0;
Deque<Integer> q1 = new ArrayDeque<>();
Deque<Integer> q2 = new ArrayDeque<>();
for(int i =0;i<queue1.length;i++){
sum1+=queue1[i];
sum2+=queue2[i];
q1.add(queue1[i]);
q2.add(queue2[i]);
}
if(sum1==sum2)return 0;
answer=func(q1,q2,sum1,sum2);
return answer;
}
}
answer=func(q1,q2,sum1,sum2);
return answer;
}
}
|
cs |
내가 작성한 코드보다 빠른 처리속도
두 큐 값이 같지 않은 경우 원소 빼고 넣는 과정을 func함수로 정의,
두 큐 값의 합을 비교하는 과정이 더욱 깔끔해져서 빨라진 것으로 추정