开关(区间数) (智算之道初赛高校组第一场)

题目列表

共 10 个测试点 每个测试点 10 分
每个测试点限时 1 秒 运行内存上限 512MB

小 B 面前有 n 个开关，开始时第 i个开关的状态是 ai，其中 ai = 1 表示第 i个开关是开的，ai = 0表示第 i 个开关是关的。
现在 小 B 获得了一种魔法，他可以进行若干次操作，每次操作可以选择一个数 x，然后把 x 号开关及其之前的所有开关状态反转（开变关，关变开），请问小 B 最少需要多少次操作才能使所有开关都变为关的状态。

输入格式
第一行一个整数表示 n
第二行为一个长度为 n的 01 字符串，即每一位只会是 0 或者 1
输出格式
输出一个整数，表示最少需要多少次操作才能使所有开关都变为关的状态

数据规模与约定
对于 30% 的数据，1 <= a <= 20;
对于 60%的数据，1 <= n <= 2000
对于100% 的数据，1 ≤ n ≤ 200000
样例输入
4
1001
样例输出
3

思路： 不难发现，只能从后往前处理，即从后往前的第一个1开始算，有多少个区间就需要多少次(也包括0区间)。

代码实现：

#include
#define endl '\n'
#define null NULL
#define ll long long
#define int long long
#define pii pair
#define lowbit(x) (x &(-x))
#define ls(x) x<<1
#define rs(x) (x<<1+1)
#define me(ar) memset(ar, 0, sizeof ar)
#define mem(ar,num) memset(ar, num, sizeof ar)
#define rp(i, n) for(int i = 0, i < n; i ++)
#define rep(i, a, n) for(int i = a; i <= n; i ++)
#define pre(i, n, a) for(int i = n; i >= a; i --)
#define IOS ios::sync_with_stdio(0); cin.tie(0);cout.tie(0);
const int way[4][2] = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
using namespace std;
const int  inf = 0x7fffffff;
const double PI = acos(-1.0);
const double eps = 1e-6;
const ll   mod = 1e9 + 7;
const int  N = 2e5 + 5;

int n;
string s;

signed main()
{
    IOS;
    
    cin >> n >> s;
    int pos = -1;
    for(int i = s.size() - 1; ~i; i --){
        if(s[i] != '0'){
            pos = i; 
            break;
        }
    }
    //如果没有1区间
    if(pos == -1){
        cout << 0 << endl;
        return 0;
    }
    int ans = 1;
    for(int i = pos - 1; ~i; i --){
        if(s[i] != s[i + 1] && i + 1 < s.size())
            ans ++;
    }
    cout << ans << endl;

    return 0;
}