[코드트리/C++] 왕실의 기사 대결

왕실의 기사 대결

#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
using namespace std;
int L,N,Q;
int dx[]={-1,0,1,0};
int dy[]={0,1,0,-1};
vector <vector <int>> chess;
vector <vector <int>> knLo;

struct knight {
    int r;
    int c;
    int h;
    int w;
    int k;
    int dm;
};

vector <knight> Knight;

void move(int n,int d,set<int> s){
    for (auto element : s){
        // 기존 위치 다 0으로 설정
        for(int j=0;j<Knight[element].h;j++){
            for(int k=0;k<Knight[element].w;k++){
                knLo[Knight[element].r+j][Knight[element].c+k]=0;
            }
        }
    }
    for (auto element : s){
        // 위치정보 r,c 변경
        Knight[element].r+=dx[d];
        Knight[element].c+=dy[d];

        // 명령 받은애 아니면 데미지 계산, 죽나 확인
        if(element!=n){
            for(int j=0;j<Knight[element].h;j++){
                for(int k=0;k<Knight[element].w;k++){
                    if(chess[Knight[element].r+j][Knight[element].c+k]==1){ // 함정이라면
                        Knight[element].dm++; //데미지 계산
                        Knight[element].k--; // 체력 계산
                        if(Knight[element].k<=0){ // 체력 고갈된 경우 죽음
                            Knight[element].r=-1; 
                            Knight[element].c=-1;
                            Knight[element].h=-1;
                            Knight[element].w=-1;
                            break;
                        }
                    }
                }
                if(Knight[element].r==-1)break;
            }
        }
    }
    for (auto element : s){
        if(Knight[element].r!=-1){ //안죽었으면 다시 배치
            for(int j=0;j<Knight[element].h;j++){
                for(int k=0;k<Knight[element].w;k++){
                    knLo[Knight[element].r+j][Knight[element].c+k]=element;
                }
            }
        }
    }


    
}

void play(int n,int d){

    int r=Knight[n].r;
    int c=Knight[n].c;
    int h=Knight[n].h;
    int w=Knight[n].w;
    if(r==-1) return; // 마지막에 추가. 없으면 테스트케이스 18번에서 틀림

    queue <int> nextK;
    set <int> tmp;
    tmp.insert(n);
    int can=1;
    for(int j=0;j<h;j++){
        for(int k=0;k<w;k++){
            if(chess[r+j+dx[d]][c+k+dy[d]]==2){
                can=0;
                break;
            }
            int next= knLo[r+j+dx[d]][c+k+dy[d]];
            if(next!=0&&next!=n) {
                nextK.push(next);
                tmp.insert(next);
            }
        }
        if(can==0)break;
    }
    if(can==1){
        while(!nextK.empty()&&can==1){
            int nn=nextK.front();
            nextK.pop();
            for(int j=0;j<Knight[nn].h;j++){
                for(int k=0;k<Knight[nn].w;k++){
                    if(chess[Knight[nn].r+j+dx[d]][Knight[nn].c+k+dy[d]]==2){
                        can=0;
                        break;
                    }
                    int next= knLo[Knight[nn].r+j+dx[d]][Knight[nn].c+k+dy[d]];
                    
                    if(next!=0&&next!=nn) {
                        nextK.push(next);
                        tmp.insert(next);
                    }
                }
                if(can==0)break;
            }
        }
    }
    if(can) move(n,d,tmp);
    
}



int main() {
    cin>>L>>N>>Q;
    chess.resize(L+2,vector <int>(L+2,2));
    knLo.resize(L+1,vector <int>(L+1,0));
    Knight.resize(N+1); 

    for(int i=1;i<=L;i++){
        for(int j=1;j<=L;j++){
            cin>>chess[i][j];
        }
    }
    int r,c,h,w,k;
    for(int i=1;i<=N;i++){
        cin>>r>>c>>h>>w>>k;
        Knight[i].r=r;
        Knight[i].c=c;
        Knight[i].h=h;
        Knight[i].w=w;
        Knight[i].k=k;
        Knight[i].dm=0;
        for(int j=0;j<h;j++){
            for(int k=0;k<w;k++){
                knLo[r+j][c+k]=i;
            }
        }
    }

    int n,d,rr=0;
    while(rr++<Q){
        cin>>n>>d;
        play(n,d);
        // for(int i=1;i<=L;i++){for(int j=1;j<=L;j++){cout<<knLo[i][j];}cout<<endl;}
    }
    int ans=0;
    for(knight k: Knight){
        if(k.r!=-1) ans+=k.dm;
    }
    cout<<ans;
    return 0;
}

1. 처음에 옆에 있는 기사들을 큐에 저장했더니 같은 기사가 여러 번 저장돼서 데미지가 여러번 계산됐다. 중복 안되게 set로 바꿨다.
2. deque나 map이나 set 써볼때 문법 찾아봤다. 안보고도 쓸줄알기
3. if(r==-1) return; 이거 마지막에 추가했다. 안하면 틀림

테스트 케이스 여러개가 아니면 이런 사소한 부분들을 다 안틀리고 챙길 수 있을지 모르겠다🥲