QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#350773 | #8330. Count off 3 | hos_lyric | AC ✓ | 51ms | 17400kb | C++14 | 6.4kb | 2024-03-11 03:58:55 | 2024-04-16 16:08:56 |
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")
////////////////////////////////////////////////////////////////////////////////
template <unsigned M_> struct ModInt {
static constexpr unsigned M = M_;
unsigned x;
constexpr ModInt() : x(0U) {}
constexpr ModInt(unsigned x_) : x(x_ % M) {}
constexpr ModInt(unsigned long long x_) : x(x_ % M) {}
constexpr ModInt(int x_) : x(((x_ %= static_cast<int>(M)) < 0) ? (x_ + static_cast<int>(M)) : x_) {}
constexpr ModInt(long long x_) : x(((x_ %= static_cast<long long>(M)) < 0) ? (x_ + static_cast<long long>(M)) : x_) {}
ModInt &operator+=(const ModInt &a) { x = ((x += a.x) >= M) ? (x - M) : x; return *this; }
ModInt &operator-=(const ModInt &a) { x = ((x -= a.x) >= M) ? (x + M) : x; return *this; }
ModInt &operator*=(const ModInt &a) { x = (static_cast<unsigned long long>(x) * a.x) % M; return *this; }
ModInt &operator/=(const ModInt &a) { return (*this *= a.inv()); }
ModInt pow(long long e) const {
if (e < 0) return inv().pow(-e);
ModInt a = *this, b = 1U; for (; e; e >>= 1) { if (e & 1) b *= a; a *= a; } return b;
}
ModInt inv() const {
unsigned a = M, b = x; int y = 0, z = 1;
for (; b; ) { const unsigned q = a / b; const unsigned c = a - q * b; a = b; b = c; const int w = y - static_cast<int>(q) * z; y = z; z = w; }
assert(a == 1U); return ModInt(y);
}
ModInt operator+() const { return *this; }
ModInt operator-() const { ModInt a; a.x = x ? (M - x) : 0U; return a; }
ModInt operator+(const ModInt &a) const { return (ModInt(*this) += a); }
ModInt operator-(const ModInt &a) const { return (ModInt(*this) -= a); }
ModInt operator*(const ModInt &a) const { return (ModInt(*this) *= a); }
ModInt operator/(const ModInt &a) const { return (ModInt(*this) /= a); }
template <class T> friend ModInt operator+(T a, const ModInt &b) { return (ModInt(a) += b); }
template <class T> friend ModInt operator-(T a, const ModInt &b) { return (ModInt(a) -= b); }
template <class T> friend ModInt operator*(T a, const ModInt &b) { return (ModInt(a) *= b); }
template <class T> friend ModInt operator/(T a, const ModInt &b) { return (ModInt(a) /= b); }
explicit operator bool() const { return x; }
bool operator==(const ModInt &a) const { return (x == a.x); }
bool operator!=(const ModInt &a) const { return (x != a.x); }
friend std::ostream &operator<<(std::ostream &os, const ModInt &a) { return os << a.x; }
};
////////////////////////////////////////////////////////////////////////////////
constexpr unsigned MO = 1000000007;
using Mint = ModInt<MO>;
#define rep4(a) for (int a = 0; a < 4; ++a)
#define rep5(a) for (int a = 0; a < 5; ++a)
#define rep7(a) for (int a = 0; a < 7; ++a)
inline int add(int a, int b) { return ((a += b) >= 7) ? (a - 7) : a; }
inline int red(int a) { return min(a, 7 - a); }
constexpr int PW2[] = {1, 2, 4, 1, 2, 4};
constexpr int PW3[] = {1, 3, 2, 6, 4, 5};
constexpr int LIM = 10004;
/*
at even/odd degree
values at 1,2,3
*/
Mint dp[2][5010][7][7][7];
int SLen;
char S[10010];
int main() {
for (int h = 0; h < 2; ++h) {
dp[h][0][0][0][0] = 1;
for (int i = 0; i < LIM/2; ++i) {
if (2 * i + h != 0) {
memcpy(dp[h][i + 1], dp[h][i], sizeof(dp[h][i]));
}
const int p2 = PW2[(2 * i + h) % 6];
const int p3 = PW3[(2 * i + h) % 6];
rep7(a1) rep7(a2) rep7(a3) {
dp[h][i + 1][add(a1, 1)][add(a2, p2)][add(a3, p3)] += dp[h][i][a1][a2][a3];
}
}
}
for (int numCases; ~scanf("%d", &numCases); ) { for (int caseId = 1; caseId <= numCases; ++caseId) {
scanf("%s", S);
SLen = strlen(S);
reverse(S, S + SLen);
Mint ans = 0;
int now[2][4] = {};
for (int i = SLen; --i >= 0; ) if (S[i] == '1') {
if (i != 0) {
// smaller here
// 4 for wildcard
Mint f[2][5][5][5] = {};
for (int h = 0; h < 2; ++h) {
const int ii = (i + 1 - h) / 2;
rep7(a1) rep7(a2) rep7(a3) {
f[h]
[red(add(a1, now[h][1]))]
[red(add(a2, now[h][2]))]
[red(add(a3, now[h][3]))]
+= dp[h][ii][a1][a2][a3];
}
rep4(a1) rep5(a2) rep5(a3) f[h][4][a2][a3] += f[h][a1][a2][a3];
rep5(a1) rep4(a2) rep5(a3) f[h][a1][4][a3] += f[h][a1][a2][a3];
rep5(a1) rep5(a2) rep4(a3) f[h][a1][a2][4] += f[h][a1][a2][a3];
}
// PIE
Mint here = 0;
rep5(a1) rep5(a2) rep5(a3) {
const Mint t = f[0][a1][a2][a3] * f[1][a1][a2][a3];
(((a1 < 4) + (a2 < 4) + (a3 < 4)) & 1) ? (here -= t) : (here += t);
}
// cerr<<i<<": here = "<<here<<endl;
ans += here;
}
now[i & 1][1] = add(now[i & 1][1], 1);
now[i & 1][2] = add(now[i & 1][2], PW2[i % 6]);
now[i & 1][3] = add(now[i & 1][3], PW3[i % 6]);
}
if (S[0] == '1') {
bool ok = true;
for (int x = 1; x <= 3; ++x) {
ok = ok && (now[0][x] != now[1][x]);
ok = ok && (now[0][x] + now[1][x] != 7);
}
if (ok) {
ans += 1;
}
}
printf("%u\n", ans.x);
}
#ifndef LOCAL
break;
#endif
}
return 0;
}
这程序好像有点Bug,我给组数据试试?
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 0ms
memory: 17356kb
input:
5 1 1010 110101 1000111000 101101001000
output:
1 2 15 114 514
result:
ok 5 number(s): "1 2 15 114 514"
Test #2:
score: 0
Accepted
time: 3ms
memory: 17216kb
input:
10 1 11 1000 10111011 1000110100101001 11101110000001000011010011011000 110011000111110001101010101100100011010010101000011111001101011 11010111011101000010101111011111011011100001001101010011101011111111011011111101110110010011001101000001000111100010010111000010 10000000000000000000000000000000000...
output:
1 1 2 45 6591 814196699 193088128 849103726 497125329 363076920
result:
ok 10 numbers
Test #3:
score: 0
Accepted
time: 18ms
memory: 17356kb
input:
10 1 101 100101000 111111001111011001111100111 100001101010101000101110010111010010001101101110011111000001010001111100101010000 111001010100100100110011110111000111001001001001000100000011000110011000110101010010100000100101110101000011000011100010011001011000101110100111000110011011010111011111011...
output:
1 2 64 27062688 486363229 184013394 580592021 118930214 772664718 344619804
result:
ok 10 numbers
Test #4:
score: 0
Accepted
time: 26ms
memory: 17292kb
input:
10 1 1011 1101001010001110 1000010101110010000010010000000000001010111001001001110011001101 1001100101110111001000100100110111110001110010111011010101010111011101111101111100010000001100001001011100111100010110011010000010000000001100111011000001110011010000100000110101010011111100010111111100011011...
output:
1 3 10053 860833891 537931408 329471109 368911735 157523156 595487313 534701861
result:
ok 10 numbers
Test #5:
score: 0
Accepted
time: 32ms
memory: 17296kb
input:
10 1 11111 1010110010010000111001001 10011010100100001000110000111101101000111100001000000101010001100111111010001000101000000011100101000101100111100001001101100 11000100100010011101010101001011100010001100001010110011110101001101011000110001000111101010010000110111010001100100100111001000001000010...
output:
1 10 4692555 763463648 464152115 115362567 880780461 578723006 560068977 423846910
result:
ok 10 numbers
Test #6:
score: 0
Accepted
time: 41ms
memory: 17356kb
input:
10 1 101011 100100110100111100001101000101011100 1011011000011001101101010110000111011001001100110101111100110000100100101010000000110110010001110011101011000001011001000010001011101110110100110010111111000101101010110000101010101011001111100111011001101111011101 101000000111000010111000110000011000...
output:
1 13 955673880 266148454 368723690 496979115 190983211 772121423 932555320 843716403
result:
ok 10 numbers
Test #7:
score: 0
Accepted
time: 38ms
memory: 17332kb
input:
10 1 1101101 1100111111001000111100000010000111000000010101001 110111101101011000111101100110010011011100101110101111110011001111001101001000011001110011110101001110010110011110011001111010010101010011010011101101111010111000001110110111011011101000100001000001101110010111100110001110011101110111100...
output:
1 29 912933242 912560788 607401363 477602366 394403189 275067439 592568023 75193370
result:
ok 10 numbers
Test #8:
score: 0
Accepted
time: 42ms
memory: 17376kb
input:
10 1 10000010 100101110110100111100000100011111111010001100010100001100110001 111111000010011010111011111110000010101101011110100001101011110000001111001110001111110101000000010000001011000101101011010111011101111110101001000110011101010000111001011111100100010000010110110101010001110100111110110001...
output:
1 32 959140870 614330473 849221876 787816311 359958989 239371459 534701861 254356877
result:
ok 10 numbers
Test #9:
score: 0
Accepted
time: 44ms
memory: 17292kb
input:
10 1 111110011 111101001101011110100011110000100110011101010111111110100001111001101000100101101 11011010000110101011111110110011101010100100110001001111111011010000101111110001001011000011010101001000101110000100011011100101010110010101000101010111101100101110111100011011011000101101001001001100111...
output:
1 99 286317277 694681686 723919544 789291149 680541846 694957099 453387561 757810824
result:
ok 10 numbers
Test #10:
score: 0
Accepted
time: 45ms
memory: 17336kb
input:
10 1 1001001110 10110010111110100100111100111101101010111110111011001110000111101011011010110011110000000001110100 11000101011111110110100101100100100000010110001011100010101111111000000111000101100110111110101010111110010110111111110010110010001100000000111000101100010001110010001011111110101011111...
output:
1 120 987933828 449323095 435643580 557750562 122442298 758115947 388795572 87146822
result:
ok 10 numbers
Test #11:
score: 0
Accepted
time: 49ms
memory: 17200kb
input:
10 1 11010010100 100110110000100111011101001111000000000110111100011110111011110100001010101000000000100000101100100101110101011100111000 110111100000010010111000111011111100010100100111101001001101111010011011100100001010100010100011110111111100101011100111111011011000000111001111000101111010110111...
output:
1 325 391030697 323231960 401473132 822267612 841573845 283856764 804647498 76347459
result:
ok 10 numbers
Test #12:
score: 0
Accepted
time: 48ms
memory: 17336kb
input:
10 1 111111110011 100111010001010010111100011101110011110100101101010111001110101111000111010001111110000001000011010111010001001000011101100011100010010100010000 11001011101100010011111001010110110000110110011001011001000001001110010100100000000101100010001011010010001101000101110000111100100000001...
output:
1 704 677678115 593427859 667002509 574438492 664907465 979953874 8529137 613727900
result:
ok 10 numbers
Test #13:
score: 0
Accepted
time: 43ms
memory: 17336kb
input:
10 1 1100110101011 100111000010011100111101101100110010110000100011110101100100011001101011100011101101110111001101000001110010111001110011100101000111111000010101101100011000010100010101 1111010000010011010010011000010000000001000110111011101100111011100010110011100110011110011011110011110100011001...
output:
1 1146 832402516 402106502 689225542 416112434 991952024 938688647 880733772 630306115
result:
ok 10 numbers
Test #14:
score: 0
Accepted
time: 50ms
memory: 17400kb
input:
10 1 11000000000111 110010000100100011111101001100111010110111011101101011001001010110101111111101001000000100110011110101100111010110100100010100000100000011100010101011010001100001000111000101000011010110010100000 101001000110100011110100011001010101001010011111010111111001100111111100101111110111...
output:
1 2087 659256442 942088668 754989716 908871865 566839365 111034927 696022638 206335876
result:
ok 10 numbers
Test #15:
score: 0
Accepted
time: 45ms
memory: 17212kb
input:
10 1 111100001011110 101100010011110001000011110010011010011100110010111110100111111111100101100111010101001101111001111010111011011111000111000101101010001100111010100110110000111110100100101000001101111100000101101100010110101000011001110001101 11111110001100110111110110100010111010100010010010010...
output:
1 5612 120730460 903512843 440620378 736669452 35297346 414402862 87146822 461180872
result:
ok 10 numbers
Test #16:
score: 0
Accepted
time: 45ms
memory: 17288kb
input:
10 1 1011110111101110 11111011001101100000000011011111011000101001100010000000001010011110010000100000111100101011101111111111001000110011110000011001000111000010101100001001100111100100000010101101111100100100101110101100000000011101011100010111111010000101011000110010011000 11110011001110110000001...
output:
1 9074 47298040 806126372 607928251 829797230 861514498 6535505 135611721 148853296
result:
ok 10 numbers
Test #17:
score: 0
Accepted
time: 47ms
memory: 17352kb
input:
10 1 11101111010010001 1111011100010101011101000110010001011001010101000100111100010110101010010001100101001001011111101001101100110100100100101101011001100000101011000100001011101000101000110000110100100100001001000011000111000010100011001111111011010001110111101111010011100010110001010010000100001...
output:
1 25268 485539600 497476229 129697011 91489334 354698980 228961474 875061949 618786188
result:
ok 10 numbers
Test #18:
score: 0
Accepted
time: 42ms
memory: 17216kb
input:
10 1 101101011100110001 100111001100000111101000101110011011011011111101011110101111000010000101010001010110100101001001100010101100101001110110001101101100111111100001000100010010110101110010111101100110010000010001101011001110001001100111101100111010100100000000000010000100101001000110000111100100...
output:
1 38342 769759919 33866310 945890505 127750526 125262837 888967227 757810824 441419016
result:
ok 10 numbers
Test #19:
score: 0
Accepted
time: 51ms
memory: 17356kb
input:
10 1 1001100011010110111 11100110010001111111110100011111100100011011110000110100000111101100111110111010111010111001111111100111011000000101001111000010001100010001111001011000111111001111100100101101011001110011100000111011111110101000111011101101110101101101110101100000011000001001011011100001111...
output:
1 65218 438898572 348219276 776140964 704823526 170625715 198310775 477853700 897436999
result:
ok 10 numbers
Test #20:
score: 0
Accepted
time: 48ms
memory: 17264kb
input:
10 1 11010100111111001011 1000011001001001110110101001100001001101001001010010101010001110011001001000000000110001100110001110110111010100101011011001100101001110111001101101000111101100010110101100101110111101000101111010100110001011110111000110000110110111011101110010010011010001110101010110010000...
output:
1 183823 238142747 846693477 959968477 260267123 642987070 779134130 951392182 679687101
result:
ok 10 numbers
Extra Test:
score: 0
Extra Test Passed