LeetCode #3756

Concatenate Non-Zero Digits and Multiply by Sum II

1개의 풀이 · C++

문제 원문 보기 ↗

SOLUTION INFO

C++ · main.cpp

main.cpp
using ll = long long;
const ll MOD = 1e9 + 7;

class Solution {
public:
    vector<int> sumAndMultiply(string s, vector<vector<int>>& queries) {
        int N = (int)s.size();
        vector<ll> prefix_sum(N + 1), prefix_concat(N + 1);
        vector<int> cnt(N + 1);
        vector<ll> pw(N + 1, 1);
        for(int i = 1; i <= N; ++i) pw[i] = pw[i - 1] * 10 % MOD;
        for(int i = 1; i <= N; ++i) {
            int x = s[i - 1] - 48;
            prefix_sum[i] = prefix_sum[i - 1] + x;
            if(x != 0) {
                prefix_concat[i] = (prefix_concat[i - 1] * 10 % MOD + x) % MOD;
                cnt[i] = cnt[i - 1] + 1;
            }
            else {
                prefix_concat[i] = prefix_concat[i - 1];
                cnt[i] = cnt[i - 1];
            }
        }
        int Q = (int)queries.size();
        vector<int> answer;
        for(int i = 0; i < Q; ++i) {
            int l = queries[i][0] + 1;
            int r = queries[i][1] + 1;
            int sum = prefix_sum[r] - prefix_sum[l - 1];
            int c = cnt[r] - cnt[l - 1];
            ll ret = (prefix_concat[r] - pw[c] * prefix_concat[l - 1] % MOD) % MOD;
            if(ret < 0) ret += MOD;
            ret = ret * sum % MOD;
            answer.push_back(ret);
        }
        return answer;
    }
};

SOLUTION DESCRIPTION

풀이 설명

등록된 풀이 설명이 없습니다.

GITHUB COMMUNITY

커뮤니티 평가

GitHub 계정으로 남긴 최신 평가 1건만 반영하며, 데이터는 매일 저장소에 동기화됩니다.

최근 동기화

체감 난이도

아직 평가 없음

첫 난이도 평가를 남겨주세요.

문제 추천

아직 평가 없음

이 문제가 도움이 되었는지 알려주세요.

이 문제 평가하기

GitHub 로그인 후 난이도를 제출하고, 원하는 경우 추천 여부도 함께 남길 수 있습니다.

추천 여부 (선택)

평가하려면 GitHub로 로그인해 주세요.

평가는 자동으로 저장소 Discussion에 기록되므로 Discussion 화면을 직접 열 필요가 없습니다. 같은 문제를 다시 평가하면 기존 평가가 갱신됩니다.

DISCUSSION

댓글

GitHub 로그인 후 작성할 수 있으며 모든 댓글은 이 저장소의 Discussions에 보관됩니다.