POJ 2632 Crashing Robots【模拟】

刺骨的言语ヽ痛彻心扉 2022-10-16 09:57 144阅读 0赞

#include <iostream>
    
    using namespace std;
    
    #define debug(x) cout<<#x<<": "<<(x)<<endl;
    #pragma warning(disable:4996)
    
    int K, A, B, N, M;
    
    int a[1002][1002];
    
    struct rob { 
        int x, y;
        int d;
        // n 0 , e 1
    };
    rob rbs[1002];
    
    int getind(char c) { 
        if (c == 'N') { 
            return 0;
        }else if (c == 'E') { 
            return 1;
        }else if (c == 'S') { 
            return 2;
        }else if (c == 'W') { 
            return 3;
        }
        return -1;
    }
    
    int isok(rob& rb) { 
        if (rb.x <= 0 || rb.x >= A + 1
            || rb.y <= 0 || rb.y >= B + 1) { 
            return 0;
        }
        //debug(a[rb.y][rb.x])
    
        if (a[rb.y][rb.x] > 0) { 
            return a[rb.y][rb.x];
        }
        return -1;
    }
    
    bool cal() { 
        cin >> A >> B;
        cin >> N >> M;
    
        for (int i = 1; i <= N; ++i) { 
            cin >> rbs[i].x;
            cin >> rbs[i].y;
            char d;
            cin >> d;
            rbs[i].d = getind(d);
            a[rbs[i].y][rbs[i].x] = i;
        }
    
        bool isp = false;
    
        for (int i = 1; i <= M; ++i) { 
    
            int rid,rp;
            char ac;
            cin >> rid >> ac >> rp;
    
            rob& rb = rbs[rid];
    
            if (isp) { 
                continue;
            }
    
            if (ac == 'L') { 
                rb.d += 3*rp;
                rb.d %= 4;
            }else if (ac == 'R') { 
                rb.d += 1*rp;
                rb.d %= 4;
            }else if (ac == 'F') { 
    
                // -1 OK,0 wall ,other rbs
                for (int j = 0; j < rp; ++j) { 
    
                    int ox = rb.x;
                    int oy = rb.y;
    
                    a[oy][ox] = -1;
    
                    if (rb.d == 0) { 
                        rb.y += 1;
                    }
                    else if (rb.d == 1) { 
                        rb.x += 1;
                    }
                    else if (rb.d == 2) { 
                        rb.y -= 1;
                    }
                    else if (rb.d == 3) { 
                        rb.x -= 1;
                    }
    
                    int iso = isok(rb);
                    
                    if (iso >= 0) { 
                        if (isp == false) { 
                            if (iso == 0) { 
                                cout << "Robot " << rid << " crashes into the wall" << endl;
                            }
                            else if (iso > 0) { 
                                cout << "Robot " << rid << " crashes into robot " << iso << endl;
                            }
                            isp = true;
                        }
                        break;
                    }
                    else { 
                        a[rb.y][rb.x] = rid;
                    }
                }
            }
        }
    
        if (!isp) { 
            cout << "OK" << endl;
        }
    
        return 0;
    }
    
    int main()
    { 
        //freopen("../in2.txt","r",stdin);
        cin >> K;
        for (int i = 0; i < K; ++i) { 
            fill(a[0], a[0] + sizeof(a) / 4, -1);
            cal();
        }
    }

![在这里插入图片描述][20210529182601689.png]

[20210529182601689.png]: /images/20221014/7c83dadba8c645898bd9558b50d71980.png