/ SeriousOJ /

Record Detail

Accepted


  
# Status Time Cost Memory Cost
#1 Accepted 1ms 320.0 KiB

Code

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

int main() {
    string s;
    getline(cin, s); // Read the full first line
    int a, b, last;

    // Remove all whitespace just in case
    s.erase(remove(s.begin(), s.end(), ' '), s.end());

    // Check if input contains "..."
    bool end = s.find("...") != string::npos;

    vector<int> terms;
    string temp = "";
    for (char ch : s) {
        if (isdigit(ch)) {
            temp += ch;
        } else if (ch == '+' || ch == '.') {
            if (!temp.empty()) {
                terms.push_back(stoi(temp));
                temp = "";
            }
        }
    }
    if (!temp.empty()) terms.push_back(stoi(temp)); // last number if no end

    if (terms.size() < 2) {
        cerr << "Invalid input format\n";
        return 1;
    }

    a = terms[0];
    b = terms[1];

    cin >> last;

    int d = b - a;
    int sum = (2 * a + (last - 1) * d) * last / 2;

    cout << sum << "\n";
    return 0;
}

Information

Submit By
Type
Pretest
Problem
P1231 Busy Friend
Language
C++17 (G++ 13.2.0)
Submit At
2025-09-02 16:56:21
Judged At
2025-09-02 16:56:21
Judged By
Score
10
Total Time
1ms
Peak Memory
320.0 KiB