ACM_蛋糕小王子铁头娃

Problem Description:

铁头娃制作了很多蛋糕,想分给他的小伙伴们,他的小伙伴很喜欢铁头娃做的蛋糕,每个人都想分到最多蛋糕
铁头娃想到了一个头铁主意:先给小伙伴们从1到N编号,在1-N区间随机选择一个数字K,每个编号取余K等于0的小伙伴会分
到一个蛋糕,重复多次,分蛋糕的过程有时铁头娃想皮一下,就会随机选择一个编号D的小伙伴,问他分到了多少块蛋糕。

Input:

输入数据有多组,每一组第一行包含两个数字N,M(1 <= N <= 50000 , 1 <= M <= 50000)
接下来M行,每一行包含两个数,
第一个数输入1则表示铁头娃要分蛋糕,接下来一个数字K,表示所有编号取余K为0的小伙伴分到一个蛋糕。
第一个数如果是2,表示铁头娃想皮一下,接下来一个数字D。

Output:

对于每次铁头娃的皮,输出D号小伙伴拥有的蛋糕

Sample Input:

5 5
1 1
1 3
1 4
2 2
2 3

Sample Output:

1
2
解题思路:刚开始做这道题时,不知道怎么把因子k的倍数的个数依次加1,(菜鸡弱,做题少,想不出来QAQ,生硬暴力结果TLE,流下了没技术的眼泪Orz)。当看到这个题解时恍然大悟,原来可以这么做。做法:定义一个一维数组,其元素类型是vector容器,用来存放每个数组元素下标的所有因子。再定义一个一维数组用来存放每个数被标记的次数。这里要转换一下求解思路:把分蛋糕看做是数字出现的次数,每次取编号k仅将其次数加1,当访问编号D元素分到多少个蛋糕时,只需将D中每个因子的个数加起来即为编号D分到蛋糕的个数,这样就大大降低了时间复杂度。
AC代码:
 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 const int maxn=50005;
 4 vector<int> vec[maxn];//把数组中每一个元素定义为vector容器,存放的是对应下标的所有因子
 5 void init(){
 6     for(int i=0;i<maxn;++i)vec[i].clear();
 7     vec[1].push_back(1);//1的因子只有1
 8     for(int i=2;i<=50000;++i){
 9         vec[i].push_back(1);
10         vec[i].push_back(i);//包括1和它本身的因子先放进容器
11     }
12     for(int i=2;i<=25000;++i)//i是j的因子
13         for(int j=i*2;j<=maxn;j+=i)//i的倍数即为j的因子
14             vec[j].push_back(i);
15 }
16 int n,m,x,y,cnt[maxn];//数组cnt记录每个因子的个数
17 int main(){
18     init();//打表
19     while(cin>>n>>m){
20         memset(cnt,0,sizeof(cnt));//初始化
21         while(m--){
22             cin>>x>>y;
23             if(x==1)cnt[y]++;//将y这个因子的个数加1
24             else{
25                 int ans=0;
26                 for(size_t i=0;i<vec[y].size();++i)
27                     ans+=cnt[vec[y][i]];//将D中每个因子的个数全部加起来
28                 cout<<ans<<endl;
29             }
30         }
31     }
32     return 0;
33 }

原文地址:https://www.cnblogs.com/acgoto/p/9061941.html