[BZOJ2793][Poi2012]Vouchers

2793: [Poi2012]Vouchers

Time Limit: 20 Sec  Memory Limit: 64 MB
Submit: 316  Solved: 148
[Submit][Status][Discuss]

Description

考虑正整数集合，现在有n组人依次来取数，假设第i组来了x人，他们每个取的数一定是x的倍数，并且是还剩下的最小的x个。
正整数中有m个数被标成了幸运数，问有哪些人取到了幸运数。

 

Input

第一行一个正整数m (m<=1,000,000)，下面m行每行一个正整数x (x<=1,000,000)，表示x是一个幸运数。
接下来一行一个正整数n (n<=1,000,000)，下面n行每行一个正整数x (x<=1,000,000)，表示这一组来了x个人。

 

Output

第一行输出一个非负整数k，表示k个人取到了幸运数，下面k行依次表示取到幸运数的人的编号，人按照来的顺序从1开始编号。

 

Sample Input

4
1
6
8
16
3
4
2
4

Sample Output

3
2
4
6

HINT

 

Hint

总共来了10个人，他们取走的数依次是4 8 12 16 2 6 20 24 28 32。

第2、4、6个人取到的是幸运数8、16、6。

（别把这题想太难，其实很水的）

 

 

Source

鸣谢Oimaster

[Submit][Status][Discuss]

对于每一个x维护一个next数组，代表下一个x的位置，然后暴力处理即可。

简要估计一下上限复杂度$\sum\limits_{i=1}^n\frac mi=O(mlogn) 其中m=\max{a_i}$

不过肯定不会达到这个复杂度的

#include<cstdio>
#include<algorithm>
#define N 1000010
#define ll long long
using namespace std;
int n,next[N],maxn,tot,m,a[N];
ll ans[N],sum,vis[N];
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    scanf("%d",&a[i]),maxn=max(a[i],maxn);
    scanf("%d",&m);
    while(m--)
    {
        int x;
        scanf("%d",&x);
        ll now=sum;sum+=x;
        for(int i=x;i&&next[x]+x<=maxn;)
        if(!vis[next[x]+=x])
        vis[next[x]]=++now,i--;
    }
    for(int i=1;i<=n;i++)
    if(vis[a[i]])
    ans[++tot]=vis[a[i]];
    sort(ans+1,ans+tot+1);
    printf("%d\n",tot);
    for(int i=1;i<=tot;i++)
    printf("%lld\n",ans[i]);
}