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

안녕하세요 파이팅님 ㅎㅎ

제 생각에는

                if (j > p) {
                    ret[i][j] = j - p;
                } else if (j < p) {
                    ret[i][j] = -1;
                } else if (j == p) {
                    ret[i][j] = -1;
                    p++;
                }

이부분에 대한 로직이 좀 꼬인 것 같은데요. 굳이 저렇게 해야하나 싶어서 좀 다듬어봤습니다.

제 생각에는 -1이 아닐 때 j - p가 되어야 하는 것 같아요.

1.처음에 그냥 -1로 다 초기화

2. 삼항연산식, 불필요한 로직 제거

    

또한 최대한 파이팅님이 이렇게 해야할것 같아. 라는 것을 유추해서 최적화해봤는데요.

#include <bits/stdc++.h>
using namespace std;

int main() {
    int h, w;
    cin >> h >> w;
    vector<vector<int>> a(h + 1, vector<int>(w + 1));
    vector<vector<int>> ret(h + 1, vector<int>(w + 1, -1));
    string s;

    for (int i = 1; i <= h; i++) {
        cin >> s;
        for (int j = 1; j <= w; j++) {
            a[i][j] = (s[j - 1] == 'c') ? 1 : 0;
        }
    }

    for (int i = 1; i <= h; i++) {
        int lastC = -1; 
        for (int j = 1; j <= w; j++) {
            if (a[i][j] == 1) {
                lastC = j;  
                ret[i][j] = 0;
            } else if (lastC != -1) {
                ret[i][j] = j - lastC;
            }
        }
    }

    for (int i = 1; i <= h; i++) {
        for (int j = 1; j <= w; j++) {
            cout << ret[i][j] << " ";
        }
        cout << "\n";
    }

    return 0;
}

참고부탁드립니다.

통과합니다가 뜹니다.

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

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

감사합니다.

강사 큰돌 올림.