洛谷 P1056 排座椅 题解

我终于走进了普及组的大门
这是一道很水的排序题，矩阵大小限制在1000*1000内，所以用桶排很容易解决
然后先按照每行能隔开多少对同学cnt排序
排序之后还要对序号num排序，因为答案要求升序输出（洛谷的排版有点问题，第一次没有看明白条件WA掉了）
用了stdlib里的qsort()

#include
#include
#define _for(i,j,k) for(int i=j;i
#define _rep(i,j,k) for(int i=j;i<=k;i++)

typedef struct{
    int num;
    int cnt;
} buk;
buk K[1002]={0},L[1002]={0};

int cmp(const void *a,const void *b)	//对同学对数进行排序
{
    buk *aa=(buk *)a;
    buk *bb=(buk *)b;
    return bb->cnt-aa->cnt;
}

int cmp2(const void *a,const void *b)	//对输出的答案排序号
{
    buk *aa=(node *)a;
    buk *bb=(node *)b;
    return aa->num-bb->num;
}

int main()
{
    _for(i,1,1000) {			//桶初始化
        K[i].num=i;
        L[i].num=i;
    }
    int m,n,k,l,d;
    scanf("%d%d%d%d%d",&m,&n,&k,&l,&d);
    int x1,y1,x2,y2;
    _for(i,0,d) {
        scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
        if (x1==x2)			//因为答案保证相邻，所以只需简单判断
            L[y1<y2?y1:y2].cnt++;
        else
            K[x1<x2?x1:x2].cnt++;
    }
    
    qsort(K+1,999,sizeof(buk),cmp);	//见正文
    qsort(K+1,k,sizeof(buk),cmp2);
    qsort(L+1,999,sizeof(buk),cmp);
    qsort(L+1,l,sizeof(buk),cmp2);
    
    _for(i,1,k)
        printf("%d ",K[i].num);
    printf("%d\n",K[k].num);		//题目要求行尾不能有空格
    _for(i,1,l)
        printf("%d ",L[i].num);
    printf("%d",L[l].num);
    return 0;
}