AtCoder Regular Contest 170（A~B）

A - Yet Another AB Problem

给你两个字符串S和T，你可以对S执行操作，选择两个字符，将前面的改为A，后面的改为B，最少操作几次可以把S改成T。如果改不成就输出-1。

从左往右一个一个改过去，分类讨论，如果是要把A改成B。

S：A->B

T：B

那么T中该位置前面一定要有一个A，否则无法修改。

如果要把B改成A。

S：B->A

T：A

那么T中该位置后面一定要有一个B，否则无法修改。

其中可以本次修改可以更优，即S中后面有一个A，对应T后面的B（一次修改，完成两次对应）

#include <bits/stdc++.h>
//#define int long long
#define per(i,j,k) for(int (i)=(j);(i)<=(k);++(i))
#define rep(i,j,k) for(int (i)=(j);(i)>=(k);--(i))
#define fr first
#define se second
#define endl '\n'
using namespace std;
const int N=2e5+5;

int n,ans,a;
string s,t;

queue<int>q;
int b[N];

void solve(){
    cin>>n>>s>>t;
    per(i,0,n-1){
        if(t[i]=='B' and s[i]=='A')q.push(i);
    }
    rep(i,n-1,0){
        if(t[i]=='B')b[i]++;
        if(i>=1)b[i-1]=b[i];
    }
    per(i,0,n-1){
        if(t[i]=='A')a++;
        if(s[i]!=t[i]){
            if(s[i]=='A'){//需要改成B,前面至少有一个A
                if(!a){
                    cout<<-1<<endl;
                    return ;
                }
                ans++;
            }else{//需要改成A,后面至少有一个B
                if(!b[i+1]){
                    cout<<-1<<endl;
                    return;
                }
                ans++;
                while(!q.empty() and q.front()<i)q.pop();
                if(!q.empty()){
                    s[q.front()]='B';
                    q.pop();
                }
            }
        }
    }

    cout<<ans<<endl;
}

signed main(){
    ios::sync_with_stdio(false),cin.tie(nullptr);
    int t=1;
    while(t--)solve();
    return 0;
}

补题：B - Arithmetic Progression Subsequence

给你1e5个数，每个数是1~10。对于l和r区间，如果区间中有三个数（不管顺序）a[j]，a[k]，a[l]，满足1 2 3或3 2 1（差为1） 或者1 5 9，9 5 1（差为4）这种差相等的，说明l和r是一个好区间，号区间有几个。

思路1：考虑差值最大只有可能是4，对于一个数a[i]，只需要枚举所有差值（sub：1~4），a[i]，a[i]+sub，a[i]+2*sub（注意也可以是减法），如果在a[i]之后存在这样的值，那么第三个数的下标及其以后都是好区间。所以只需要想一个算法维护每个数后面的1~10第一次出现的位置。

思路2：找到一个好区间之后就可以无限左右扩展，还需要去判断内部是否有好空间，如果内部有一个好空间，那么外部也都是好空间，所以不应该是从每一个数开始枚举，应该枚举好空间序列长度，从3开始往上扩展。

正确思路：正难则反，只会出现1~10的数，尝试构建无解的情况，从差为0开始往上，如果差重复了就必然有解，如1 1 2 4 8，几乎就没别的数可以填了，就会开始重复了。

AC代码

#include <bits/stdc++.h>
#define int long long
#define per(i,j,k) for(int (i)=(j);(i)<=(k);++(i))
#define rep(i,j,k) for(int (i)=(j);(i)>=(k);--(i))
#define fr first
#define se second
#define endl '\n'
using namespace std;
const int N=1e5+5;

int n,a[N],ans;

bool check(int l,int r){//约1~100次查询
    per(i,l,r){
        per(j,i+1,r){
            per(k,j+1,r){
                if(a[j]-a[i]==a[k]-a[j])return true;
            }
        }
    }
    return false;
}

void solve(){
    cin>>n;
    per(i,1,n)cin>>a[i];
    per(i,1,n){
        per(j,i+1,n){//i和j差到10以内就必然有解,复杂度是带系数的O(n),check比较烂总体约1e9
            if(check(i,j)){
                ans+=n-j+1;
                break;
            }
        }
    }
    cout<<ans<<endl;
}

signed main(){
    ios::sync_with_stdio(false),cin.tie(nullptr);
    int t=1;
    while(t--)solve();
    return 0;
}

顺带一提不开long long见祖宗。