康托展开和逆康托展开

康托展开

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;
}
相关推荐
自信1504130575941 分钟前
初学者小白复盘15之指针(4)
c语言·数据结构·算法
郝学胜-神的一滴41 分钟前
Cesium绘制线:从基础到高级技巧
前端·javascript·程序人生·线性代数·算法·矩阵·图形渲染
东皇太星1 小时前
机器学习概念,算法原理及应用
算法·机器学习·梯度下降法
.格子衫.1 小时前
021数据结构之并查集——算法备赛
数据结构·算法
im_AMBER2 小时前
Leetcode 35
笔记·学习·算法·leetcode
码农多耕地呗2 小时前
力扣101.对称二叉树(java)
算法·leetcode
小龙报2 小时前
《算法通关指南之C++编程篇(5)----- 条件判断与循环(下)》
c语言·开发语言·c++·算法·visualstudio·学习方法·visual studio
郝学胜-神的一滴2 小时前
C++ STL(标准模板库)深度解析:从基础到实践
linux·服务器·开发语言·c++·算法
刚入坑的新人编程2 小时前
算法训练.17
开发语言·数据结构·c++·算法
TinpeaV2 小时前
String[ ] 和 List<String> 的区别
数据结构·list