(200627) 코드포스

Codeforces materials usage license (v. 0.1) 에 따라 codeforces.com 의 문제임을 알립니다.

Original problem link: Problem - 1368D - Codeforces


Example

Input

1
123

Output

15129

Input

3
1 3 5

Output

51

Input

2
349525 699050

Output

1099509530625


Common codes for Codeforces
#include <algorithm>
#include <array>
#include <cmath>
#include <functional> 
#include <iostream>
#include <queue>
#include <string>
#include <vector>

namespace predefined {

#ifdef CODEFORCES_DEBUG
  constexpr bool kIsDebug = true;
#else
  constexpr bool kIsDebug = false;
#endif

enum class LogType {
  INFO,
  DEBUG
};

template <LogType type, bool... Bools>
struct Log;

template <>
struct Log<LogType::INFO> {
  Log() {}
  ~Log() { std::cout << '\n'; }

  template <typename T>
  Log& operator<< (const T &t) {
    std::cout << t;
    return *this;
  }
};

template <>
struct Log<LogType::DEBUG, true>{
  Log() {}
  ~Log() { std::cout << '\n'; }

  void Open() {
    freopen("input.txt", "r", stdin);
  }

  template <typename T>
  Log& operator<< (const T &t) {
    std::cout << t;
    return *this;
  }
};

template <>
struct Log<LogType::DEBUG, false> {
  Log() {}
  ~Log() {}

  void Open() {}

  template <typename T>
  Log& operator<< (const T &t) { return *this; }
};

template<typename T>
struct BoolMap {
  explicit BoolMap(const T &_true_value, const T &_false_value) 
    : true_value(_true_value),
      false_value(_false_value) {}

  const T true_value;
  const T false_value;

  const T inline GetValue(const bool bool_key) const {
    return bool_key ? true_value : false_value;  
  };
};

}  // predefined;

#define LOG_INFO predefined::Log<predefined::LogType::INFO>()
#define LOG_DEBUG predefined::Log<predefined::LogType::DEBUG, predefined::kIsDebug>()
#define LOG(LEVEL) LOG_##LEVEL
#define LABEL(variable_name) #variable_name << ": " << variable_name

#define INIT_ANSWER(type, true_value, false_value) \
  const predefined::BoolMap<type> predefined_bool_map(true_value, false_value)
#define INIT_STRING_ANSWER(true_string, false_string) \
  INIT_ANSWER(std::string, true_string, false_string)
#define GET_ANSWER(bool_key) \
  predefined_bool_map.GetValue(bool_key)

#define INIT_CODEFORCES() \
  std::ios::sync_with_stdio(false); \
  std::cin.tie(NULL); \
  LOG(DEBUG).Open(); \

1368C
namespace {
using namespace std;

constexpr size_t kMaxIndices = 20;
}  // unnamed namespace;


int main(int argc, char** argv) {
  INIT_CODEFORCES();

  int n;
  cin >> n;

  vector<int64_t> exp2seqs({1});
  for (int i = 1; i < kMaxIndices; ++i) exp2seqs.emplace_back(exp2seqs[i - 1] * 2);
  vector<int> bit_counts(kMaxIndices, 0);

  LOG(DEBUG) << LABEL(exp2seqs[0]) << LABEL(exp2seqs[1]) << LABEL(exp2seqs[2]);
  
  while (n--) {
    int64_t input;
    cin >> input;
    for (int i = 0; input; ++i) {
      if (exp2seqs[i] & input) {
        input -= exp2seqs[i];
        ++bit_counts[i];
      }
    }
  }

  int64_t ans = 0;

  bool is_not_empty = true;

  while (is_not_empty) {
    is_not_empty = false;

    int64_t temp = 0;
    for (size_t i = 0; i < kMaxIndices; ++i) {
      if (bit_counts[i]-- > 0) {
        is_not_empty = true;
        temp |= exp2seqs[i];
      }
    }

    ans += temp * temp;
  }

  LOG(INFO) << ans;

  return 0;
}