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