글 읽기 - 반복문 중 iterator 삭제

pmch0325 3년 전 0

아래 코드는 틀린코드이긴 한데 틀린 부분은 일단 넘어가고 이해안되는 점이 있습니다.

43번 줄부터 iterator 변수인 iter에서 vector의 end까지 삭제하는 부분이 있는데 여기서 종료조건이 제대로 작동을 안합니다...

vector의 erase함수는 삭제한 iterator의 다음 iterator를 반환하는 걸로 알고있어서

제 의도대로라면 vector원소가 1개인 경우에 그 원소를 삭제하고 iter는 end()를 가리키게 되어서 while문에서 나와야 하는데

3번정도 의미없는 반복을 하다가 에러가 납니다. 뭐가 틀린걸까요????

#include <iostream>
#include <vector>
#include <algorithm>
#include <fstream>

using namespace std;

typedef struct{int r,c,dir;}mal;
int arr[13][13];
mal mals[10];
vector<int> board[13][13];
int dr[]={0,1,0,-1}, dc[]={1,0,-1,0}, chdir[]={0,2,1,4,3};
int N,K, ans=1;

void solution(){
    int nr,nc;
    bool retry = false;
    while( ans<1001 ){
        for(int i=0; i<K; i++){
            // 칸 이동
            nr = mals[i].r + dr[mals[i].dir];
            nc = mals[i].c + dc[mals[i].dir];
            
            // 파란 칸이거나 범위 밖일 때
            if( nr==-1 || nr==N || nc==-1 || nc==N || arr[nr][nc]==2 ){
                if( retry ){
                    retry = false;
                    continue;
                }
                retry = true;
                mals[i].dir = chdir[mals[i].dir];
                i--;
                continue;
            }
            retry = false;
            
            // 종료조건(틀림)
            if( board[mals[i].r][mals[i].c].size() + board[nr][nc].size() > 3 ){
                cout<<ans;
                return;
            }
            
            // 말 이동
            auto iter = find(board[mals[i].r][mals[i].c].begin(), board[mals[i].r][mals[i].c].end(), i);
            if( arr[nr][nc]==1 ) reverse(iter, board[mals[i].r][mals[i].c].end());
            while( iter!=board[mals[i].r][mals[i].c].end() ){
                board[nr][nc].push_back(*iter);
                mals[*iter].r = nr, mals[*iter].c = nc;
                iter = board[mals[i].r][mals[i].c].erase(iter);
            }
        }
        ans++;
    }
    cout<<-1;
}

int main(){
    // ios::sync_with_stdio(false);
    // cin.tie(0);
    /*in>>N>>K;
    for(int i=0; i<N; i++){
        for(int j=0; j<N; j++){
            cin>>arr[i][j];
        }
    }
    for(int i=0; i<K; i++){
        cin>>mals[i].r>>mals[i].c>>mals[i].dir;
        board[mals[i].r][mals[i].c].push_back(i);
    }*/
    ifstream in("test.txt");
    if( in.is_open() ){
        in>>N>>K;
        for(int i=1; i<=N; i++){
            for(int j=1; j<=N; j++){
                in>>arr[i][j];
            }
        }
        for(int i=0; i<K; i++){
            in>>mals[i].r>>mals[i].c>>mals[i].dir;
            board[mals[i].r][mals[i].c].push_back(i);
        }
    }
    solution();
    return 0;
}

pmch0325 3년 전 0

조건문에 쓰이는 변수가 반복문 안에서 바뀌어서 꼬여버렸네요

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

반복문 중 iterator 삭제

pmch0325 3년 전 0

pmch0325 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말