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
", ans);
17     return 0;
18 }

View Code