Codeforces Div.2 212 C Insertion Sort (DP)

题目链接：https://codeforces.com/problemset/problem/362/C

题目大意：

        题目给你一个长为n的0~n-1的排列，并以此作插入排序，现在允许你先交换两个数的位置，使得插入排序中交换值的操作最少，问你有多少种交换方法。

 

解题思路：

        数组下标从1开始。 令 dp[i][j] 表示a[1] ~ a[i] 中小于j 的数字个数。显然有dp[i][j] = dp[i-1][j] + (a[i] < j ? 1 : 0). 总交换次数为sigma(i - dp[i][a[i]] - 1),i属于[2, n]。从2开始是因为如果只有一个数，交换次数必为0。

        令i < j, 且a[i] > a[j], 设p为a[i+1] ~ a[j-1]中小于a[i]的个数，q为a[i+1] ~ a[j-1]中小于a[i]的个数，若这两个数交换位置，贡献增量可以分为三部分：

    1、a[i+1]~a[j-1]贡献增量为q - p

    2、a[i] 在 j 位置上贡献增量为（j - i + 1） - p (补上区间跨度，减去a[i+1]~a[j-1]中小于a[i]的个数，满足计算规律)

    3、a[j] 在 i 位置上贡献增量为 -（j - i + 1）+ q - 1 （减1是因为a[i] > a[j], 现在位置交换了）

    其中p = dp[j-1][a[i]] - dp[i][a[i]];  q = dp[j-1][a[j]] - dp[i][a[j]]；合起来贡献增量为2*(q-p) - 1.

代码如下：

# include <bits/stdc++.h>

using namespace std;

const int maxn = 5e3 + 5;
int a[maxn], dp[maxn][maxn];
int n, ans;

int main(){
    std::ios::sync_with_stdio(false);
    while(cin >> n){
        for(int i = 1; i <= n; ++i)
            cin >> a[i];

        /*从0开始的话要单独初始化值
        for(int i = 0; i < n; ++i)
            dp[0][i] = a[0] < i ? 1 : 0;*/

        int all = 0;
        for(int i = 1; i <= n; ++i){
            for(int j = 0; j < n; ++j)
                dp[i][j] = dp[i-1][j] + (a[i] < j ? 1 : 0);
            if(i > 1)   all += i - dp[i][a[i]] - 1;  //下标从1开始所以要-1，只有一个的话就不用换，所以从2开始累加
        }

        int minn = all;
        for(int i = 1; i <= n; ++i){
            for(int j = i+1; j <= n; ++j){
                if(a[i] > a[j]){
                    int p = dp[j-1][a[i]] - dp[i][a[i]];
                    int q = dp[j-1][a[j]] - dp[i][a[j]];
                    int tmp = all + 2 * (q - p) - 1;
                    if(tmp < minn){
                        minn = tmp;
                        ans = 1;
                    }
                    else if(tmp == minn) ans++;
                }
            }
        }
        cout << minn << ' ' << ans << endl;

        memset(dp, 0, sizeof(dp));
        memset(a, 0, sizeof(a));
    }

    return 0;
}