HDU 2040 亲和数

问题描述

古希腊数学家毕达哥拉斯在自然数研究中发现，220 的所有真约数(即不是自身的约数)之和 为：1+2+4+5+10+11+20+22+44+55+110＝284。 而284 的所有真约数为 1、2、4、71、 142，加起来恰好为 220。人们对这样的数感到很惊奇， 并称之为亲和数。一般地讲，如果两个数中任何一个数都是另一个数的真约数之和，则这两 个数就是亲和数。 你的任务就编写一个程序，判断给定的两个数是否是亲和数

输入

输入数据第一行包含一个数 M，接下有M 行，每行一个实例,包含两个整数 A,B； 其中 0 <= A,B <= 600000 ;

输出量

对于每个测试实例，如果A 和 B 是亲和数的话输出 YES，否则输出 NO。

样本输入

2
220 284 
100 200

样本输出

YES
NO

代码

#include <iostream>
#include <cmath>
using namespace std;
int sum(int a){
    int res=1;
    for (int i = 2; i <=sqrt(a) ; ++i) {
            if(a%i==0){
                if(i!= sqrt(a)){

                    res=res+i+a/i;
                }
                else{
                    res=res+i;
                }
             }
    }
    return res;
}
int main(){
    int n,a,b;
    cin>>n;
    while(n--){
        cin>>a>>b;
        if(sum(a)==b && sum(b)==a){
            cout<<"YES"<<endl;
        }
        else {
            cout<<"NO"<<endl;
        }
    }
    return 0;
}