洛谷2409Y的积木（二分答案）

题目背景

Y是个大建筑师，他总能用最简单的积木拼出最有创意的造型。

题目描述

Y手上有n盒积木，每个积木有个重量。现在他想从每盒积木中拿一块积木，放在一起，这一堆积木的重量为每块积木的重量和。现在他想知道重量和最小的k种取法的重量分别是多少。（只要任意更换一块积木，就视为一种不同的取法。如果多种取法重量总和一样，我们需要输出多次。）

输入输出格式

输入格式：
第一行输入两个整数，n,k，意义如题目所描述。

每组数据接下来的n行，第一个整数为mi，表示第i盒积木的数量，在同一行有mi个整数，分别表示每个积木的重量。

输出格式：
一行，重量最小的k种取法的重量，要求对于每个数据，从小到大输出

输入输出样例

输入样例#1：
3 10
4 1 3 4 5
3 1 7 9
4 1 2 3 5
输出样例#1：
3 4 5 5 6 6 7 7 7 7
说明

对于30%的数据：2<=mi<=10,1<=n<=10

对于50%的数据：2<=mi<=50,1<=n<=50

对于100%的数据：2<=mi<=100,1<=n<=100,1<=k<=10000,每个积木的重量为不超过100的正整数，所有mi的积大于等于k。本题不卡常。

【思路】在最小值与最大值之间二分答案mid（即第k小的数），若比mid小的个数大于等于k则向前搜索，反之向后搜索。

【注意】一定要找出准确的第k小的数，未到k位用mid补齐。

*注意数组大小

#include <iostream>
#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <queue>
#include <algorithm>
#define maxn 102
using namespace std;
int n,k,a[maxn][maxn],mn=0,mx=0,lst[maxn],tot,ans[maxn*maxn*maxn/2];
int dfs(int x,int lim,int sum){
    if(x==n+1){ans[++tot]=sum;return 1;}
    int res=0;
    for(int i=1;i<=a[x][0];++i){
        if(a[x][i]+sum+mn-lst[x]<lim)
          res=res+dfs(x+1,lim,sum+a[x][i]);
        else break;
        if(res>=k)return res;
    }
    return res;
}
bool okit(int lim){
    tot=0;
    int res=dfs(1,lim,0);
    if(res>=k)return 0;
    else return 1;
}
void erfen(int l,int r){
    while(l<r){
        int mid=(l+r)>>1;
        if(okit(mid))l=mid+1;
        else r=mid;
    }
    bool ok=okit(l);
    if(!ok){
        ok=okit(l-1);
        for(int i=tot+1;i<=k;++i)ans[i]=l-1;
    }
    else{
        for(int i=tot+1;i<=k;++i)ans[i]=l;
    }
}
int main(){
    scanf("%d%d",&n,&k);int tk=k;
    for(int i=1;i<=n;++i){
        scanf("%d",&a[i][0]);
        for(int j=1;j<=a[i][0];++j)scanf("%d",&a[i][j]);
        sort(a[i]+1,a[i]+a[i][0]+1);
        lst[i]=lst[i-1]+a[i][1];
        mn+=a[i][1];mx+=a[i][a[i][0]];
    }
    erfen(mn,mx);
    sort(ans+1,ans+1+k);
    for(int i=1;i<=k;++i)printf("%d ",ans[i]);
    return 0;
}