2023/1/4习题日志 分糖果

时间：1s 空间：256M

题目描述：

一个班里有n个男生站成一圈，每两个男生之间站着一个女生，共n个女生，男女生分别按顺时针编号依次为1号，2号，以此类推。i号女生站在i号和i+1号男生中间（n+1号男生即1号男生）。

老师依次给男生们发糖，每个男生会收到不少于0的偶数颗糖且会把全部糖果等量分给左右两边的女生。最后第i名女生总共收到了ai颗糖。请计算老师分别给每个男生发了多少糖。

保证n是奇数，可以证明答案唯一。

输入格式：

第一行，包含一个正整数 n，表示男、女生人数。 第二行，包含n个整数a1,a2,...,an，表示每个女生收到的糖果数。

输出格式：

输出n个整数，表示老师分别给每个男生发的糖果数。

方程逆推可得到公式    
        b[i] = a[i - 1] * 2 - b[i - 1];

利用公式对本题进行求解即可AC。

下附AC代码

#include<bits/stdc++.h>
using namespace std;
vector<pair<int, int>> v;
int s[100005];
int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int n, k;
    string ss;
    cin >> n >> k >> ss;
    for (int i = 1; i <= n;++i){
        s[i] = ss[i - 1] - '0';
    }
    s[n + 1] = 2;
    int c = s[1], sum = 1;
    for (int i = 2; i <= n + 1;i++){
        if(c!=s[i]){
            v.push_back({c, sum});
            sum = 1;
            c = s[i];
        }else{
            sum++;
        }
    }
    for (int i = 1; i < v.size();i++){
        v[i].second += v[i - 1].second;
    }
    n = v.size();
    sum = 0;
    int ans = 0, l = 0;
    for (int r = 0; r < n;++r){
        if(v[r].first==0){
            sum++;
        }
        while(sum>k){
            if(v[l].first==0){
                sum--;
            }
            l++;
        }
        if(l-1<0){
            ans = max(ans, v[r].second);
        }
        else{
            ans = max(ans, v[r].second - v[l - 1].second);
        }
    }
    cout << ans;
}