hdu 2639

#include <stdio.h>#define N 102#define M 1002int a[N], b[N], k, heap[M][31], hsize[M],c[33];inline void merge(int a[], int an, int w, int b[], int &bn ){    int cn = 0, i=1, j=1, d=0;    while( i<=an&&j<=bn&&d<k) if( a[i]+w == b[j] )i++;        else if(a[i]+w>b[j]) c[++d]=a[i++]+w;        else c[++d]=b[j++];    while(i<=an&&d<k) c[++d]=a[i++]+w;    while(j<=bn&&d<k) c[++d]=b[j++];    for( i=1; i<=d; i ++ ) b[i]=c[i];    bn=d;}int main(){    int cas, V, n, i, j,sum;    scanf("%d",&cas);    while( cas-- )    {        scanf("%d%d%d",&n,&V,&k);        for(i=0; i<n; i++) scanf("%d",&a[i]);        for(i=0; i<n; i++) scanf("%d",&b[i]);        for(i=1; i<=V; i++ ) hsize[i] = 0;        hsize[0]=1;        heap[0][1]=sum=0 ;        for(i=0; i<n; i ++  )        {            sum=b[i]+sum<V?b[i]+sum:V;            for(j=sum; j >= b[i]; j -- )            if( hsize[j-b[i]]>0 )                merge(heap[j-b[i]],hsize[j-b[i]],a[i],heap[j],hsize[j]);        }        for(i=V-1;i>=1;i-- ) merge(heap[i],hsize[i],0,heap[V],hsize[V]);        if(hsize[V]<k) puts("0");        else printf("%d\n",heap[V][k]);    }    return 0;}

堆实现

#include <stdio.h>#define N 102#define M 1002int a[N], b[N], k, heap[M][31], hsize[M];void keep_heap( int heap[], int s, int u ){    int x = u, l = u<<1, r = u<<1|1;    if( l<=s && heap[x] > heap[l] ) x = l;    if( r<=s && heap[x] > heap[r] ) x = r;    if( x != u )    {        l = heap[x],heap[x]=heap[u],heap[u]=l;        keep_heap(heap,s,x);    }}void insert_heap( int heap[31], int &s, int x ){    if( s == k && x <= heap[1] ) return;    for( int i = 1; i <= s; i ++ )        if( x == heap[i] ) return ;    if( s<k )    {        heap[++s] = x;        for( int u = s; u>1 && heap[u]<heap[u/2]; u/=2 )        x = heap[u], heap[u] = heap[u/2], heap[u/2] = x;    }    else if( s == k && heap[1]<x )    {        heap[1] = x;        keep_heap(heap,s,1);    }}int main(){    int cas, V, n, i, j,t,sum;    scanf("%d",&cas);    while( cas-- )    {        scanf("%d%d%d",&n,&V,&k);        for(i=0; i<n; i++) scanf("%d",&a[i]);        for(i=0; i<n; i++) scanf("%d",&b[i]);        for(i=1; i<=V; i++ ) hsize[i] = 0;        hsize[0]=1;        heap[0][1]=sum=0 ;        for(i = 0; i < n; i ++  )        {            sum=b[i]+sum<V?b[i]+sum:V;            for(j=sum; j >= b[i]; j -- )            if( hsize[j-b[i]]>0 )                for(t = hsize[j-b[i]]; t>=1; t-- )                insert_heap(heap[j],hsize[j],heap[j-b[i]][t]+a[i]);        }        for(i=V-1;i>=1;i-- )        for(j=hsize[i];j>=1; j-- )        insert_heap(heap[V],hsize[V],heap[i][j]);        if(hsize[V]<k) puts("0");        else printf("%d\n",heap[V][1]);    }    return 0;}