A1128 | 逻辑想象能力、简洁高效美观的代码、memset的使用情景

#include <stdio.h>
#include <memory.h>
#include <math.h>
#include <string>
#include <vector>
#include <set>
#include <stack>
#include <queue>
#include <algorithm>
#include <map>


#define I scanf
#define OL puts
#define O printf
#define F(a,b,c) for(a=b;a<c;a++)
#define FF(a,b) for(a=0;a<b;a++)
#define FG(a,b) for(a=b-1;a>=0;a--)
#define LEN 10000
#define MAX 0x06FFFFFF
#define V vector<int>

using namespace std;

int C[3][LEN];//col, diag1, diag2

int main(){
//    freopen("d:/input/A1128.txt","r",stdin);
    int i,n,m;
    scanf("%d",&n);
    while(n-->0){
        scanf("%d",&m);
        memset(C,0,sizeof(C));
        bool ok=1;
        F(i,1,m+1){
            int r=i;
            int c;
            scanf("%d",&c);
            int d1=r-c+m;
            int d2=c-r+m;
            if(!ok) continue;
            if(C[0][c] || C[1][d1] || C[2][d2]){
                ok=0;
            }else{
                C[0][c]=1;
                C[1][d1]=1;
                C[2][d2]=1;
            }
        }
        OL(ok ? "YES" : "NO");
    }
    return 0;
}