51nod1305 Pairwise Sum and Divide
题目链接:51nod 1305 Pairwise Sum and Divide
看完题我想都没想就直接暴力做了,AC后突然就反应过来了。。。
Floor( (a+b)/(a*b) )=Floor( (1/b)+(1/a) )
当a=1时:若b=1,则该式等于2,否则该式等于1
当a=b=2时:该式等于1
其余情况都等于0,所以只需要统计1和2的个数就行了。
1 #include<cstdio> 2 #include<algorithm> 3 using namespace std; 4 typedef long long ll; 5 int main(){ 6 int n, i, a; 7 int s1 = 0, s2 = 0, ss = 0; 8 scanf("%d", &n); 9 for(i = 0; i < n; ++i){ 10 scanf("%d", &a); 11 if(a == 1) s1++; 12 else if(a == 2) s2++; 13 else ss++; 14 } 15 ll ans = (ll)s1*(s1-1) + (ll)s1*(s2+ss) + (ll)s2*(s2-1)/2; 16 printf("%lld\n", ans); 17 return 0; 18 }