[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=;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=;i<=n;i++)
     if(vis[a[i]])
     ans[++tot]=vis[a[i]];
     sort(ans+,ans+tot+);
     printf("%d\n",tot);
     for(int i=;i<=tot;i++)
     printf("%lld\n",ans[i]);
 }