Submission #652067

#	Time	Username	Problem	Language	Result	Execution time	Memory
652067		TimDee	Flight to the Ford (BOI22_communication)	C++17	0 / 100	869 ms	24132 KiB

communication

#include"communication.h"
#pragma GCC optimize("O3")
#pragma GCC optimize("Ofast")
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
#define forn(i,n) for (int i=0; i<n; ++i)

int V[3000000]; int ptr=0; int asz=0;
void brut(int x, int n, int v=0, int i=0, int l=0) {
    if (i>=n) {V[ptr++]=v; return;}
    int b=(x>>i)&1;
    brut(x,n,v+(b<<i),i+1,0);
    if (l==0) brut(x,n,v+((!b)<<i),i+1,1);
}

vector<int> get_possible(int n) {
    ll x=0, q=0;
    forn(i,29) {
        q=receive();
        x+=q<<i;
    }
    brut(x,29);
    vector<int> b;
    for (int j=0; j<ptr; ++j) {
        int x=V[j];
        if (x>=n) continue;
        for (ll i=0; i+x<n; i+=1ll<<29) {
            b.push_back(x);
        }
    }
    sort(b.begin(), b.end());
    ptr=0;
    return b;
}

vector<int> dequery(vector<int>&a,int bit) {
    int x=0, q;
    forn(i,bit) x+=receive()<<i;
    brut(x,bit);

    vector<int> b;
    for (int j=0; j<ptr; ++j) {
        int x=V[j];
        if (x>=a.size()) continue;
        for (int i=0; i+x<a.size(); i+=(1<<bit)) {
            b.push_back(a[i+x]);
        }
    }
    sort(b.begin(), b.end());
    ptr=0;
    return b;

}

pair<int,int> decode(int n) {
    
    if (n>3) return{1,1};
    vector<int> a=get_possible(n); //for (auto x:a) cout<<x<<' '; cout<<'\n';
    a=dequery(a,21);
    a=dequery(a,15);
    a=dequery(a,11);
    a=dequery(a,8);
    a=dequery(a,6);
    a=dequery(a,4);
    a=dequery(a,3);

    a=dequery(a,4); //for (auto x:a) cout<<x<<' '; cout<<'\n';
    a=dequery(a,3); //for (auto x:a) cout<<x<<' '; cout<<'\n';
    a=dequery(a,2); //for (auto x:a) cout<<x<<' '; cout<<'\n';
    a=dequery(a,2); //for (auto x:a) cout<<x<<' '; cout<<'\n';

    if (a.size()==1) return{a[0]+1,a[0]+1};
    if (a.size()==2) return{a[0]+1,a[1]+1};
    assert(a.size()==3);

    int q,x=0;
    q=receive();
    x+=q<<0;
    q=receive();
    x+=q<<1;
    q=receive();
    x+=q<<2;
    q=receive();
    x+=q<<3;

    vector<int> ans;
    vector<int> A={0,1,5,10,9,8,2,4};
    vector<int> B={1,8,9,11,13,3,12,0};
    vector<int> C={6,15,2,4,12,3,7,14};
    for (auto v:A) if (v==x) ans.push_back(a[0]);
    for (auto v:B) if (v==x) ans.push_back(a[2]);
    for (auto v:C) if (v==x) ans.push_back(a[1]);

    if (ans.size()==1) return {ans[0]+1,ans[0]+1};
    return {ans[0]+1,ans[1]+1};

}

int Send(int x, int bit) {
    ll ret=0;
    forn(i,bit) ret+=send((x>>i)&1)<<i;
    return ret;
}

int a[3000000], b[3000000], bsz=0;
void get_possible(int n, int x) {
    x=Send(x,29);
	brut(x,29);
    return;
    int cnt=0;
	for (int j=0; j<ptr; ++j) {
        int x=V[j];
        if (x>=n) continue;
        for (ll i=0; i+x<n; i+=1ll<<29) {
            b[bsz++]=i+x;
        }
    }
	ptr=0;
    asz=0;
    sort(b,b+bsz);
	forn(i,bsz) a[asz++]=b[i];
    bsz=0;
}

void query(int x,int bit) {
	int r=0; while (a[r]!=x) ++r;
    r=Send(r,bit);
	brut(r,bit);
	for (int j=0; j<ptr; ++j) {
        int x=V[j];
		if (x>=asz) continue;
		for (int i=0; i+x<asz; i+=(1<<bit)) {
			b[bsz++]=a[i+x];
		}
	}
    sort(b,b+bsz);
	ptr=0;
    asz=0;
	forn(i,bsz) a[asz++]=b[i];
    bsz=0;

}

void encode(int n, int x) {

	--x;
    get_possible(n,x); //for (auto x:a) cout<<x<<' '; cout<<'\n';
    if (n>3) return;
    query(x,21);
    query(x,15);
    query(x,11);
    query(x,8);
    query(x,6);
    query(x,4);
    query(x,3);

    query(x,4); //for (auto x:a) cout<<x<<' '; cout<<'\n';
    query(x,3); //for (auto x:a) cout<<x<<' '; cout<<'\n';
    query(x,2); //for (auto x:a) cout<<x<<' '; cout<<'\n';
    query(x,2); //for (auto x:a) cout<<x<<' '; cout<<'\n';

    if (asz<3) return;

    //assert(asz==3);
    int v=0,b=0;;
    if (x==a[0]) {
        b=send(0);
        v+=b<<0;
        b=send(0);
        v+=b<<1;
        b=send(0);
        v+=b<<2;
        b=send(0);
        v+=b<<3;
    } else if (x==a[1]) {
        b=send(0);
        v+=b<<0;
        b=send(1);
        v+=b<<1;
        b=send(1);
        v+=b<<2;
        b=send(0);
        v+=b<<3;
    } else {
        b=send(1);
        v+=b<<0;
        b=send(0);
        v+=b<<1;
        b=send(0);
        v+=b<<2;
        b=send(1);
        v+=b<<3;
    }
    
}

Compilation message (stderr)

communication.cpp: In function 'std::vector<int> dequery(std::vector<int>&, int)':
communication.cpp:45:14: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   45 |         if (x>=a.size()) continue;
      |             ~^~~~~~~~~~
communication.cpp:46:26: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   46 |         for (int i=0; i+x<a.size(); i+=(1<<bit)) {
      |                       ~~~^~~~~~~~~
communication.cpp:38:14: warning: unused variable 'q' [-Wunused-variable]
   38 |     int x=0, q;
      |              ^
communication.cpp: In function 'void get_possible(int, int)':
communication.cpp:111:9: warning: unused variable 'cnt' [-Wunused-variable]
  111 |     int cnt=0;
      |         ^~~

• Subtask 1: 0 / 15
• Subtask 2: 0 / 85