#include "grader.h"
#include "encoder.h"
#include<bits/stdc++.h>
#define st first
#define nd second
#define pb push_back
#define eb emplace_back
#define pp pop_back
#define all(x) (x).begin(), (x).end()
using namespace std;
using vi = vector<int>;
using pii = pair<int, int>;
using vii = vector<pii>;
void nadaj(vi &V, int x, int b){
for(int i=0; i<b; i++){
V.pb(!!(x&(1<<i)));
}
}
void encode2(vi &ans, int n, int h, int m, int *v1, int *v2){
vi E[n];
int czy[n][h], dist[n][h];
for(int i=0; i<n; i++){
for(int j=0; j<h; j++){
czy[i][j]=(i==j);
dist[i][j]=1e9*(i!=j);
}
}
vi edg;
for(int i=0; i<m; i++){
E[v1[i]].pb(edg.size());
edg.pb(v2[i]);
E[v2[i]].pb(edg.size());
edg.pb(v1[i]);
}
for(int s=0; s<h; s++){
vi V;
V.pb(s);
for(int i=0; i<V.size(); i++){
int v=V[i];
for(int e:E[v]){
if(dist[edg[e]][s]==1e9){
dist[edg[e]][s]=dist[v][s]+1;
V.pb(edg[e]);
}
}
}
}
vi E2[n];
bool spec[n];
for(int v=0; v<n; v++){
int ile=0;
vii V2;
for(int u=0; u<n; u++){
if(u==v)continue;
bool b=1;
for(int i=0; i<h; i++){
if(dist[u][i]+1<dist[v][i] || dist[v][i]+1<dist[u][i]){
b=0;
}
}
if(!b)continue;
b=0;
for(int i=0; i<h; i++){
if(!czy[u][i] && dist[u][i]==dist[v][i]+1){
V2.eb(u, i);
b=1;
}
if(!czy[v][i] && dist[v][i]==dist[u][i]+1){
V2.eb(v, i);
b=1;
}
}
if(b)ile++;
}
sort(all(V2));
V2.resize(unique(all(V2))-V2.begin());
spec[v]=(ile>=111);
if(ile>=111){
for(int u=0; u<n; u++){
if(u==v)continue;
bool b=1;
for(int i=0; i<h; i++){
if(dist[u][i]+1<dist[v][i] || dist[v][i]+1<dist[u][i]){
b=0;
}
}
if(!b)continue;
for(int i=0; i<h; i++){
if(!czy[u][i] && dist[u][i]==dist[v][i]+1){
czy[u][i]=1;
E2[v].pb(u);
}
if(!czy[v][i] && dist[v][i]==dist[u][i]+1){
czy[v][i]=1;
E2[v].pb(u);
}
}
}
}
}
int sum=0;
for(int t=h; t>0; t--){
for(int u=0; u<n; u++){
for(int v=0; v<u; v++){
bool b=1;
for(int i=0; i<h; i++){
if(dist[u][i]+1<dist[v][i] || dist[v][i]+1<dist[u][i]){
b=0;
}
}
if(!b)continue;
int ile=0;
for(int i=0; i<h; i++){
if(!czy[u][i] && dist[u][i]==dist[v][i]+1){
ile++;
}
if(!czy[v][i] && dist[v][i]==dist[u][i]+1){
ile++;
}
}
if(ile==t){
//cerr<<t<<"\n";
sum+=t;
if(2*(u-v)<=n)E2[v].pb(u);
else E2[u].pb(v);
for(int i=0; i<h; i++){
if(!czy[u][i] && dist[u][i]==dist[v][i]+1){
czy[u][i]=1;
}
if(!czy[v][i] && dist[v][i]==dist[u][i]+1){
czy[v][i]=1;
}
}
}
}
}
}
cerr<<sum<<"\n";
for(int i=0; i<n; i++){
sort(E2[i].begin(), E2[i].end());
if(spec[i])nadaj(ans, 111, 7);
else{
assert(E2[i].size()<100);
nadaj(ans, E2[i].size(), 7);
}
//assert(spec[i]==(E2[i].size()>=111));
if(spec[i]){
vi V(n);
for(int u:E2[i]){
V[u]=1;
}
for(int j:V){
nadaj(ans, j, 1);
}
}
else{
for(int u:E2[i]){
nadaj(ans, (u-i+n)%n, 9);
}
}
}
return;
}
void encode3(vi &ans, int n, int h, int m, int *v1, int *v2){
vi E[n];
int czy[n][h], dist[n][h];
for(int i=0; i<n; i++){
for(int j=0; j<h; j++){
czy[i][j]=(i==j);
dist[i][j]=1e9*(i!=j);
}
}
vi edg;
for(int i=0; i<m; i++){
E[v1[i]].pb(edg.size());
edg.pb(v2[i]);
E[v2[i]].pb(edg.size());
edg.pb(v1[i]);
}
for(int s=0; s<h; s++){
vi V;
V.pb(s);
for(int i=0; i<V.size(); i++){
int v=V[i];
for(int e:E[v]){
if(dist[edg[e]][s]==1e9){
dist[edg[e]][s]=dist[v][s]+1;
V.pb(edg[e]);
}
}
}
}
vi E2[n];
int sum=0;
for(int t=h; t>0; t--){
for(int u=0; u<n; u++){
for(int v=0; v<u; v++){
bool b=1;
for(int i=0; i<h; i++){
if(dist[u][i]+1<dist[v][i] || dist[v][i]+1<dist[u][i]){
b=0;
}
}
if(!b)continue;
int ile=0;
for(int i=0; i<h; i++){
if(!czy[u][i] && dist[u][i]==dist[v][i]+1){
ile++;
}
if(!czy[v][i] && dist[v][i]==dist[u][i]+1){
ile++;
}
}
if(ile==t){
//cerr<<t<<"\n";
sum+=t;
if(2*(u-v)<=n)E2[v].pb(u);
else E2[u].pb(v);
for(int i=0; i<h; i++){
if(!czy[u][i] && dist[u][i]==dist[v][i]+1){
czy[u][i]=1;
}
if(!czy[v][i] && dist[v][i]==dist[u][i]+1){
czy[v][i]=1;
}
}
}
}
}
}
cerr<<sum<<"\n";
for(int i=0; i<n; i++){
sort(E2[i].begin(), E2[i].end());
nadaj(ans, min((int)E2[i].size(), 111), 7);
if(E2[i].size()>=111){
vi V(n);
for(int u:E2[i]){
V[u]=1;
}
for(int j:V){
nadaj(ans, j, 1);
}
}
else{
for(int u:E2[i]){
nadaj(ans, (u-i+n)%n, 9);
}
}
}
return;
}
void encode(int n, int h, int m, int *v1, int *v2){
vi V, V2;
encode2(V, n, h, m, v1, v2);
encode3(V2, n, h, m, v1, v2);
if(V2.size()<V.size())V=V2;
for(int i:V){
encode_bit(i);
}
}
#include "grader.h"
#include "decoder.h"
#include<bits/stdc++.h>
#define st first
#define nd second
#define pb push_back
#define pp pop_back;
using namespace std;
using vi = vector<int>;
using pii = pair<int, int>;
int czytaj(int b){
int x=0;
for(int i=0; i<b; i++){
x+=(int(decode_bit())<<i);
}
return x;
}
void decode(int n, int h) {
vi E[n];
for(int i=0;i<n; i++){
int k=czytaj(7);
//cerr<<k<<"\n";
if(k>=100){
for(int j=0; j<n; j++){
if(czytaj(1)){
E[i].pb(j);
E[j].pb(i);
}
}
}
else{
while(k--){
int j=czytaj(9);
E[i].pb((j+i)%n);
E[(j+i)%n].pb(i);
}
}
}
for(int s=0; s<h; s++){
vi V;
V.pb(s);
vi dist(n, 1e9);
dist[s]=0;
for(int i=0; i<V.size(); i++){
int v=V[i];
for(int u:E[v]){
if(dist[u]==1e9){
dist[u]=dist[v]+1;
V.pb(u);
}
}
}
for(int i=0; i<n; i++){
hops(s, i, dist[i]);
}
}
}
Compilation message
encoder.cpp: In function 'void encode2(vi&, int, int, int, int*, int*)':
encoder.cpp:39:17: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
39 | for(int i=0; i<V.size(); i++){
| ~^~~~~~~~~
encoder.cpp: In function 'void encode3(vi&, int, int, int, int*, int*)':
encoder.cpp:184:17: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
184 | for(int i=0; i<V.size(); i++){
| ~^~~~~~~~~
decoder.cpp: In function 'void decode(int, int)':
decoder.cpp:44:17: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
44 | for(int i=0; i<V.size(); i++){
| ~^~~~~~~~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
6667 ms |
13840 KB |
Output is correct - 43000 call(s) of encode_bit() |
2 |
Correct |
2 ms |
4604 KB |
Output is correct - 98 call(s) of encode_bit() |
3 |
Correct |
1589 ms |
5864 KB |
Output is correct - 39510 call(s) of encode_bit() |
4 |
Correct |
1 ms |
4604 KB |
Output is correct - 116 call(s) of encode_bit() |
5 |
Correct |
2767 ms |
6840 KB |
Output is correct - 66717 call(s) of encode_bit() |
6 |
Correct |
3280 ms |
6920 KB |
Output is correct - 69064 call(s) of encode_bit() |
7 |
Incorrect |
3954 ms |
7628 KB |
too many decode_bit() calls |
8 |
Correct |
2886 ms |
5516 KB |
Output is correct - 21172 call(s) of encode_bit() |
9 |
Correct |
1659 ms |
5608 KB |
Output is correct - 21265 call(s) of encode_bit() |
10 |
Correct |
1652 ms |
5696 KB |
Output is correct - 21049 call(s) of encode_bit() |
11 |
Correct |
1873 ms |
6088 KB |
Output is correct - 35530 call(s) of encode_bit() |
12 |
Correct |
1913 ms |
5544 KB |
Output is correct - 15991 call(s) of encode_bit() |
13 |
Correct |
3658 ms |
7592 KB |
Output is correct - 56430 call(s) of encode_bit() |
14 |
Correct |
1949 ms |
5668 KB |
Output is correct - 24019 call(s) of encode_bit() |
15 |
Correct |
1920 ms |
5788 KB |
Output is correct - 25045 call(s) of encode_bit() |
16 |
Correct |
2473 ms |
6816 KB |
Output is correct - 33199 call(s) of encode_bit() |
17 |
Correct |
2408 ms |
6824 KB |
Output is correct - 33704 call(s) of encode_bit() |
18 |
Correct |
3251 ms |
7536 KB |
Output is correct - 57050 call(s) of encode_bit() |
19 |
Correct |
2865 ms |
6656 KB |
Output is correct - 48265 call(s) of encode_bit() |
20 |
Correct |
3444 ms |
7960 KB |
Output is correct - 53294 call(s) of encode_bit() |
21 |
Correct |
3836 ms |
8368 KB |
Output is correct - 58356 call(s) of encode_bit() |
22 |
Correct |
3992 ms |
7852 KB |
Output is partially correct - 73692 call(s) of encode_bit() |
23 |
Correct |
4156 ms |
8960 KB |
Output is correct - 66017 call(s) of encode_bit() |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
6667 ms |
13840 KB |
Output is correct - 43000 call(s) of encode_bit() |
2 |
Correct |
2 ms |
4604 KB |
Output is correct - 98 call(s) of encode_bit() |
3 |
Correct |
1589 ms |
5864 KB |
Output is correct - 39510 call(s) of encode_bit() |
4 |
Correct |
1 ms |
4604 KB |
Output is correct - 116 call(s) of encode_bit() |
5 |
Correct |
2767 ms |
6840 KB |
Output is correct - 66717 call(s) of encode_bit() |
6 |
Correct |
3280 ms |
6920 KB |
Output is correct - 69064 call(s) of encode_bit() |
7 |
Incorrect |
3954 ms |
7628 KB |
too many decode_bit() calls |
8 |
Correct |
2886 ms |
5516 KB |
Output is correct - 21172 call(s) of encode_bit() |
9 |
Correct |
1659 ms |
5608 KB |
Output is correct - 21265 call(s) of encode_bit() |
10 |
Correct |
1652 ms |
5696 KB |
Output is correct - 21049 call(s) of encode_bit() |
11 |
Correct |
1873 ms |
6088 KB |
Output is correct - 35530 call(s) of encode_bit() |
12 |
Correct |
1913 ms |
5544 KB |
Output is correct - 15991 call(s) of encode_bit() |
13 |
Correct |
3658 ms |
7592 KB |
Output is correct - 56430 call(s) of encode_bit() |
14 |
Correct |
1949 ms |
5668 KB |
Output is correct - 24019 call(s) of encode_bit() |
15 |
Correct |
1920 ms |
5788 KB |
Output is correct - 25045 call(s) of encode_bit() |
16 |
Correct |
2473 ms |
6816 KB |
Output is correct - 33199 call(s) of encode_bit() |
17 |
Correct |
2408 ms |
6824 KB |
Output is correct - 33704 call(s) of encode_bit() |
18 |
Correct |
3251 ms |
7536 KB |
Output is correct - 57050 call(s) of encode_bit() |
19 |
Correct |
2865 ms |
6656 KB |
Output is correct - 48265 call(s) of encode_bit() |
20 |
Correct |
3444 ms |
7960 KB |
Output is correct - 53294 call(s) of encode_bit() |
21 |
Correct |
3836 ms |
8368 KB |
Output is correct - 58356 call(s) of encode_bit() |
22 |
Correct |
3992 ms |
7852 KB |
Output is partially correct - 73692 call(s) of encode_bit() |
23 |
Correct |
4156 ms |
8960 KB |
Output is correct - 66017 call(s) of encode_bit() |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
6667 ms |
13840 KB |
Output is correct - 43000 call(s) of encode_bit() |
2 |
Correct |
2 ms |
4604 KB |
Output is correct - 98 call(s) of encode_bit() |
3 |
Correct |
1589 ms |
5864 KB |
Output is correct - 39510 call(s) of encode_bit() |
4 |
Correct |
1 ms |
4604 KB |
Output is correct - 116 call(s) of encode_bit() |
5 |
Correct |
2767 ms |
6840 KB |
Output is correct - 66717 call(s) of encode_bit() |
6 |
Correct |
3280 ms |
6920 KB |
Output is correct - 69064 call(s) of encode_bit() |
7 |
Incorrect |
3954 ms |
7628 KB |
too many decode_bit() calls |
8 |
Correct |
2886 ms |
5516 KB |
Output is correct - 21172 call(s) of encode_bit() |
9 |
Correct |
1659 ms |
5608 KB |
Output is correct - 21265 call(s) of encode_bit() |
10 |
Correct |
1652 ms |
5696 KB |
Output is correct - 21049 call(s) of encode_bit() |
11 |
Correct |
1873 ms |
6088 KB |
Output is correct - 35530 call(s) of encode_bit() |
12 |
Correct |
1913 ms |
5544 KB |
Output is correct - 15991 call(s) of encode_bit() |
13 |
Correct |
3658 ms |
7592 KB |
Output is correct - 56430 call(s) of encode_bit() |
14 |
Correct |
1949 ms |
5668 KB |
Output is correct - 24019 call(s) of encode_bit() |
15 |
Correct |
1920 ms |
5788 KB |
Output is correct - 25045 call(s) of encode_bit() |
16 |
Correct |
2473 ms |
6816 KB |
Output is correct - 33199 call(s) of encode_bit() |
17 |
Correct |
2408 ms |
6824 KB |
Output is correct - 33704 call(s) of encode_bit() |
18 |
Correct |
3251 ms |
7536 KB |
Output is correct - 57050 call(s) of encode_bit() |
19 |
Correct |
2865 ms |
6656 KB |
Output is correct - 48265 call(s) of encode_bit() |
20 |
Correct |
3444 ms |
7960 KB |
Output is correct - 53294 call(s) of encode_bit() |
21 |
Correct |
3836 ms |
8368 KB |
Output is correct - 58356 call(s) of encode_bit() |
22 |
Correct |
3992 ms |
7852 KB |
Output is partially correct - 73692 call(s) of encode_bit() |
23 |
Correct |
4156 ms |
8960 KB |
Output is correct - 66017 call(s) of encode_bit() |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
6667 ms |
13840 KB |
Output is correct - 43000 call(s) of encode_bit() |
2 |
Correct |
2 ms |
4604 KB |
Output is correct - 98 call(s) of encode_bit() |
3 |
Correct |
1589 ms |
5864 KB |
Output is correct - 39510 call(s) of encode_bit() |
4 |
Correct |
1 ms |
4604 KB |
Output is correct - 116 call(s) of encode_bit() |
5 |
Correct |
2767 ms |
6840 KB |
Output is correct - 66717 call(s) of encode_bit() |
6 |
Correct |
3280 ms |
6920 KB |
Output is correct - 69064 call(s) of encode_bit() |
7 |
Incorrect |
3954 ms |
7628 KB |
too many decode_bit() calls |
8 |
Correct |
2886 ms |
5516 KB |
Output is correct - 21172 call(s) of encode_bit() |
9 |
Correct |
1659 ms |
5608 KB |
Output is correct - 21265 call(s) of encode_bit() |
10 |
Correct |
1652 ms |
5696 KB |
Output is correct - 21049 call(s) of encode_bit() |
11 |
Correct |
1873 ms |
6088 KB |
Output is correct - 35530 call(s) of encode_bit() |
12 |
Correct |
1913 ms |
5544 KB |
Output is correct - 15991 call(s) of encode_bit() |
13 |
Correct |
3658 ms |
7592 KB |
Output is correct - 56430 call(s) of encode_bit() |
14 |
Correct |
1949 ms |
5668 KB |
Output is correct - 24019 call(s) of encode_bit() |
15 |
Correct |
1920 ms |
5788 KB |
Output is correct - 25045 call(s) of encode_bit() |
16 |
Correct |
2473 ms |
6816 KB |
Output is correct - 33199 call(s) of encode_bit() |
17 |
Correct |
2408 ms |
6824 KB |
Output is correct - 33704 call(s) of encode_bit() |
18 |
Correct |
3251 ms |
7536 KB |
Output is correct - 57050 call(s) of encode_bit() |
19 |
Correct |
2865 ms |
6656 KB |
Output is correct - 48265 call(s) of encode_bit() |
20 |
Correct |
3444 ms |
7960 KB |
Output is correct - 53294 call(s) of encode_bit() |
21 |
Correct |
3836 ms |
8368 KB |
Output is correct - 58356 call(s) of encode_bit() |
22 |
Correct |
3992 ms |
7852 KB |
Output is partially correct - 73692 call(s) of encode_bit() |
23 |
Correct |
4156 ms |
8960 KB |
Output is correct - 66017 call(s) of encode_bit() |