【CZY选讲·吃东西】

最新推荐文章于 2022-10-27 15:29:02 发布

转载最新推荐文章于 2022-10-27 15:29:02 发布 · 138 阅读

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/Damitu/p/7654414.html

题目描述

一个神秘的村庄里有4家美食店。这四家店分别有A,B,C,D种不同的美食。LYK想在每一家店都吃其中一种美食。每种美食需要吃的时间可能是不一样的。现在给定第1家店A种不同的美食所需要吃的时间a1,a2,…,aA。给定第2家店B种不同的美食所需要吃的时间b1,b2,…,bB。以及c和d。LYK拥有n个时间，问它有几种吃的方案。

数据范围

1<=n<=100000000，1<=A,B,C,D<=5000,0<=ai,bi,ci,di<=100000000。

此题空间足够大，时间2s。

Ps.不要在意复杂度，事实证明O(n)能过。

题解：
①时间复杂度能够承受：5000²，因此考虑暴力先讲AB CD两组的组合结果求出来。

②对于得到的两个数组进行组合，首先肯定要排序。

③排序使用桶排序O(n)。然后维护两个指针，一个在1数组开头,一个在2数组结尾。

④根据单调性,两个指针只会向相反方向移动，时间复杂度O(n)

#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#include <string>
#include <cstring>
using namespace std;
int f[100000005],c[25000005],cc[25000005];
int A,B,C,D,n,a[5005],b[5005],aa[5005],bb[5005],MAX,cnt,cntt,i,j,now;
long long ans;
int main()
{
    freopen("eat.in","r",stdin);
    freopen("eat.out","w",stdout);
    scanf("%d%d%d%d%d",&n,&A,&B,&C,&D);
    if (n==0 && A==0 && B==0 && C==0 && D==0) return 0;
    for (i=1; i<=A; i++) scanf("%d",&a[i]);
    for (i=1; i<=B; i++) scanf("%d",&b[i]);
    MAX=0;
    cnt=cntt=0;
    for (i=1; i<=A; i++)
        for (j=1; j<=B; j++)
            if (a[i]+b[j]<=n)
            {
                f[a[i]+b[j]]++;
                MAX=max(MAX,a[i]+b[j]);
            }
    for (i=0; i<=MAX; i++)
    while (f[i])
        {
            f[i]--;
            c[++cnt]=i;
        }

    for (i=1; i<=C; i++) scanf("%d",&aa[i]);
    for (i=1; i<=D; i++) scanf("%d",&bb[i]);
    MAX=0;
    for (i=1; i<=C; i++)
        for (j=1; j<=D; j++)
            if (aa[i]+bb[j]<=n)
            {
                f[aa[i]+bb[j]]++;
                MAX=max(MAX,aa[i]+bb[j]);
            }
    for (i=0; i<=MAX; i++)
    while (f[i])
        {
            f[i]--;
            cc[++cntt]=i;
        }
    for (i=cntt; i>=1; i--) if (c[1]+cc[i]<=n) break;
    now=i;
    for (i=1; i<=cnt; i++)
    {
        ans+=now;
        while (now&& c[i+1]+cc[now]>n) now--;
    }
    cout<<ans<<endl;
    ans=0;
    return 0;
}//czy020202