[level 1] Title: 기사단원의 무기, Time: 80.44 ms, Memory: 93.8 MB -BaekjoonHub

Author: wonseokyoon

프로그래머스/1/136798. 기사단원의 무기/README.md

@@ -4,7 +4,7 @@
 
 ### 성능 요약
 
-메모리: 95 MB, 시간: 10.96 ms
+메모리: 93.8 MB, 시간: 80.44 ms
 
 ### 구분
 
@@ -16,7 +16,7 @@
 
 ### 제출 일자
 
-2025년 01월 06일 18:56:35
+2025년 08월 23일 09:34:19
 
 ### 문제 설명
 

프로그래머스/1/136798. 기사단원의 무기/기사단원의 무기.java

@@ -1,22 +1,51 @@
 import java.util.*;
 class Solution {
- public int solution(int number, int limit, int power) {
-        int[] count=new int[number+1];
+    public int solution(int number, int limit, int power) {
+        // 15 -> 약수:1,3,5,15 -> 공격력 4짜리 무기 구매(단, 제한 존재)
+        // 공격력 1당 1Kg 철 필요
+        
+        // number: 기사 단원 수
+        // limit: 공격력 제한 수치
+        // power: 제한 수치를 초과한 기사가 사용할 공격력
+        // 필요한 철의 무게 return
+        
+        // 1. 1부터 number까지 약수의 갯수를 구함
+        ArrayList<Integer> list = new ArrayList<>();
         for(int i=1;i<=number;i++){
-            for(int j=1;j<=number/i;j++){
-                count[i*j]++;
+            // 1. 1부터 number까지 약수의 갯수를 구함
+            int count = getCount(i);
+            // 2. 약수의 갯수가 limit을 초과하는지 확인
+            if(count > limit){
+                // 3. 만약에 limit보다 크다면 power를 리스트에 등록                
+                list.add(power);
+                continue;
             }
+            list.add(count);
         }
-        int sum=0;
-        for(int n:count){
-            if(n>limit){
-                sum+=power;
+        // 4. list의 합 return
+        int sum = list.stream()
+            .mapToInt(Integer::intValue)
+            .sum();
+        return sum;
+    }
+    
+    int getCount(int number){
+        if(number == 1){
+            return 1;
+        }
+        
+        int sqrt = (int) Math.sqrt(number);
+        int count = 0;
+        
+        for(int i=1;i <= sqrt;i++){
+            if(i*i == number){
+                count++;
             }
-            else{
-                sum+=n;
+            else if(number % i == 0){
+                count+=2;
             }
         }
-        return sum;
-
+        
+        return count;
     }
 }