QOJ.ac
QOJ
ID | 题目 | 提交者 | 结果 | 用时 | 内存 | 语言 | 文件大小 | 提交时间 | 测评时间 |
---|---|---|---|---|---|---|---|---|---|
#333501 | #5874. Mystery Square | hos_lyric | 41 ✓ | 697ms | 3784kb | C++14 | 5.6kb | 2024-02-20 00:32:03 | 2024-02-20 00:32:04 |
Judging History
answer
#include <cassert>
#include <cmath>
#include <cstdint>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <bitset>
#include <complex>
#include <deque>
#include <functional>
#include <iostream>
#include <limits>
#include <map>
#include <numeric>
#include <queue>
#include <random>
#include <set>
#include <sstream>
#include <string>
#include <unordered_map>
#include <unordered_set>
#include <utility>
#include <vector>
using namespace std;
// using Int = long long;
template <class T1, class T2> ostream &operator<<(ostream &os, const pair<T1, T2> &a) { return os << "(" << a.first << ", " << a.second << ")"; };
template <class T> ostream &operator<<(ostream &os, const vector<T> &as) { const int sz = as.size(); os << "["; for (int i = 0; i < sz; ++i) { if (i >= 256) { os << ", ..."; break; } if (i > 0) { os << ", "; } os << as[i]; } return os << "]"; }
template <class T> void pv(T a, T b) { for (T i = a; i != b; ++i) cerr << *i << " "; cerr << endl; }
template <class T> bool chmin(T &t, const T &f) { if (t > f) { t = f; return true; } return false; }
template <class T> bool chmax(T &t, const T &f) { if (t < f) { t = f; return true; } return false; }
#define COLOR(s) ("\x1b[" s "m")
#ifndef LIBRA_OTHER_INT128_H_
#define LIBRA_OTHER_INT128_H_
#include <stdio.h>
#include <iostream>
constexpr unsigned __int128 toUInt128(const char *s) {
unsigned __int128 x = 0;
for (; *s; ++s) x = x * 10 + (*s - '0');
return x;
}
constexpr __int128 toInt128(const char *s) {
if (*s == '-') return -toInt128(s + 1);
__int128 x = 0;
for (; *s; ++s) x = x * 10 + (*s - '0');
return x;
}
unsigned __int128 inUInt128() {
static char buf[41];
scanf("%s", buf);
return toUInt128(buf);
}
__int128 inInt128() {
static char buf[41];
scanf("%s", buf);
return toInt128(buf);
}
void out(unsigned __int128 x) {
static char buf[41];
int len = 0;
do { buf[len++] = '0' + static_cast<int>(x % 10); } while (x /= 10);
for (int i = len; --i >= 0; ) putchar(buf[i]);
}
void out(__int128 x) {
if (x < 0) {
putchar('-');
out(-static_cast<unsigned __int128>(x));
} else {
out(static_cast<unsigned __int128>(x));
}
}
std::ostream &operator<<(std::ostream &os, unsigned __int128 x) {
static char buf[41];
int len = 0;
do { buf[len++] = '0' + static_cast<int>(x % 10); } while (x /= 10);
for (int i = len; --i >= 0; ) os << buf[i];
return os;
}
std::ostream &operator<<(std::ostream &os, __int128 x) {
if (x < 0) {
os << '-' << -static_cast<unsigned __int128>(x);
} else {
os << static_cast<unsigned __int128>(x);
}
return os;
}
#endif // LIBRA_OTHER_INT128_H_
void out2(unsigned __int128 x) {
static char buf[130];
int len = 0;
do { buf[len++] = '0' + static_cast<int>(x % 2); } while (x /= 2);
for (int i = len; --i >= 0; ) putchar(buf[i]);
}
using INT = __int128;
inline int half(int E) {
return (E + 1) / 2;
}
INT solveHi(int E, INT A, INT B) {
// cerr<<"[solveHi] "<<E<<" "<<A<<" "<<B<<endl;
const INT base = A >> half(E) << half(E);
const INT mask = (B - A) >> half(E) << half(E);
for (INT sub = mask; ; --sub &= mask) {
const INT l = base | sub;
const INT r = l + ((INT)1 << half(E));
INT lo = 0, hi = 1LL << half(E);
for (; lo + 1 < hi; ) {
const INT mid = (lo + hi) / 2;
((mid * mid >= l) ? hi : lo) = mid;
}
// cerr<<" ["<<l<<", "<<r<<"): ceil(sqrt(l)) = "<<hi<<endl;
for (INT x = hi, y; (y = x * x) < r; ++x) {
if (!(A & ~y) && !(y & ~B)) {
// cerr<<" found "<<x<<"^2 = "<<y<<endl;
return x;
}
}
if (!sub) break;
}
return 0;
}
// r mod 2^e
INT solveLoDfs(int E, INT A, INT B, int e, INT r) {
// cerr<<" [solveLoDfs] "<<E<<" "<<A<<" "<<B<<" "<<e<<" "<<r<<endl;
if (e == half(E)) {
const INT x = r;
const INT y = x * x;
if (!(A & ~y) && !(y & ~B)) {
// cerr<<" found "<<x<<"^2 = "<<y<<endl;
return x;
}
return 0;
} else {
// (r + 2^e s)^2 == r^2 + 2^(e+1) r s (mod 2^(e+2)) for e >= 2
if (e == 1 || e == half(E) - 1 || ((B - A) >> (e+1) & 1)) {
for (INT s = 0; s < 2; ++s) {
const INT res = solveLoDfs(E, A, B, e + 1, r | s << e);
if (res) return res;
}
return 0;
} else {
const INT s = (A ^ (r * r)) >> (e+1) & 1;
return solveLoDfs(E, A, B, e + 1, r | s << e);
}
}
}
INT solveLo(int E, INT A, INT B) {
cerr<<"[solveLo] "<<E<<" "<<A<<" "<<B<<endl;
return solveLoDfs(E, A, B, 1, 1);
}
INT solve(INT A, INT B) {
int E = 0;
for (; !(B < (INT)1 << E); ++E) {}
const int popHi = __builtin_popcountll((B - A) >> half(E));
const int popLo = __builtin_popcountll((B - A) & (((INT)1 << half(E)) - 1));
if (popHi < popLo) {
return solveHi(E, A, B);
} else {
return solveLo(E, A, B);
}
}
int main() {
char S[130];
for (int numCases; ~scanf("%d", &numCases); ) { for (int caseId = 1; caseId <= numCases; ++caseId) {
scanf("%s", S);
const int SLen = strlen(S);
INT A = 0, B = 0;
for (int e = 0; e < SLen; ++e) {
if (S[SLen - 1 - e] == '1') A |= (INT)1 << e;
if (S[SLen - 1 - e] != '0') B |= (INT)1 << e;
}
INT ans = 0;
for (int e = 0; B >> (2*e); ++e) {
const INT res = solve(A >> (2*e), B >> (2*e));
if (res) {
ans = res << e;
break;
}
}
cerr<<"ans = "<<ans<<", ans^2 = "<<ans<<endl;
printf("Case #%d: ", caseId);
out2(ans * ans);
puts("");
}
#ifndef LOCAL
break;
#endif
}
return 0;
}
詳細信息
Subtask #1:
score: 10
Accepted
Test #1:
score: 10
Accepted
time: 2ms
memory: 3776kb
input:
25 1??? 1 10??110??00??1000?? 1??010?0110?1010?0?010?0111011?11100?100010?0??0??1 1??11????00??1?1?0?1??01??110110?11?00100110?00100?0?00 11?1?1???11111?11?1??11110000?00?????00??0?000?000?1 10??000000?0?00000?00000000??0?0000???00??????0000??? 101???11??11000?????1??1?1??10??0?0100011?0001?01011001...
output:
Case #1: 1001 Case #2: 1 Case #3: 1011110110000100001 Case #4: 111010001100101000001010111011011100110001000110001 Case #5: 1101111110000101100101010111011001110010011000010000100 Case #6: 1111111111111111111111111000000000000000000000000001 Case #7: 1000000000000000000000000000000000000000000000000...
result:
ok 25 lines
Subtask #2:
score: 31
Accepted
Test #2:
score: 31
Accepted
time: 697ms
memory: 3784kb
input:
25 1????????????????????111101010000011100110101111000001011111100110000011000101100000010010110100101000???????????????????? 10?11100?000111??11?01010110100??1?100111?001000000??0101?110?0111?011?11?1??00010111?010??100?100??10?010?001001110111110?1 1000100111100100110011010111100001111010?????????...
output:
Case #1: 11100101110101010110111110101000001110011010111100000101111110011000001100010110000001001011010010100001101011000010100001 Case #2: 1011110000001110111001010110100001110011100010000001101010110001111011111110000010111101010100010010101010100100111011111001 Case #3: 1000100111100100110011010...
result:
ok 25 lines
Extra Test:
score: 0
Extra Test Passed