贪心-CF-720A

Posted 2020-09-22 Bcai

http://codeforces.com/problemset/problem/720/A

Closing ceremony

在二维平面坐标系中，给定一个n*m的矩形区域，矩形区域的两个定点为（1,1）、（n,m）。

给定两个队列q1、q2，保证q1、q2元素个数和为n*m。

现在需要将q1、q2队列中每个元素按照哈弗曼距离移动到矩形区域中的一个整数坐标点上，且每个元素有一个最大移动距离限制。

q1队列中的元素从坐标（0,0）出发，q2中的元素从坐标（0,m+1）出发。

问是否能够将所有元素移动至矩形区域内。

解题报告

思路

考虑只有一个队列的情况，此时必然有贪心原则：每次放在离起始点尽量远的地方。

而现在有两个队列，我们可以先考虑q1，将q1中的元素按距离限制升序排序，每次放在离两个起始点最远的地方开始填，即从( min(q1[i]-1,n),0 )开始填，同时注意距离限制。

q1填完后，用相似的方法填q2即可。

代码

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int maxn = 10004;

int q1[maxn], q2[maxn];
bool Map[maxn][maxn];
int n, m, k1, k2;

bool Work(int stx, int sty, int * q, int k){
    sort(q, q+k);
    int stj, dj;
    sty?(stj = m, dj = -1):(stj = 1, dj = 1);
    for(int p =0; p<k; p++){
        bool flag = false;
        if(q[p] < 2) return false;
        for(int i = min(q[p]-1, n); i>0; i--){
            for(int j = stj; j>0 && j<=m; j+=dj){
                if(abs(i-stx)+abs(j-sty) > q[p]) break;
                if(Map[i][j]) continue;
                Map[i][j] = true;
                flag = true;
                break;
            }
            if(flag) break;
        }
        if(!flag) return false;
    }
    return true;
}


int main(){
    scanf("%d%d", &n, &m);
    scanf("%d", &k1);
    for(int i=0; i<k1; i++){
        scanf("%d", q1+i);
    }
    scanf("%d", &k2);
    for(int i=0; i<k2; i++){
        scanf("%d", q2+i);
    }
    memset(Map, 0, sizeof(Map));
    if(Work(0, 0, q1, k1) && Work(0, m+1, q2, k2)){
        puts("YES");
    }else{
        puts("NO");
    }
    return 0;
}

--（完）--