【题解】codeforces 1746B Rebellion

题意：给定一个只包含0和1的数组a，可以对a进行以下操作：

选定两个下标不同的元素a_i和a_j，将a_i加到a_j上，再从数组中删除a_i。

问最少操作多少次，可以让数组a变成单调非下降子序列（即a_i<=a_j, 1 <= i < j <= n)。

 

思路：双指针扫一遍数组中的元素，i = 0, j = n - 1,（下标均从0开始） 其中指针 i 任务是找到一个1，指针 j 的任务找到一个0，然后把a_i加到a_j上，然后删除a_i，操作计数器加1。重复此过程直到两个指针相遇。以样例 1 0 0 1 1为例，首先i = 0, j = 2，然后 数组元素变成了 0 1 1 1，已经是一个单调非下降的子序列了，总共操作次数为1。

 

附上代码如下。

 

#include <iostream>
#include <stack>
using namespace std;

const int N = 100010;int n;
int a[N];

int main() {
    ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
    int t;
    cin >> t;
    while(t--) {
        cin >> n;
        bool flag = true;
        
        for(int i = 0; i < n; i++) {
            cin >> a[i];
            if(i && a[i] < a[i - 1]) flag = false;
        }
        if(flag) cout << 0 << endl;
        else {
            int i = 0, j = n - 1, cnt = 0;
            while(i < j) {
                while(i < n && !a[i]) i++;
                while(j >= 0 && a[j]) j--;
                
                if(i >= j) break;
                cnt++;i++;j--;
            }
            cout << cnt << endl;
        }
    }
    return 0;
}