链接:https://ac.nowcoder.com/acm/contest/7604/C
来源:牛客网
题目描述
牛牛的班级中有n个人,他们的性格各不相同。
牛牛现在想要从这n个人中选出一些人组成一个兴趣小组,但是他想让参加这个兴趣小组的人数尽可能的多。但是他有不想让其中有任何一对人之间由于性格问题产生矛盾。
具体来说,如果这个兴趣小组中出现两个人性格值的乘积开三次方根是一个正整数,就认为他们两个性格不合。
比如一个性格值为2的同学和一个性格值为4的同学就是性格不合的,因为2 * 4=8,而一个性格值为2的同学和一个性格值为8的同学性格相合,可以出现在同一个兴趣小组中,因为2*8=16,16开三次方根不是一个正整数。
请你告诉牛牛,他们班的同学组成的最大兴趣小组的人数是多少。
输入描述:
第一行输入一个正整数n表示牛牛所在的班级中的人数。
接下来输入一行n个正整数a[i]表示每个人的性格值。
输出描述:
输出一行一个正整数,表示最大兴趣小组的人数。
示例1
输入
4
4 2 16 27
输出
3
思路:重要的是对数据的初始化(去除立方因子),使用map标记和查找,代码中有详细的注释,我就不再啰嗦了
using namespace std; #define ll long long ll vi[1000],pri[1000]; int cn=0; void init()//用素数筛得方式得到2~1000所有素数的立方 { for(int i=2;i<1000;++i) if(!vi[i]) { pri[++cn]=i*i*i; for(int j=i*i;j<1000;j+=i) vi[j]=1; } } ll fun(ll x)//清除x中所有的立方因子 { for(int i=1;i<=cn;++i) { if(x<pri[i])break; while(x%pri[i]==0)x/=pri[i]; } return x; } int main() { init(); int n,ans=0; ll x; scanf("%d",&n); map<ll,int>mp; for(int i=1;i<=n;++i) scanf("%lld",&x),x=fun(x),mp[x]++; map<ll,int>::iterator it,ite; for(it=mp.begin();it!=mp.end();++it) { if(it->first==1) { ++ans;continue; } ite=mp.find(fun((it->first)*(it->first)));//寻找是否有(*it)*(*it) if(ite==mp.end()) { ans+=it->second; it->second=0;//注意只能只能置0,不能erase,会影响ans+=max(it->second,ite->second); continue; } ans+=max(it->second,ite->second);//题目要求人数最多,相当于去除与小组中许多人都能开立方的人 it->second=0;ite->second=0; } printf("%d\n",ans); }