康托展开和逆康托展开

康托展开

cpp 复制代码
//康托展开
constexpr int M = 998244353;
template<class T>
struct Fenwick {
    //比如说数组长度为5
    //-----------------a[3]
    //------a[1]
    //--a[0]    --a[3]     --a[4]
    //    1    2    3    4    5
    //add(x)修改的是x+1
    //sum(x)查询的是x
    int n;
    std::vector<T>a;
    Fenwick(int n_=0) {
        init(n_);
    }
    void init(int n_) {
        n = n_;
        a.assign(n, T{});
    }
    void add(int x, const T& v) {

        for (int i = x + 1; i <= n; i += i & -i) {
            a[i - 1] = a[i - 1] + v;
        }
    }
    T sum(int x) {
        T ans{};
        
        for (int i = x; i > 0; i -= i & -i) {
            ans = ans + a[i - 1];
        }
        return ans;
    }
    T rangeSum(int l, int r) {
        return sum(r) - sum(l);
    }
};

struct cantor {
    int n;
    std::vector<ll>fac;
    Fenwick<ll> f;
    cantor(int n_) {
        init(n_);
    }
    void init(int n_) {
        n = n_;
        fac.assign(n, 0);
        f.init(n);
        fac[0] = 1;
        for (int i = 1; i < n; i++) {
            fac[i] = i*fac[i - 1] % M;
        }
        for (int i = 0; i < n; i++) {
            f.add(i, 1);
        }
    }
    ll query(int x, std::vector<ll>a) {
        //树状数组求逆序对不就是
        ll sum = 0;
        for (int i = 0; i < x; i++) {
            sum = (sum + fac[n - i - 1] * f.sum(a[i]-1) % M) % M;
            f.add(a[i] - 1, -1);
        }
        return (sum+1) % M;
    }
};

void solve() {
    int n;
    std::cin >> n;
    cantor c(n);
    std::vector<ll>a(n);
    for (int i = 0; i < n; i++) {
        std::cin >> a[i];

    }
    std::cout << c.query(n, a) << "\n";
}

逆康托展开

cpp 复制代码
std::string invcantor(int n, ll k) {
    // 计算阶乘
    std::vector<ll> fac(n, 1);
    for (int i = 1; i < n; i++) {
        fac[i] = fac[i - 1] * i;
    }

    // 初始化未使用的数字集合
    std::vector<int> vis(n);
    std::iota(vis.begin(), vis.end(), 1);

    // 调整 k 为从 0 开始的索引
    k--;

    std::string ans;
    for (int i = 0; i < n; i++) {
        ll fact = fac[n - i - 1];
        int pos = k / fact; // 当前位数字的索引
        ans.push_back(vis[pos]+'0');
        vis.erase(vis.begin() + pos); // 从集合中移除已选数字
        k %= fact; // 更新 k
    }

    return ans;
}
相关推荐
YY_TJJ1 天前
算法题——贪心算法
算法·贪心算法
C++ 老炮儿的技术栈1 天前
include″″与includ<>的区别
c语言·开发语言·c++·算法·visual studio
RainbowC01 天前
GapBuffer高效标记管理算法
android·算法
liu****1 天前
10.queue的模拟实现
开发语言·数据结构·c++·算法
mit6.8241 天前
10.17 枚举中间|图论
算法
shinelord明1 天前
【大数据技术实战】Kafka 认证机制全解析
大数据·数据结构·分布式·架构·kafka
让我们一起加油好吗1 天前
【基础算法】01BFS
数据结构·c++·算法·bfs·01bfs
孤狼灬笑1 天前
机器学习十大经典算法解析与对比
人工智能·算法·机器学习
草莓工作室1 天前
数据结构3:线性表2-顺序存储的线性表
数据结构·windows
1白天的黑夜11 天前
递归-24.两两交换链表中的节点-力扣(LeetCode)
数据结构·c++·leetcode·链表·递归