糯米

TI DaVinci, gstreamer, ffmpeg
随笔 - 167, 文章 - 0, 评论 - 47, 引用 - 0
数据加载中……

POJ 2010 Moo University - Financial Aid 堆

昨天做了2008,今天准备做2009。但是看了下题目,发现爆难,才100个人过。
觉得这种题还是别碰了,等以后牛逼了再做。
于是跳过2008年,直接到2010年了!呵呵。

这题还是算容易的,比较适合自己水平发挥,用堆来做,速度尚可 188ms 。


思路:
先把牛按照score排序一下,然后从后往前找,把每一头牛当做是位于中间的那头牛。
那现在就是求:
该头牛前面的所有牛中,哪 (N - 1) / 2 头牛aid值的和最小。
该头牛后面的所有牛中,哪 (N - 1) / 2 头牛aid值的和最小。
这就是典型的用堆可以解决的问题了。

#include <stdio.h>
#include 
<stdlib.h>

#define MAX_C 100032
#define MAX_N 20032

struct node {
    
int score, aid;
}
;
struct node in[MAX_C];
int N, C, F;
int after[MAX_C], before[MAX_C];
int heap_size, heap_sum, heap[MAX_N];

int cmp(const void *a, const void *b)
{
    
return ((struct node *)a)->score - ((struct node *)b)->score;
}


__inline 
void shift_down(int idx)
{
    
int val = heap[idx];
    
while (1{
        idx 
*= 2;
        
if (idx > heap_size)
            
break ;
        
if (idx + 1 <= heap_size && heap[idx + 1> heap[idx])
            idx
++;
        
if (heap[idx] <= val)
            
break;
        heap[idx 
/ 2= heap[idx];
    }

    heap[idx 
/ 2= val;
}


__inline 
int heap_init(int start, int len)
{
    
int i;

    heap_sum 
= 0;
    
for (i = start; i < start + len; i++{
        heap[i 
- start + 1= in[i].aid;
        heap_sum 
+= in[i].aid;
    }

    
for (i = heap_size / 2; i >= 1; i--
        shift_down(i);
    
return heap_sum;
}


__inline 
int heap_update(int aid)
{
    
if (aid < heap[1]) {
        heap_sum 
-= heap[1- aid;
        heap[
1= aid;
        shift_down(
1);
    }

    
return heap_sum;
}


int main()
{
    
int i;

    freopen(
"e:\\test\\in.txt""r", stdin);

    scanf(
"%d%d%d"&N, &C, &F);
    
for (i = 0; i < C; i++)
        scanf(
"%d%d"&in[i].score, &in[i].aid);
    qsort(
in, C, sizeof(in[0]), cmp);
    
    heap_size 
= (N - 1/ 2;
    before[heap_size 
- 1= heap_init(0, heap_size);
    
for (i = heap_size; i < C; i++
        before[i] 
= heap_update(in[i].aid);
    after[C 
- heap_size] = heap_init(C - heap_size, heap_size);
    
for (i = C - heap_size - 1; i >= 0; i--)
        after[i] 
= heap_update(in[i].aid);
    
for (i = C - heap_size - 1; i - heap_size >= 0; i--{
        
if (in[i].aid + before[i - 1+ after[i + 1<= F)
            
break;
    }

    printf(
"%d\n", i - heap_size < 0 ? -1 : in[i].score);

    
return 0;
}

posted on 2010-03-13 19:25 糯米 阅读(602) 评论(0)  编辑 收藏 引用 所属分类: POJ


只有注册用户登录后才能发表评论。
网站导航: 博客园   IT新闻   BlogJava   知识库   博问   管理