这题本来觉得是道没什么的水题,结果没想到空间卡得这么死
于是注意观察式子,发现每个推到后面最多加7 * 9 (7位数,每位是9)
于是就只要开64的空间,利用滚动数组去优化即可
另外还要注意,答案表全打出来也是会MLE的,只能根据输入的k去存放k个答案
代码:
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
int n, k, dp[65], sum[10005];
int get(int x) {
int sum = 0;
while (x) {
sum += x % 10;
x /= 10;
}
return sum;
}
struct A {
int a, id;
void read(int id) {
this->id = id;
scanf("%d", &a);
}
} x[5005];
bool cmp(A a, A b) {
return a.a < b.a;
}
int ans[5005];
int main() {
for (int i = 1; i <= 10000; i++) sum[i] = get(i);
scanf("%d%d", &n, &k);
int out = 0;
for (int i = 0; i < k; i++)
x[i].read(i);
sort(x, x + k, cmp);
int u = 0;
for (int i = 1; i <= n; i++) {
if (!dp[i % 64]) {
out++;
while (x[u].a <= out) {
if (x[u].a == out) ans[x[u].id] = i;
u++;
}
}
dp[(i + sum[i / 10000] + sum[i % 10000]) % 64] = true;
dp[i % 64] = false;
}
printf("%d\n", out);
for (int i = 0; i < k; i++)
printf("%d%c", ans[i], i == k - 1 ? '\n' : ' ');
return 0;
}