AcWing 338. 计数问题
输入样例:
1 10 44 497 346 542 1199 1748 1496 1403 1004 503 1714 190 1317 854 1976 494 1001 1960 0 0
输出样例:
1 2 1 1 1 1 1 1 1 1 85 185 185 185 190 96 96 96 95 93 40 40 40 93 136 82 40 40 40 40 115 666 215 215 214 205 205 154 105 106 16 113 19 20 114 20 20 19 19 16 107 105 100 101 101 197 200 200 200 200 413 1133 503 503 503 502 502 417 402 412 196 512 186 104 87 93 97 97 142 196 398 1375 398 398 405 499 499 495 488 471 294 1256 296 296 296 296 287 286 286 247
分情况讨论 :[a,b] 0 ~ 9
利用 count(n, x) 表示 1 ~ n 中 x 出现的次数(前缀和思想)
然后利用 count(b,x) - count(a,x)
思想:求区间值的时候可以转换成前缀和求和然后作差
以 为例,分别求出1在每一位上出现的次数
求1在第4位上出现的次数:
(1)
(2)
(2.1)
(2.2)
(2.3)
#include <iostream> #include <algorithm> #include <vector> using namespace std; const int N = 10; /* 001~abc-1, 999 abc 1. num[i] < x, 0 2. num[i] == x, 0~efg 3. num[i] > x, 0~999 */ int get(vector<int> num, int l, int r) { int res = 0; for (int i = l; i >= r; i -- ) res = res * 10 + num[i]; return res; } int power10(int x) { int res = 1; while (x -- ) res *= 10; return res; } int count(int n, int x) { if (!n) return 0; vector<int> num; while (n) { num.push_back(n % 10); n /= 10; } n = num.size(); int res = 0; for (int i = n - 1 - !x; i >= 0; i -- ) { if (i < n - 1) { res += get(num, n - 1, i + 1) * power10(i); if (!x) res -= power10(i); } if (num[i] == x) res += get(num, i - 1, 0) + 1; else if (num[i] > x) res += power10(i); } return res; } int main() { int a, b; while (cin >> a >> b , a) { if (a > b) swap(a, b); for (int i = 0; i <= 9; i ++ ) cout << count(b, i) - count(a - 1, i) << ' '; cout << endl; } return 0; }