NOIP 2008 - 排座椅

题目描述

      上课的时候总有一些同学和前后左右的人交头接耳，这是令小学班主任十分头疼的一件事情。不过，班主任小雪发现了一些有趣的现象，当同学们的座次确定下来之后，只有有限的D对同学上课时会交头接耳。同学们在教室中坐成了M行N列，坐在第i行第j列的同学的位置是（i，j），为了方便同学们进出，在教室中设置了K条横向的通道，L条纵向的通道。于是，聪明的小雪想到了一个办法，或许可以减少上课时学生交头接耳的问题：她打算重新摆放桌椅，改变同学们桌椅间通道的位置，因为如果一条通道隔开了两个会交头接耳的同学，那么他们就不会交头接耳了。        请你帮忙给小雪编写一个程序，给出最好的通道划分方案。在该方案下，上课时交头接耳的学生对数最少。

输入

输入的第一行，有5各用空格隔开的整数，分别是M，N，K，L，D（2< =N，M< =1000，0< =K< M，0< =L< N，D< =2000）。  接下来D行，每行有4个用空格隔开的整数，第i行的4个整数Xi，Yi，Pi，Qi，表示坐在位置(Xi,Yi)与(Pi,Qi)的两个同学会交头接耳（输入保证他们前后相邻或者左右相邻）。  输入数据保证最优方案的唯一性。

输出

第一行包含K个整数，a1a2……aK，表示第a1行和a1+1行之间、第a2行和第a2+1行之间、…、第aK行和第aK+1行之间要开辟通道，其中ai<   ai+1，每两个整数之间用空格隔开（行尾没有空格）。  第二行包含L个整数，b1b2……bk，表示第b1列和b1+1列之间、第b2列和第b2+1列之间、…、第bL列和第bL+1列之间要开辟通道，其中bi<   bi+1，每两个整数之间用空格隔开（行尾没有空格）。

样例输入

4 5 1 2 3 4 2 4 3 2 3 3 3 2 5 2 4

样例输出

2 2 4

贪心也可，DP也可，把横向的和纵向的分开计算。

#include <cstdio>
#include <algorithm>
using namespace std;

struct seat
{
    int v;
    int n;
};
seat row[1005], col[1005];
int ans_r[1005], ans_c[1005];

int cmp_seat(const seat &a, const seat &b)
{
    return a.v > b.v;
}

int cmp_ans(const int a, const int b)
{
    return a < b;
}

int main()
{
    int M, N, K, L, D;
    int x1, x2;
    int y1, y2;
    scanf("%d%d%d%d%d", &M, &N, &K, &L, &D);
    for (int i = 1; i <= 1000; ++i)
    {
        row[i].n = col[i].n = i;
        row[i].v = col[i].v = 0;
    }
    for (int i = 1; i <= D; ++i)
    {
        scanf("%d%d%d%d", &x1, &y1, &x2, &y2);
        if (x1 - x2 == 1 || x2 - x1 == 1)
            row[min(x1, x2)].v++;
        else
            col[min(y1, y2)].v++;
    }
    sort(row + 1, row + N + 1, cmp_seat);
    sort(col + 1, col + N + 1, cmp_seat);

    for (int i = 1; i <= K; ++i)
        ans_r[i] = row[i].n;
    for (int i = 1; i <= L; ++i)
        ans_c[i] = col[i].n;

    sort(ans_r + 1, ans_r + K + 1);
    sort(ans_c + 1, ans_c + L + 1);
    for (int i = 1; i <= K; ++i)
    {
        printf("%d", ans_r[i]);
        if (i < L) printf(" ");
    }
    printf("\n");
    for (int i = 1; i <= L; ++i)
    {
        printf("%d", ans_c[i]);
        if (i < L) printf(" ");
    }
    return 0;
}