LeetCode
找出字符串的可整除数组
题目链接:2575. 找出字符串的可整除数组 - 力扣(LeetCode)
题目描述
给你一个下标从 0 开始的字符串 word
,长度为 n
,由从 0
到 9
的数字组成。另给你一个正整数 m
。
word
的 可整除数组 div
是一个长度为 n
的整数数组,并满足:
- 如果
word[0,...,i]
所表示的 数值 能被m
整除,div[i] = 1
- 否则,
div[i] = 0
返回 word
的可整除数组。
示例 1:
输入:word = "998244353", m = 3
输出:[1,1,0,0,0,1,1,0,0]
解释:仅有 4 个前缀可以被 3 整除:"9"、"99"、"998244" 和 "9982443" 。
示例 2:
输入:word = "1010", m = 10
输出:[0,1,0,1]
解释:仅有 2 个前缀可以被 10 整除:"10" 和 "1010" 。
提示:
1 <= n <= 105
word.length == n
word
由数字0
到9
组成1 <= m <= 109
思路
代码
C++
class Solution {
public:
vector<int> divisibilityArray(string word, int m) {
vector<int> ans(word.length());
long long x = 0;
for(int i = 0; i < word.length(); i++){
x = (x * 10 + (word[i] - '0')) % m;
ans[i] = x == 0;
}
return ans;
}
};
Java
class Solution {
public int[] divisibilityArray(String word, int m) {
char[] s = word.toCharArray();
int[] ans = new int[s.length];
long x = 0;
for(int i = 0; i < s.length; i++){
x = (x * 10 + (s[i] - '0')) % m;
if(x == 0){
ans[i] = 1;
}
}
return ans;
}
}