2015-2016 ACM ICPC Baltic Selection Contest E. Permutation Polygon

---

题目描述：

http://codeforces.com/gym/100796/problem/E

题解：

其实很裸的一道题.
对于i和p[i],我们考虑在i和p[i]之间,但是值却在它之外的,那么和他相交.
同时,为了便于计算,我们也可以算在1~i和p[i]~n的值在i到p[i]的个数,
这个很简单,我们分成两段,左边推一次,右边推一次,用树状数组.然后结束
但是注意:要将二元组i和p[i]同时放在点上,这样才对.但是我们计算的时候只用一次该用的. 这点是重点

重点：

1.扫两次分开算
2.注意二元组同时更新(相当于放在点上)

代码：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>

using namespace std;

typedef long long ll;

const int N = 1e5 + 100;
int C1[N], C2[N];
int n, p[N];
void update1(int x, int val) {
    while (x <= n) {
        C1[x] += val;
        x += x & -x;
    }
}
void update2(int x, int val) {
    while (x <= n) {
        C2[x] += val;
        x += x & -x;
    }
}

int get_sum1(int x) {
    int ret = 0;
    while (x)  {
        ret += C1[x];
        x -= x & -x;
    }
    return ret;
}

int get_sum2(int x) {
    int ret = 0;
    while (x) {
        ret += C2[x];
        x -= x & -x;
    }
    return ret;
}

int query1(int l, int r) {
    if(r < l)
        return 0;
    return get_sum1(r) - get_sum1(l - 1);
}

int query2(int l, int r) {
    if(r < l)
        return 0;
    return get_sum2(r) - get_sum2(l - 1);
}
struct node {
    int a, b;
    node(int _a = 0, int _b = 0) {
        a = _a;
        b = _b;
    }
};
vector<node> t[N];
void solve() {
    ll ans = 0;
    for(int i = 1; i <= n; i++) 
        t[i].clear();
    for(int i = 1; i <= n; i++) {
        int a = i, b = p[i];
        if(a > b)
            swap(a, b);
        t[a].push_back(node(a, b));
        t[b].push_back(node(a, b));
    }
    memset(C1, 0, sizeof(C1));
    for(int i = 1; i <= n; i++) {
        for(int j = 0; j < t[i].size(); j++) {
             int a = t[i][j].a, b = t[i][j].b;
            if(a == i) {
                ans = ans + query1(a + 1, b - 1);
            }
        }
        for(int j = 0;j < t[i].size(); j++) {
            int a = t[i][j].a, b = t[i][j].b;
            update1(a, 1);
            update1(b, 1);
        }
    }
    /* for(int i = 1; i <= n; i++) {
        t[i].clear();
    }
    for(int i = 1; i <= n; i++) {
        int a = i, b = p[i];
        if(a > b)
            swap(a, b);
        t[b].push_back(node(a, b));
        }*/
    // printf("11    %I64d\n", ans);
    memset(C1, 0, sizeof(C1));
    for(int i = n; i >= 1; i--) {
        for(int j = 0; j < t[i].size(); j++) {
            int a = t[i][j].a, b = t[i][j].b;
            if(b == i)
                ans = ans + query1(a + 1, b - 1);
        }
        for(int j = 0; j < t[i].size(); j++) {
            int a = t[i][j].a, b = t[i][j].b;
            update1(a, 1);
            update1(b, 1);
        }
    }
    ans /= 2;
    printf("%I64d\n", ans);
}

int main() {
   // freopen("e.txt", "r", stdin);
    while(scanf("%d", &n) != EOF) {
        for(int i = 1; i <= n; i++) {
            scanf("%d", &p[i]);
        }
        solve();
    }
    return 0;
}