BZOJ1607 [Usaco2008 Dec]Patting Heads

Posted 2020-09-08 nhc2014

题目链接：http://www.lydsy.com/JudgeOnline/problem.php?id=1607

解：这么简单的题目我一开始竟然完全没思路。。。

只要一输入一头牛的数字，我们就把是它的倍数的位置加上1，这样询问的时候，就是O(1)的复杂度了。

因为题目保证了所有牛的数字不同，所以时间复杂度还是蛮优的（吧。。。），1000000的int数组也开得下。

程序：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
using namespace std;
int num[2000000],maxx,n,ans[2000000],a[2000000];
int main()
{
  scanf("%d",&n);
  for (int i=1;i<=n;i++) 
  {
    scanf("%d",&a[i]);
    num[a[i]]++;
    maxx=max(a[i],maxx);
  }
  for (int i=1;i<=maxx;i++)
   for (int j=1;j*i<=maxx;j++)
    ans[j*i]+=num[i];
  for (int i=1;i<=n;i++) printf("%d\n",ans[a[i]]-1);
  return 0;
}