AtCoder初学者竞赛418

A - I'm a teapot

#include<bits/stdc++.h>
using namespace std;

bool solve()
{
    int n;
    cin >> n;
    string s;
    cin >> s;
    
    if(n < 3)return false;
    return s.substr(n - 3) == "tea";
}

int main()
{
    if(solve())cout << "Yes" << endl;
    else cout << "No" << endl;
    return 0;
}

B - You're a teapot

双for暴力计算每个区间最大值，注意题意要求头尾都是't'这个字符

#include<bits/stdc++.h>
using namespace std;

void solve()
{
    string s;
    cin >> s;
    int n = (int)s.length();
    double maxn = 0.0;
    for(int i = 0;i < n;i ++)
    {
        for(int j = i + 1;j < n;j ++)
        {
            if(s[i] == 't' && s[j] == 't')
            {
                int cnt = 0;
                for(int k = i;k <= j;k ++)
                    if(s[k] == s[i])
                        cnt ++;
                if(cnt < 3)continue;
                maxn = max(maxn,double(cnt - 2) / (j - i - 1));
            }
        }
    }
    printf("%.12lf",maxn);
}

int main()
{
    solve();
    return 0;
}

C - Flush

手动二分查找可以改成

lower_bound(迭代器，迭代器，val)：找到第一个大于等于val的位置

upper_bound(迭代器，迭代器，val)：找到第一个大于val的位置 

#include<bits/stdc++.h>
using namespace std;
using ll = long long;
void solve()
{
    int n,q;
    cin >> n >> q;
    vector<ll>a(n + 1,0);
    vector<ll>S(n + 1,0);
    ll sum = 0;
    for(int i = 1;i <= n;i ++)
    {
        cin >> a[i];
        sum += a[i];
    }
    sort(a.begin() + 1,a.end());
    for(int i = 1;i <= n;i ++)
        S[i] = S[i - 1] + a[i];
    ll maxn = a[n];
//    cout <<"sum:" << sum << endl;
    while(q --)
    {
        ll b;
        cin >> b;
        if(b > maxn)cout << -1 << endl;
        else {
            int l = 1,r = n;
            int t = 0;
            while(l <= r)
            {
                int m = (l + r) / 2;
                if(a[m] >= b){
                    t = m;
                    r = m - 1;
                }else l = m + 1;
            }
//            cout << t << endl;
//            cout << S[t - 1] << " " << (n - t + 1) << " " << (b - 1) << endl;
            cout << S[t - 1] + (n - t + 1) * (b - 1) + 1 << endl;
        }
    }
    
}

int main()
{
    solve();
    return 0;
}

 赛后优化：

#include<bits/stdc++.h>
using namespace std;
using ll = long long;
void solve()
{
    int n,q;
    cin >> n >> q;
    vector<ll>a(n + 1,0);
    vector<ll>S(n + 1,0);
    for(int i = 1;i <= n;i ++)
        cin >> a[i];
    sort(a.begin() + 1,a.end());
    for(int i = 1;i <= n;i ++)
        S[i] = S[i - 1] + a[i];
    while(q --)
    {
        ll b;
        cin >> b;
        if(b > a[n])cout << -1 << endl;
        else {
            int t = distance(a.begin(),lower_bound(a.begin(),a.end(),b));
            t --;
            cout << S[t] + (n - t) * (b - 1) + 1 << endl;
        }
    }
}

int main()
{
    solve();
    return 0;
}

D - XNOR Operation

对于一个子序列，如果其具有偶数个（包括0）个0，则其同或结果必定可以为1

问题转化为区间内有偶数个0的个数，记录pre为前i个数0的个数，odd_cnt,even_cnt分别记录全局的pre为奇数的个数和偶数的个数，即l,r区间要满足题意，则pre_r和pre_(l - 1)的奇偶性质应该一样，又因为所有满足pre为偶数的除了和别的位置配对，也可以到该位置的所有字符视为一个区间（也可以认为是最前面有个pre为偶数（0）的空字符位置）：如1100，下标1-4，如果只配对，只有：1(1，2位置配对)，100（1，4位置配对），00（2，4位置配对），显然缺少：1(0位置和1位置)，11(0位置和2位置)，1100（0位置和4位置）

使用pre ^= 1快速更改奇偶性，pre初始为0

处理偶数，可以初始化偶数计数器为1，或者改n(n - 1)/2为(n + 1)n / 2

#include "bits/stdc++.h"
using namespace std;
int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int n;
    string T;
    cin >> n >> T;
    long long a[2] = {0, 0};
    a[0] = 1;
    int pre = 0;
    for (char ch : T) {
        if (ch == '0') {
            pre ^= 1;
        }
        a[pre]++;
    }
    long long ans = a[0] * (a[0] - 1) / 2 + a[1] * (a[1] - 1) / 2;
    cout << ans << "\n";
    return 0;
}

E - Trapezium

用一个unorderd_map记录斜率相同的边的个数，用另一个map记录斜率相同且边长度相同边的个数，前配对数减后配对数的一半就是梯形数（后者实际在记录平行四边形的数量，由于前面计算梯形时将平四计算了两倍，且后面专门计算平四也计算了两倍，所以要减去一半的后者）

第一个开map超时

#include<bits/stdc++.h>
using namespace std;
using ll = long long;
int n;
void solve()
{
    cin >> n;
    vector<pair<ll,ll>>pt(n);
    unordered_map<double,ll>mp;
    map<pair<double,ll>,int>rect;
    ll ans = 0,rects = 0;
    for(int i = 0;i < n;i ++)
    {
        cin >> pt[i].first >> pt[i].second;
        for(int j = 0;j < i;j ++)
        {
            auto [ix,iy] = pt[i];
            auto [jx,jy] = pt[j];
            double k = 0;
            if(ix - jx == 0)k = 1000000.0;
            else k = (iy - jy) / (double)(ix - jx);
            
            ans += mp[k];
            mp[k] ++;
            
            //两条边平行就是梯形，两边平行且相等就是平四，且两边对于平四都计算了两次
            ll dis = (ix - jx) * (ix - jx) + (iy - jy) * (iy - jy);
            rects += rect[{k,dis}];
            rect[{k,dis}] ++;
        }
    }
    cout << ans - rects / 2 << endl;
}

int main()
{
    ios_base::sync_with_stdio(0),cin.tie(NULL),cout.tie(NULL);
    solve();
    return 0;
}