인프런 커뮤니티 질문&답변

안녕하세요kim님ㅎㅎ

일단 endl을 쓰시면 안됩니다. "\n"을 써주세요. 이부분에 대한 설명은 교안 참고 부탁드립니다.

    for (int i=0; i<k; i++) {
        cin >> s;
        if (atoll(s.c_str()) == 0) 
            cout << mp_s[s] << "\n";
        else cout << mp_n[atoll(s.c_str())] <<"\n";

    }

앞의 코드처럼 고쳐보시겠어요?

또한 cin,cout을 쓰는데 시간초과가 나는 문제의 경우 이 부분을 추가시켜보면 풀리는 문제가 많습니다. (해당 부분도 교안 참고 부탁드립니다.)

int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL); cout.tie(NULL);
        cin >> n >> k;

이렇게 추가해보시겠어요?

나머지 코드는 다 잘 짜셨습니다.

또 질문 있으시면 언제든지 질문 부탁드립니다.

좋은 수강평과 별점 5점은 제가 큰 힘이 됩니다. :)

감사합니다.

강사 큰돌 올림.

안녕하세요, 인프런 AI 인턴이에요.

제가 보았을 때 이 코드는 인풋 개수(N)에 대해서 O(N^2)의 시간 복잡도를 가지고 있습니다. 따라서 인풋 개수가 커질수록 처리 시간이 급격히 증가하여 시간초과가 발생할 수 있습니다.

이유는 while문에서 조건을 갱신해주는 문장(inner index를 증가시키는 부분)을 빠뜨리셔서 N^2만큼 반복문이 돌아 시간 복잡도가 높아진 것 같습니다. 이를 수정하여 적절한 인덱스를 증가시켜주시면 더 효율적인 코드를 작성할 수 있을 것입니다.

또한, 입력 부분에서 input() 대신 sys.stdin.readline() 함수를 사용하시면 입력 시간을 조금 더 줄일 수 있으니 참고해주세요.

감사합니다.