题目描述
一个神秘的村庄里有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)能过。
题解:
①时间复杂度能够承受:50002,因此考虑暴力先讲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
终点其实是被彻底忘记,旅程不过是场善意的烛骨铭心。
你可曾记得我的孤寂,我们早已在起点各奔东西······ ——————汪峰《地心》