(C++) - 백준(BOJ) 9455번 : 박스 답

#include <iostream>
using namespace std;
int M, N, a[101][101], t,ans,cnt;
int main() {
    cin >> t;
    while (t--)
    {
        cin >> M >> N;
        for (int i = 1; i <= M; i++)
            for (int j = 1; j <= N; j++)
                a[i][j] = 0;
        ans = 0;
        for (int i = 1; i <= M; i++)
            for (int j = 1; j <= N; j++)
                cin >> a[i][j];
        for (int j = 1; j <= N; j++)
        {
            for (int i = M; i >= 1; i--)
            {
                cnt = 0;
                if (a[i][j] == 1)//행의 원소가 1이라면
                {
                    for (int k = i+1; k <= M; k++)//1이나올때까지 아래로 내려가며 0의 개수를 세준다
                    {
                        if (a[k][j] == 1) { break; }
                        else
                            cnt++;
                    }
                    ans += cnt;
                //마지막 1이 나오기 직전 0의 인텍스와 처음 1이 나온 인덱스의 값을 바꿔서 박스를 옮겨준다
                    swap(a[i+cnt][j], a[i][j]);
                }
            }
        }
        cout << ans << '\n';
    }
}