B. Fixed Points

time limit per test
2 seconds
memory limit per test
256 megabytes
input
standard input
output
standard output
 

A permutation of length n is an integer sequence such that each integer from 0 to (n - 1) appears exactly once in it. For example, sequence [0, 2, 1] is a permutation of length 3 while both [0, 2, 2] and [1, 2, 3] are not.

A fixed point of a function is a point that is mapped to itself by the function. A permutation can be regarded as a bijective function. We'll get a definition of a fixed point in a permutation. An integer i is a fixed point of permutation a0, a1, ..., an - 1 if and only if ai = i. For example, permutation [0, 2, 1] has 1 fixed point and permutation [0, 1, 2] has 3 fixed points.

You are given permutation a. You are allowed to swap two elements of the permutation at most once. Your task is to maximize the number of fixed points in the resulting permutation. Note that you are allowed to make at most one swap operation.

Input

The first line contains a single integer n (1 ≤ n ≤ 105). The second line contains n integers a0, a1, ..., an - 1 — the given permutation.

Output

Print a single integer — the maximum possible number of fixed points in the permutation after at most one swap operation.

Examples
input
5
0 1 3 4 2
output
3

水题 求n个0~n-1的数字他们所在的位置等于自身数值的个数。

分别记录位置和数值,找的两个交换成立则+2,否则有一个可换+1,具体看代码。

附AC代码:

#include<bits/stdc++.h>
using namespace std;

int a[100010],b[100010];

int main(){
    memset(a,-1,sizeof(a));
    memset(b,-1,sizeof(b));
    int n;
    int flag1=0;
    cin>>n;
    int ans=0;
    for(int i=0;i<n;i++){
        cin>>a[i];
        if(a[i]!=i){
            b[i]=a[i];
            flag1=1;
        }
        else{
            ans++;
        }
    }
    int flag=0;
    for(int i=0;i<n;i++){
        if(a[i]!=i){
            if(a[b[i]]==i){
                flag=1;
                break;
            }
        }
    }
    if(flag==1){
        ans+=2;
        cout<<ans<<endl;
    }
    else{
        if(flag1==0){
            cout<<ans<<endl;
        }
        else{
        ans++;
        cout<<ans<<endl;
        }
        
    }
    return 0;
}
原文地址:https://www.cnblogs.com/Kiven5197/p/5908611.html