QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#402876 | #1286. Ternary String Counting | Eray | WA | 1ms | 3888kb | C++14 | 2.6kb | 2024-05-01 17:06:53 | 2024-05-01 17:06:54 |
Judging History
answer
#include <bits/stdc++.h>
typedef long long LL;
const int N = 5000 + 5;
const int M = 1e6 + 5;
const LL MOD = 1e9 + 7;
int n, m;
int jl[N], jr[N], kl[N], kr[N];
int nl[N], nr[N];
LL f[N][N], sumr[N], sumc[N];
int main() {
int T; scanf("%d", &T);
while(T--) {
scanf("%d%d", &n, &m);
for(int i = 1; i <= n; i++) jl[i] = 0, jr[i] = i - 1, kl[i] = 0, kr[i] = std::max(i - 2, 0);
for(int i = 1; i <= m; i++) {
int l, r, x;
scanf("%d%d%d", &l, &r, &x);
if(x == 1) jr[r] = std::min(jr[r], l - 1), kr[r] = std::min(kr[r], std::max(l - 2, 0));
else if(x == 2) jl[r] = std::max(jl[r], l), kr[r] = std::min(kr[r], l - 1);
else if(x == 3) kl[r] = std::max(kl[r], l);
}
bool flag = true;
for(int i = 1; i <= n; i++) if(jl[i] > jr[i] || kl[i] > kr[i]) { puts("0"); flag = false; break; }
if(!flag) continue;
for(int j = 0; j <= n; j++) for(int k = 0; k <= n; k++) f[j][k] = 0;
for(int i = 0; i <= n; i++) sumr[i] = sumc[i] = 0, nl[i] = 0, nr[i] = n;
f[0][0] = 1, sumr[0] = sumc[0] = 1;
auto reset = [&](int i, int j) { (sumr[i] += MOD - f[i][j]) %= MOD, (sumc[j] += MOD - f[i][j]) %= MOD, f[i][j] = 0; };
for(int i = 1; i <= n; i++) {
// f[i - 1][j][k] -> f[i][i - 1][k]
for(int k = 0; k <= i - 3; k++) {
LL val = (sumc[k] - (k == 0 ? f[0][0] : 0LL) + MOD) % MOD;
(f[i - 1][k] += val) %= MOD, (sumr[i - 1] += val) %= MOD, (sumc[k] += val) %= MOD;
}
// f[i - 1][j][k] -> f[i][i - 1][j]
for(int j = 0; j <= i - 2; j++) {
LL val = sumr[j];
(f[i - 1][j] += val) %= MOD, (sumr[i - 1] += val) %= MOD, (sumc[j] += val) %= MOD;
}
// remain (jl[i]..jr[i], kl[i]..kr[i])
for(int j = 0; j < jl[i]; j++) {
for(int k = nl[j]; k <= nr[j]; k++) reset(j, k);
nl[j] = 1, nr[j] = 0;
}
for(int j = jr[i] + 1; j <= i - 1; j++) {
for(int k = nl[j]; k <= nr[j]; k++) reset(j, k);
nl[j] = 1, nr[j] = 0;
}
for(int j = jl[i]; j <= jr[i]; j++) {
while(nl[j] <= nr[j] && nl[j] < kl[j]) reset(j, nl[j]++);
while(nr[j] >= nl[j] && nr[j] > kr[j]) reset(j, nr[j]--);
}
// printf("i = %d: j[%d, %d], k[%d, %d]\n", i, jl[i], jr[i], kl[i], kr[i]);
// for(int j = 0; j < i; j++) for(int k = 0; k < std::max(j, 1); k++)
// if(f[j][k]) printf("f[%d][%d][%d] = %lld\n", i, j, k, f[j][k]);
}
LL ans = 0;
for(int j = jl[n]; j <= jr[n]; j++) for(int k = kl[n]; k <= kr[n]; k++)
if(!j && !k) (ans += f[j][k] * 3) %= MOD;
else (ans += f[j][k] * 6) %= MOD;
printf("%lld\n", ans);
}
return 0;
} /*
1
9 3
2 3 2
1 9 2
2 4 2
1
9 2
2 3 2
1 9 2
*/
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 0
Wrong Answer
time: 1ms
memory: 3888kb
input:
4 1 0 2 0 3 0 5 2 1 3 3 4 5 1
output:
3 9 21 27
result:
wrong answer 3rd words differ - expected: '27', found: '21'