CodeForces 347 B. Fixed Points

感觉这个题不难，但是对思维要求比较高．．．

给出n个数找出所有满足a[i]==i的数的个数

最后允许交换两个数的位置，找出a[i]==i的个数

但只允许交换一次

交换时就是找a[i]==j && a[j]==i的情况

可以化为a[a[i]]==i

另外要把所有a数组读入完后才能处理．．．

代码如下：

#include <cstdio>
#include <iostream>
#include <algorithm>
#define MAXN 10010
#define ll long long
using namespace std;

int a[MAXN];

int main(void) {
    int n;
    while(cin >> n) {
        int count = 0;
        int flag = 0;
        for(int i=0; i<n; ++i) {
            cin >> a[i];
            if(a[i] == i)
                count++;
        }
        for(int i=0; i<n; ++i) {
            for(int j=i+1; j<n; ++j) {
                if(a[i]==j && a[j]==i) {
                    flag = 1;
                    break;
                }
            }
            if(flag) break;
        }
        count++;
        if(flag) count++;
        cout << count << endl;
    }
}