강의로 돌아가기
진성호

효율성 테스트 시간은 통과인데...

시간은 통과인것 같은데 정작 효율성 테스트의 정답을 틀리는 것 같네요 ㅋㅋㅋ;;;
다른 테스트는 다 통과 나왔는데 뭐가 문제일까요?

작성중인 코드―solution.cpp
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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
#include <string>
#include <vector>
#include <algorithm>

using namespace std;

long long solution(int k, vector<int> w) 
{
    long long answer = 0;
    int cnt = 1;
    int tmp;
    int i, j;
    int r, l;
    int sum;

    sort(w.begin(), w.end());
    sum = w[w.size() - 1];

    for(i = w.size() - 1; i > 0; i--, cnt++)
    {
        sum += w[i - 1];
        tmp = w[i] - w[i - 1];
        k -= tmp * cnt;

        if(k > 0)
            tmp = 0;
        else
        {
            if(k == 0)
            {
                for(j = 0; j < i; j++)
                {
                    answer += w[j] * w[j];
                }

                answer += w[i - 1] * w[i - 1] * cnt;
                break;
            }
            else
            {
                for(j = 0; j < i; j++)
                {
                    answer += w[j] * w[j];
                }

                r = (-k / cnt);
                l = (-k % cnt);

                tmp = w[i - 1] + r;
                answer += tmp * tmp * (cnt - l);
                tmp++;
                answer += tmp * tmp * l;
                break;
            }
        }
    }

    if(i == 0 && sum > k)
    {
        cnt = w.size();
        r = (k / cnt);
        l = (k % cnt);

        tmp = w[0] - r;
        answer += tmp * tmp * (cnt - l);
        tmp--;
        answer += tmp * tmp * l;
    }
    else if(i == 0)
    {
        answer = 0;
    }

    return answer;
}
0 개의 답변
답변 쓰기
이 입력폼은 마크다운 문법을 지원합니다. 마크다운 가이드 를 참고하세요.