atcoder abc 360

A a healthy breakfast

问题：

思路：扫描给定字符串，看M和R哪个先出现

代码：

#include <iostream>

using namespace std;

int main() {
    int cnt = 0;
    string str;
    cin >> str;
    for(auto t: str) {
        if(t == 'R') {
            cout << "Yes";
            return 0;
        } else if(t == 'M') {
            cout << "No";
            return 0;
        }
    }

    return 0; 
}

B vertical reading

问题：

思路：三层循环，第一层循环枚举w，第二层循环枚举w个字母中的第i位，第三层循环枚举被划分后的每个字符串的第i位

代码：

#include <bits/stdc++.h>

using namespace std;

int main() {
    string a, b;
    cin >> a >> b;
    
    int n = max(a.size(), b.size());
    for(int i = 1; i < n; i ++ ) {
        for(int k = 1; k <= i; k ++ ) {
            string str;
            for(int j = k; j <= a.size(); j += i ) {
                str += a[j - 1];
            }
            if(str == b) {
                cout << "Yes";
                return 0;
            }
        }
    }
    cout << "No" << endl;
    return 0;
}

C Move it

问题：

思路：输出所有装有至少2个物品的盒子中除了最大质量的物品其余物品的质量和

代码：

#include <bits/stdc++.h>

using namespace std;

int main() {
    int n;
    cin >> n;
    vector<int> a(n + 1);
    vector<int> w(n + 1);
    for(int i = 1; i <= n; i ++ ) cin >> a[i];
    for(int i = 1; i <= n; i ++ ) cin >> w[i];
    vector<vector<int>> q((n + 1));
    for(int i = 1; i <= n; i ++ ) {
        q[a[i]].push_back(w[i]);
    }

    long long ans = 0;
    for(int i = 1; i <= n; i ++ ) {
        sort(q[i].begin(), q[i].end());
        if(q[i].size() > 1) {
            for(int j = 0; j < q[i].size() - 1; j ++ ) ans += q[i][j];
        }
    }
    cout << ans;
    return 0;
}

D ghost ants

问题：

思路：

将向左移动的蚂蚁视作静止，同时向右移动的蚂蚁每个单位时间移动2个单位长度

将向左移动的蚂蚁与向右移动的蚂蚁分开存储

对向左移动的蚂蚁排序

枚举向右移动的蚂蚁x

开始二分：

第一个二分找到向左移动的蚂蚁中第一个坐标大于等于x蚂蚁的坐标，如果找不到，则说明x不会对答案有贡献，直接continue，反之如果找到了，那么开始第二个二分，第二个二分查找的是最后一个小于等于 x蚂蚁的坐标 y 加上 2 * T的蚂蚁

代码：

#include <bits/stdc++.h>

using namespace std;

int main() {
    int n, T;
    cin >> n >> T;
    vector<char> str(n + 1);
    for(int i = 1; i <= n; i ++ ) cin >> str[i];
    vector<long long> fl, fr;
    fl.push_back(0);
    fr.push_back(0);

    for(int i = 1; i <= n; i ++ ) {
        int x;
        cin >> x;
        if(str[i] == '0') fl.push_back(x);
        else fr.push_back(x);
    }
    
    sort(fl.begin() + 1, fl.end());

    long long ans = 0;    
    for(int i = 1; i <= fr.size() - 1; i ++ ) {
        int l = 1, r = fl.size() - 1;
        long long x = fr[i];
        while(l < r) {
            int mid = l + r >> 1;
            if(x <= fl[mid]) r = mid;
            else l = mid + 1;
        }
        if(x > fl[l]) continue;
        x += 2 * T;
        int tmp = l;
        r = fl.size() - 1;
        while(l < r) {
            int mid = l + r + 1 >> 1;
            if(x >= fl[mid]) l = mid;
            else r = mid - 1;
        }
        if(x >= fl[l]) ans += l - tmp + 1;
    }
    cout << ans;
    return 0;
}