codeforces Edu 142 D. Fixed Prefix Permutations 【思维、字典树求LCP】

最新推荐文章于 2024-07-06 15:43:38 发布

吵闹的人群保持笑容多冷静

最新推荐文章于 2024-07-06 15:43:38 发布

阅读量949

点赞数 15

分类专栏： codeforces 练习文章标签：算法

本文链接：https://blog.csdn.net/m0_73500785/article/details/137339882

版权

codeforces 练习专栏收录该内容

49 篇文章 0 订阅

订阅专栏

文章讲述了如何利用Trie数据结构解决给定排列问题，通过计算排列的逆序并查找最长公共前缀(LCP)，找到每对排列的最大公共部分价值。时间复杂度为O(nm)。

摘要由CSDN通过智能技术生成

D. Fixed Prefix Permutations

题意

给定 $n$ 个长度为 $m$ 的排列 $a_1,a_2,...a_n$

定义一个排列 $p$ 的价值为 最大顺序长度 $k$ ： $p_1 = 1,p_2 = 2, p_3 = 3, ... p_k = k$ ，如果 $p_1 \neq 1$ ，那么 $p$ 的价值为 $0$

定义两个排列的乘积为： $\cdot q = r，其中 r_j = q_{p_j}$

现在对于每个 $\in [1, n]$ ，求出最大的 $a_i \cdot a_j$ （注意 $j$ 可以等于 $i$ ）

思路

对于当前的 $a_i$ ，我们需要找到一个 $j$ ，使得 $a_i \cdot a_j = (1,2,3,4,....k, r_{k + 1},....r_m)$ 中的 $k$ 最大
如果 $k$ 刚好等于 $m$ 的话， $a_i \cdot a_j = (1, 2, 3, 4,...., m)$

此时我们转换一下： $a_i = (1,2,3,4,..., m) \cdot a_j ^ {-1}$

不难发现，对于一个顺序度为 $m$ 的排列 $(1, 2, 3, 4, ..., m)$ ，它乘上 $a_j ^ {-1}$ 不会改变 $a_j ^ {-1}$ 的内容

也就是： $a_i = a_j ^ {-1}$ ；
更一般地，当 $k < m$ 时， $a_i$ 的前 $k$ 位与 $a_j ^ {-1}$ 的前 $k$ 位是一样的！

那么我们只需要求出每个 $a_i$ 的逆，然后再与当前的 $a_i$ 比较，最长公共前缀就是当前 $a_i$ 的答案

如何求逆？由 $\cdot p ^ {-1} = [1,2,3,4,..., m]$ 得： ${-1}_{p_j} = j$ ，也就是以 $p$ 为下标映射， $p_j$ 指向的位置一定得为 $j$

那么我们只需要将每个逆存在 $T r i e$ 上，对于当前的 $a_i$ 在 $T r i e$ 上跑 $L CP$ （最大公共前缀）即可

时间复杂度： $O (nm)$

#include<bits/stdc++.h>
#define fore(i,l,r)	for(int i=(int)(l);i<(int)(r);++i)
#define fi first
#define se second
#define endl '\n'
#define ull unsigned long long
#define ALL(v) v.begin(), v.end()
#define Debug(x, ed) std::cerr << #x << " = " << x << ed;

const int INF=0x3f3f3f3f;
const long long INFLL=1e18;

typedef long long ll;

const int N = 500050;

struct node{
    int son[11];
}tree[N];

int cnt = 0;

int main(){
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    std::cout.tie(nullptr);
    int t;
    std::cin >> t;
    while(t--){
        int n, m;
        std::cin >> n >> m;
        std::vector<int> ans(n + 1, 0);
        std::vector<std::vector<int>> a(n + 1, std::vector<int>(m + 1, 0));

        fore(i, 1, n + 1)
            fore(j, 1, m + 1)
                std::cin >> a[i][j];
        
        fore(i, 1, n + 1){
            std::vector<int> b(m + 1);
            fore(j, 1, m + 1) b[a[i][j]] = j;
            int now = 0;
            fore(j, 1, m + 1){ //Trie 插入 排列的逆
                int val =  b[j];
                if(!tree[now].son[val]) tree[now].son[val] = ++cnt;
                now = tree[now].son[val];
            }
        }

        fore(i, 1, n + 1){
            int now = 0;
            fore(j, 1, m + 1){ //Trie 查询最长前缀
                int val = a[i][j];
                if(!tree[now].son[val]) break;
                ++ans[i];
                now = tree[now].son[val];
            }
        }

        fore(i, 1, n + 1) std::cout << ans[i] << " \n"[i == n];

        fore(i, 0, cnt + 1) //清空 Trie
            fore(j, 1, m + 1)
                tree[i].son[j] = 0;
        cnt = 0;
    }
    return 0;
}