codeforces 1792D Fixed Prefix Permutations

最新推荐文章于 2024-11-05 21:58:11 发布

_fairyland

最新推荐文章于 2024-11-05 21:58:11 发布

阅读量176

点赞数 1

分类专栏： # 思维 # codeforces # 二分文章标签：算法 c++

本文链接：https://blog.csdn.net/qq_41418557/article/details/129394512

版权

codeforces 同时被 3 个专栏收录

29 篇文章 0 订阅

订阅专栏

思维

16 篇文章 0 订阅

订阅专栏

二分

5 篇文章 0 订阅

订阅专栏

文章讨论了一种算法问题，即寻找数组a中的元素a_i与数组b中元素的最长公共前缀。通过对b数组进行排序，然后使用二分查找找到与a_i最接近的元素，以O(mnlogn)的时间复杂度求解，降低了原始算法的复杂度。

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

Fixed Prefix Permutations

$i-->p_i-->q_{pi}$
要满足条件 $i=q_{pi}$ ，那么只需 $q_j=i$ 中的 $j=p_i$
为此我们求出每个数字在数组中的pos，存在b中
之后验证只需查看 $a_i$ , $b_j$ 中最长共前缀

对于一个 $a_i$ 暴力查找最长公共前缀的 $b_j$ 需要O(n*m），再循环操作总共需要 $O(m n^2)$
先对 $b$ 整体排序，之后二分查找与 $a_i$ 最相近的 $b_j$ 即可排序预处理总时间复杂度为 $O (mn l o g n)$

#include <bits/stdc++.h>
#define ll long long
using namespace std;
int lcp(vector<int> a,vector<int> b){
    int i=1;
    for (i=1;a[i]==b[i]&&i<a.size();i++){
        ;
    }
    return i-1;
}
void solve(){
    int n,m;
    cin>>n>>m;
    vector<vector<int>> a(n+1,vector<int>(m+1)),b(n+1,vector<int>(m+1));
    for (int i=1;i<=n;i++){
        for (int j=1;j<=m;j++){
            cin>>a[i][j];
            b[i][a[i][j]]=j;
        }
    }
    sort(b.begin()+1,b.end());
    for (int i=1;i<=n;i++){
        int p=lower_bound(b.begin()+1,b.end(),a[i])-b.begin(),ans=0;
        if (p>=2) ans=max(ans,lcp(a[i],b[p-1]));
        if (p<=n) ans=max(ans,lcp(a[i],b[p]));
        cout<<ans<<" ";
    }
    cout<<"\n";
}
int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);

    int t;
    cin>>t;
    while (t--){
        solve();
    }
    return 0;
}