QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#175770 | #5460. Sum of Numbers | realIyxiang | WA | 1ms | 3832kb | C++23 | 3.7kb | 2023-09-10 22:50:01 | 2023-09-10 22:50:01 |
Judging History
answer
#include <bits/stdc++.h>
#define eb emplace_back
#define ep emplace
#define fi first
#define se second
#define in read<int>()
#define lin read<ll>()
#define rep(i, x, y) for(int i = (x); i <= (y); i++)
#define per(i, x, y) for(int i = (x); i >= (y); i--)
using namespace std;
using ll = long long;
using db = double;
using pii = pair < int, int >;
using vec = vector < int >;
using veg = vector < pii >;
template < typename T > T read() {
T x = 0; bool f = 0; char ch = getchar();
while(!isdigit(ch)) f |= ch == '-', ch = getchar();
while(isdigit(ch)) x = x * 10 + (ch ^ 48), ch = getchar();
return f ? -x : x;
}
template < typename T > void chkmax(T &x, const T &y) { x = x > y ? x : y; }
template < typename T > void chkmin(T &x, const T &y) { x = x < y ? x : y; }
const int N = 1e6 + 10;
//constexpr ll B = 1e16;
constexpr int B = 10;
int n, K;
string num;
vec tnum;
const bool operator < (const vec &x, const vec &y) {
//cerr << "!COMP: " << x << " " << y << endl;
if(x.size() != y.size()) return x.size() < y.size();
per(i, (int)x.size() - 1, 0)
if(x[i] != y[i]) return x[i] < y[i];
return 1;
}
ll pw[20];
void compose(vec &x) { // zhengxu 1212120
vec ret; reverse(x.begin(), x.end()); // 0121221
rep(i, 0, (int)x.size() - 1) {
if(i % 16) ret.back() = x[i] * pw[i % 16] + ret.back();
else ret.eb(x[i]);
} x = ret;
}
void depose(vec &x) { // 01212121
//reverse(x.begin(), x.end());
vec ret;
rep(i, 0, (int)x.size() - 1) {
rep(j, 0, 15) ret.eb( x[i] % 10 ), x[i] /= 10;
} while(ret.size() && ret.back() == 0) ret.pop_back(); x = ret;
}
vec getp(int l, int r) {
vec ret; l--, r--;
rep(i, l, r) ret.eb(tnum[i]);
reverse(ret.begin(), ret.end());
//compose(ret);
//cout << l << " " << r << " " << ret[0] << endl;
return ret;
}
int len;
void operator *= (vec &a, const vec &b) {
a.resize(max(a.size(), b.size()) + 3);
rep(i, 0, a.size() - 1) {
if(i < b.size()) a[i] += b[i];
if(a[i] >= B) a[i + 1] += a[i] / B, a[i] %= B;
} while(a.size() && a.back() == 0) a.pop_back();
}
vec get(const vec &pot) {
int cur = 1; vec ret(len + 2, 0);
//for(auto v : pot) cerr << v << " "; cerr << endl;
for(auto v : pot) {
int l = cur, r = l + v - 1;
//cerr << l << " " << r << " " << getp(l, r) << endl;
rep(j, l, r)
ret[l + v - 1 - j] += num[j];
} //cerr << "!" << ret << endl;
rep(j, 0, ret.size() - 1) if(ret[j] >= 10) ret[j + 1] += ret[j] / 10, ret[j] %= 10;
while(ret.size() && ret.back() == 0) ret.pop_back();
return ret;
}
void solve() {
cin >> n >> K >> num;
//reverse(num.begin(), num.end());
vec().swap(tnum);
len = n / (K + 1);
for(auto &v : num) v -= '0', tnum.eb(v);
vec ans; bool fl = 0;
auto upd = [&](const vec &ret) {
if(!fl) ans = ret, fl = 1;
else chkmin(ans, ret);
};
auto runit = [&](int chk) {
int t = 1; rep(i, 1, K) t = t * 3;
rep(s, 0, t - 1) {
vec pot(K + 1, len); int x = s;
rep(j, 0, K - 1) pot[j] += x % 3 - 1, x /= 3;
pot[K] = n;
rep(j, 0, K - 1) pot[K] -= pot[j];
int tfl = 0;
rep(j, 0, K) if(pot[j] <= 0 || pot[j] < len - 1 || pot[j] > len + 1) tfl = 1;
if(tfl) continue;
rep(j, 0, K) tfl |= pot[j] == len + 1;
if(chk && !tfl) continue;
if(!fl) fl = 1, ans = get(pot);
else chkmin(ans, get(pot));
}
};
runit(0); len++; runit(1); //depose(ans);
//cerr << ct << endl;
reverse(ans.begin(), ans.end());
for(auto v : ans) cout << v; cout << endl;
}
int main() {
#ifdef YJR_2333_TEST
freopen("1.in", "r", stdin);
#endif
pw[0] = 1; rep(i, 1, 18) pw[i] = pw[i - 1] * 10;
ios :: sync_with_stdio(false), cin.tie(0), cout.tie(0);
int T; cin >> T;
for(; T; T--) solve(); return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 0
Wrong Answer
time: 1ms
memory: 3832kb
input:
2 8 1 45455151 2 1 42
output:
10910 4
result:
wrong answer 1st lines differ - expected: '9696', found: '10910'