oj.uz

답안 #551144

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
551144 2022-04-19T23:01:08 Z Olympia XOR Sum (info1cup17_xorsum) C++17
56 / 100
 1600 ms 12328 KB 
#include <iostream>
#include <vector>
#include <iomanip>
#include <algorithm>
#include <cassert>
#include <map>
#include <complex>
#include <cmath>
#include <set>
#include <queue>
using namespace std;

#pragma GCC target ("avx2")
#pragma GCC optimization ("O3")
#pragma GCC optimization ("unroll-loops")
int64_t res (vector<int>& a, int x) {
    //how many are <= x
    int64_t r = a.size() - 1;
    int64_t ans = 0;
    for (int i = 0; i < a.size(); i++) {
        if (a[i] + a[0] > x) {
            break;
        }
        while (a[r] + a[i] > x) {
            r--;
        }
        ans += abs(r) + 1 - (r >= i);
    }
    ans /= 2;
    for (int i = 0; i < a.size(); i++) {
        ans += (2 * a[i] <= x);
    }
    return ans;
}
int64_t comp (vector<int>&a, vector<int>&b, int x) {
    //ans += res(a, x);
    //ans += res(b, x);
    int64_t r = b.size() - 1;
    int64_t ans = res(a, x) + res(b, x);
    for (int i = 0; i < a.size(); i++) {
        if (a[i] + b[0] > x) {
            break;
        }
        while (b[r] + a[i] > x) {
            r--;
        }
        ans += abs(r) + 1;
    }
    return ans;
}
int64_t comp (vector<int>&a, vector<int>&b, int l, int r) {
    return comp(a, b, r) - comp(a, b, l - 1);
}
int main () {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    int n; cin >> n;
    vector<int> v(n);
    for (int i = 0; i < n; i++) {
        cin >> v[i];
    }
    sort(v.begin(), v.end());
    int64_t ans = 0;
    for (int b = 30; b >= 0; b--) {
        vector<int> myVector;
        vector<int> v1;
        sort(v.begin(), v.end());
        for (int i = 0; i < v.size(); i++) {
            if (!(v[i] & (1 << b))) myVector.push_back(v[i] & ((1 << (b + 1)) - 1));
            else v1.push_back(v[i] & ((1 << (b + 1)) - 1));
            v[i] = v[i] & (1 << (b + 1)) - 1;
        }
        sort(myVector.begin(), myVector.end());
        sort(v1.begin(), v1.end());
        int64_t a = 0;
        a += comp(v1, myVector, (1 << b), (1 << (b + 1)) - 1);
        a += comp(v1, myVector, (1 << b) + (1 << (b + 1)), (1 << (b + 2)) - 1);
        ans += ((int64_t)1 << b) * (a % 2);
    }
    cout << ans;
}

Compilation message

xorsum.cpp:14: warning: ignoring '#pragma GCC optimization' [-Wunknown-pragmas]
   14 | #pragma GCC optimization ("O3")
      | 
xorsum.cpp:15: warning: ignoring '#pragma GCC optimization' [-Wunknown-pragmas]
   15 | #pragma GCC optimization ("unroll-loops")
      | 
xorsum.cpp: In function 'int64_t res(std::vector<int>&, int)':
xorsum.cpp:20:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   20 |     for (int i = 0; i < a.size(); i++) {
      |                     ~~^~~~~~~~~~
xorsum.cpp:30:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   30 |     for (int i = 0; i < a.size(); i++) {
      |                     ~~^~~~~~~~~~
xorsum.cpp: In function 'int64_t comp(std::vector<int>&, std::vector<int>&, int)':
xorsum.cpp:40:23: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   40 |     for (int i = 0; i < a.size(); i++) {
      |                     ~~^~~~~~~~~~
xorsum.cpp: In function 'int main()':
xorsum.cpp:68:27: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   68 |         for (int i = 0; i < v.size(); i++) {
      |                         ~~^~~~~~~~~~
xorsum.cpp:71:42: warning: suggest parentheses around '-' in operand of '&' [-Wparentheses]
   71 |             v[i] = v[i] & (1 << (b + 1)) - 1;
      |                           ~~~~~~~~~~~~~~~^~~

Subtask #1
7.0 / 7.0

# 결과 실행 시간 메모리 Grader output
1 Correct 11 ms 340 KB Output is correct
2 Correct 10 ms 372 KB Output is correct

Subtask #2
11.0 / 11.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1578 ms 12328 KB Output is correct
2 Correct 1468 ms 11824 KB Output is correct

Subtask #3
0 / 21.0

# 결과 실행 시간 메모리 Grader output
1 Correct 1578 ms 12328 KB Output is correct
2 Correct 1468 ms 11824 KB Output is correct
3 Execution timed out 1687 ms 12300 KB Time limit exceeded
4 Halted 0 ms 0 KB -

Subtask #4
38.0 / 38.0

# 결과 실행 시간 메모리 Grader output
1 Correct 11 ms 340 KB Output is correct
2 Correct 10 ms 372 KB Output is correct
3 Correct 306 ms 1616 KB Output is correct
4 Correct 315 ms 1608 KB Output is correct

Subtask #5
0 / 23.0

# 결과 실행 시간 메모리 Grader output
1 Correct 11 ms 340 KB Output is correct
2 Correct 10 ms 372 KB Output is correct
3 Correct 1578 ms 12328 KB Output is correct
4 Correct 1468 ms 11824 KB Output is correct
5 Execution timed out 1687 ms 12300 KB Time limit exceeded
6 Halted 0 ms 0 KB -