Submission #956564

#	Time	Username	Problem	Language	Result	Execution time	Memory
956564		arush_agu	Board (CEOI13_board)	C++17	100 / 100	3 ms	1360 KiB

board

#include <algorithm>
#include <cctype>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#include <functional>
#include <iomanip>
#include <iostream>
#include <iterator>
#include <list>
#include <map>
#include <numeric>
#include <queue>
#include <random>
#include <set>
#include <sstream>
#include <stack>
#include <string>
#include <unordered_map>
#include <unordered_set>
#include <vector>

#ifdef DEBUG
#include <time.h>
#endif

#define all(a) (a).begin(), (a).end()
#define rev(a) (a).rbegin(), (a).rend()
#define F first
#define S second
int recur_depth = 0;
#ifdef DEBUG
#define dbg(x)                                                                 \
  {                                                                            \
    ++recur_depth;                                                             \
    auto x_ = x;                                                               \
    --recur_depth;                                                             \
    cerr << string(recur_depth, '\t') << "\e[91m" << __func__ << ":"           \
         << __LINE__ << "\t" << #x << " = " << x_ << "\e[39m" << endl;         \
  }
#else
#define dbg(x)
#endif

using namespace std;
using namespace __gnu_pbds;

typedef pair<int, int> ii;

typedef long long ll;
typedef long double ld;
typedef pair<ll, ll> llll;

typedef vector<int> vi;
typedef vector<vi> vvi;
typedef vector<pair<int, int>> vii;
typedef vector<vii> vvii;

typedef vector<ll> vl;
typedef vector<vl> vvl;
typedef vector<pair<ll, ll>> vll;
typedef vector<vll> vvll;

typedef vector<bool> vb;

template <class type1>
using ordered_set = tree<type1, null_type, less<type1>, rb_tree_tag,
                         tree_order_statistics_node_update>;

template <typename A, typename B>
ostream &operator<<(ostream &os, const pair<A, B> &p) {
  return os << '(' << p.first << ", " << p.second << ')';
}
template <typename T_container, typename T = typename enable_if<
                                    !is_same<T_container, string>::value,
                                    typename T_container::value_type>::type>
ostream &operator<<(ostream &os, const T_container &v) {
  os << '{';
  string sep;
  for (const T &x : v)
    os << sep << x, sep = ", ";
  return os << '}';
}

const ll MOD = 1e9 + 7;
// const ll MOD = 998244353;
const ll INF = 1e9;
const ld EPS = 1e-9;

const int MAXN = 1e5 + 10;
char a_arr[MAXN], b_arr[MAXN];

int shorten(string &s, char ans[]) {
  vii st;
  for (int i = 0; i < s.size(); i++) {
    if (s[i] == '1') {
      if (st.empty() || st.back().first == 2)
        st.push_back({1, 1});
      else
        st.back().second++;
    } else if (s[i] == '2') {
      if (st.empty() || st.back().first == 1)
        st.push_back({2, 1});
      else
        st.back().second++;
    } else if (s[i] == 'U') {
      if (st.back().second == 1)
        st.pop_back();
      else
        st.back().second--;
    } else if (s[i] == 'L') {
      if (st.back().first == 2) {
        if (st.back().second == 1) {
          st.pop_back();
          if (st.empty())
            st.push_back({1, 1});
          else
            st.back().second++;
        } else
          st.back().second--, st.push_back({1, 1});
      } else {
        int len = st.back().second;
        st.pop_back();
        if (st.back().second == 1) {
          st.pop_back();
          if (st.empty())
            st.push_back({1, 1});
          else
            st.back().second++;
        } else
          st.back().second--, st.push_back({1, 1});

        st.push_back({2, len});
      }
    } else if (s[i] == 'R') {
      if (st.back().first == 1) {
        if (st.back().second == 1) {
          st.pop_back();
          if (st.empty())
            st.push_back({2, 1});
          else
            st.back().second++;
        } else
          st.back().second--, st.push_back({2, 1});
      } else {
        int len = st.back().second;
        st.pop_back();
        if (st.back().second == 1) {
          st.pop_back();
          if (st.empty())
            st.push_back({2, 1});
          else
            st.back().second++;
        } else
          st.back().second--, st.push_back({2, 1});

        st.push_back({1, len});
      }
    }
  }

  int idx = 0;
  for (int i = 0; i < st.size(); i++) {
    auto [ch, cnt] = st[i];
    for (int j = 0; j < cnt; j++)
      ans[idx++] = ch;
  }
  return idx;
}

void solve() {
  string a, b;
  cin >> a >> b;

  int depth_a = shorten(a, a_arr);
  int depth_b = shorten(b, b_arr);

  if (depth_a == depth_b) {
    bool flag = 1;
    for (int i = 0; i < depth_a; i++)
      if (a_arr[i] != b_arr[i]) {
        flag = 0;
        break;
      }
    if (flag) {
      cout << "0\n";
      return;
    }
  }

  int flip = depth_a > depth_b ? -1 : 1;
  for (int i = 0; i < min(depth_a, depth_b); i++)
    if (a_arr[i] != b_arr[i]) {
      flip = a_arr[i] == 2 ? -1 : 1;
      break;
    }

  int best = INF;
  int hor_dist = 0;
  for (int depth = 0; depth <= min(depth_a, depth_b); depth++) {
    int curr = depth_a - depth + depth_b - depth + hor_dist;
    best = min(best, curr);

    hor_dist *= 2;
    if (a_arr[depth] != b_arr[depth])
      hor_dist += a_arr[depth] == 1 ? flip : -flip;
    if (hor_dist > 1e6 + 10)
      break;
  }

  cout << best << "\n";
}

int main() {
  ios_base::sync_with_stdio(0);
  cin.tie(NULL);

  clock_t start = clock();

  int test_cases = 1;
  // cin >> test_cases;

  while (test_cases--)
    solve();

#ifdef DEBUG
  cerr << fixed << setprecision(10)
       << "\nTime Taken: " << (double)(clock() - start) / CLOCKS_PER_SEC
       << "s\n";
#endif
  return 0;
}

Compilation message (stderr)

board.cpp: In function 'int shorten(std::string&, char*)':
board.cpp:97:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::__cxx11::basic_string<char>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   97 |   for (int i = 0; i < s.size(); i++) {
      |                   ~~^~~~~~~~~~
board.cpp:165:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
  165 |   for (int i = 0; i < st.size(); i++) {
      |                   ~~^~~~~~~~~~~
board.cpp: In function 'int main()':
board.cpp:220:11: warning: unused variable 'start' [-Wunused-variable]
  220 |   clock_t start = clock();
      |           ^~~~~

• Subtask 1: 10 / 10
• Subtask 2: 10 / 10
• Subtask 3: 10 / 10
• Subtask 4: 10 / 10
• Subtask 5: 10 / 10
• Subtask 6: 10 / 10
• Subtask 7: 10 / 10
• Subtask 8: 10 / 10
• Subtask 9: 10 / 10
• Subtask 10: 10 / 10