k倍多重正整数集合

import java.util.*;

// k=1：不同数字个数
// k>1：有K倍数关系的数, 一起排序, 不能连续取 → 打家劫舍DP 🔥

public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int n  = in.nextInt(), k = in.nextInt();
        TreeMap<Long, Integer> map = new TreeMap<>();
        for (int i = 0; i < n; i++) {
            map.merge(in.nextLong(), 1, Integer::sum);
        }

        if (k == 1) {
            System.out.println(map.size());
            return;
        }

        int res = 0;
        for (long x : map.keySet()) {
            int cur = 0, pre1 = 0, pre2 = 0;
            // x  kx  kkx ...
            while (map.containsKey(x) && map.get(x) > 0 && x <= map.lastKey()) {
                int cnt = map.get(x);
                map.put(x, 0);
                cur = Math.max(cnt + pre2, pre1); // 选&nbs***bsp;不选
                pre2 = pre1;
                pre1 = cur;
                x *= k;
            }
            res += cur;
        }
        System.out.println(res);
    }
}