Codeforces Round 842 (Div. 2) D. Lucky Permutation

__night_

于 2024-03-27 22:42:06 发布

阅读量143

点赞数 1

分类专栏： codeforces 文章标签：算法图论

本文链接：https://blog.csdn.net/way_back__/article/details/137092095

版权

codeforces 专栏收录该内容

95 篇文章

订阅专栏

这段C++代码解决了一个问题，给定一个整数数组表示的链表，计算在每个元素值的连通分量中，不包含循环的部分。程序首先找出所有连通分量，然后判断是否存在环，输出连通分量的数量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目

思路：

#include <bits/stdc++.h>
using namespace std;
#define int long long
#define pb push_back
#define fi first
#define se second
#define lson p << 1
#define rson p << 1 | 1
const int maxn = 1e6 + 5, inf = 1e18, maxm = 4e4 + 5;
const int mod = 1e9 + 7;
// const int mod = 998244353;
const int N = 1e5;
// int a[505][5005];
// bool vis[505][505];
// char s[505][505];
int a[maxn], b[maxn];
int vis[maxn];
string s;
int n, m;

struct Node{
    int val, id;
    bool operator<(const Node &u)const{
        return val < u.val;
    }
}c[maxn];

int ans[maxn], pre[maxn];
int p[maxn], q[maxn];

void solve(){
    int res = 0;
    // int q, k;
    int x;
    cin >> n;
    for(int i = 1; i <= n; i++){
        vis[i] = 0;//vis下标
    }
    for(int i = 1; i <= n; i++){
        cin >> a[i];
    }
    int cnt = n;
    for(int i = 1; i <= n; i++){
        if(!vis[i]){
            int j = i;
            cnt--;
            while(!vis[j]){
                vis[j] = i;
                j = a[j];
            }
        }
    }
    bool flag = 0;
    for(int i = 2; i <= n; i++){
        if(vis[i - 1] == vis[i]){
            flag = 1;
        }
    }
    if(flag){
        res = cnt - 1;
    }
    else{
        res = cnt + 1;
    }
    cout << res << '\n';
}
    
signed main(){
    ios::sync_with_stdio(0);
    cin.tie(0);

    int T = 1;
    cin >> T;
    while (T--)
    {
        solve();
    }
    return 0;
}