題目連結

  蠻好的一道題，補題的時候用了O(N) 的演算法給過了，還有一種O(N^2)的演算法應該是指的是樹狀陣列，一會我去嘗試下，這道題，在比賽的時候，我想到的就是上次做過的DNA那道題類似，開一個多維樹狀陣列，然後去查對應的補集，然後可惜的是當時硬是沒想出來BUG在哪，因為題目有些坑點：

• K==0時候，我們取餘就會取不到，所以要特殊的分出來算；
• 還有一件事，當你K==0時候知道分開來算，也小心我們的得到的數位f()也只有1時才是不能利用的，非1還是都可以用的。

  剩下的思維：你讀程式碼吧，就是簡單的找補集，然後找對應關係即可。

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#define lowbit(x) ( x&(-x) )
#define pi 3.141592653589793
#define e 2.718281828459045
#define efs 1e-6
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
const int dir[4] = {2, 3, 5, 7};    //10內4個質數
ll N, K;
map<vector<ll>, int> mp;
vector<ll> vt(4), bu(4);   //開四倍空間 vt = vector<ll> (4);
ll get_real(ll x)
{
    if(x == 0) return 0;
    ll ans = 1;
    while(x)
    {
        ans = ans * (x%10);
        x/=10;
        if(ans == 0) return 0;
    }
    return ans;
}
int main()
{
    scanf("%lld%lld", &N, &K);
    if(K==0)
    {
        ll zero = 0;
        ll tmp;
        for(int i=1; i<=N; i++)
        {
            scanf("%lld", &tmp);
            tmp = get_real(tmp);
            if(tmp == 1) zero++;
        }
        printf("%lld\n", N*(N-1)/2 - zero*(zero-1)/2);
    }
    else
    {
        ll cnt = 0, zero = 0;
        ll tmp = 0;
        for(int i=1; i<=N; i++)
        {
            scanf("%lld", &tmp);
            tmp = get_real(tmp);
            if(tmp)
            {
                for(int i=0; i<4; i++)
                {
                    vt[i] = 0;
                    while(tmp%dir[i] == 0)
                    {
                        tmp/=dir[i];
                        vt[i]++;
                    }
                    vt[i]%=K;
                    bu[i] = (K - vt[i])%K;
                }
                cnt += mp[vt];
                mp[bu]++;
            }
            else zero++;
        }
        if(zero) cnt += zero*(zero-1)/2 + zero*(N-zero);
        cnt = N*(N-1)/2 - cnt;
        printf("%lld\n", cnt);
    }
    return 0;
}
 

一會，我再去嘗試下用下樹狀陣列去做一下這道題，蠻好的一道題，多幾種方法去弄。