ZZULI 1902: 985的因子對難題

1902: 985的因子對難題
Time Limit: 2 Sec  Memory Limit: 128 MB
Submit: 156  Solved: 45

SubmitStatusWeb BoardDescription
985有n個正整數，他想知道存在多少個不同的因子對(a[i], a[j])使得
1 <= i, j <= n && i != j && a[j] % a[i] == 0，其中i和j是元素的下標。
特別地，他認爲（a[i]，a[j]）與（a[j]，a[i]）是一樣的因子對。

Input
第一行輸入一個整數t，代表有t組測試數據。
每組數據佔兩行，第一行輸入一個n代表元素個數，下面一行輸入n個整數a[]。
注：1 <= t <= 30，1 <= n <= 1e5，1 <= a[] <= 1e6。

Output
一個整數代表最後的答案。


Sample Input
2
5
1 2 3 4 5
5
2 2 2 2 2
Sample Output
5
10
HINT

Source
hpu

SubmitStatusWeb Board

很簡單記錄下每個a[i]出現的次數然後對a[i]排序去重
最後類似於篩素數的方法篩選出來就好了

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#include <queue>
#include <stack>

using namespace std;
const int MAXN=1e5+10;
int a[MAXN];
int num[MAXN*10];
int main(){
    int T; scanf("%d",&T);
    while(T--){
        int n; scanf("%d" ,&n);
        memset(num,0,sizeof(num));
        for(int i = 1; i  <= n; i++ ){
            scanf("%d",&a[i]);
            num[a[i]]++;
        }
        sort(a + 1, a + n + 1);
        int k=unique(a+1,a+1+n)-a;//去重函數
        int sum=0;
        k--;
//        printf("%d~~~~\n",k);
        for(int i=1;i<=k;i++){
            sum=sum+(num[a[i]]-1)*(num[a[i]])/2;
            for(int j=a[i]*2;j<=a[k];j+=a[i]){
                    sum+=num[a[i]]*num[j];
            }
        }
        printf("%d\n",sum);
    }
    return 0;
}