제출 #152886

#제출 시각아이디문제언어결과실행 시간메모리
1528862qbingxuan함수컵 박물관 (FXCUP4_museum)C++17
0 / 100
2 ms376 KiB
#include <algorithm>
#include <vector>

using namespace std;
typedef int64_t ll;

int64_t check2(vector<int> &A,vector<int> &B) {
    int n = A.size();
    vector<int> v(n);
    ll res = 0;
    for(int i = 0; i < n; i++) v[i] = A[i]*200+B[i];
    sort(v.begin(),v.end());
    for(int i = 0,j; i < n; i = j) {
        for(j = i; j < n; j++) if(v[i]!=v[j]) break;
        res += (j-i)*(j-i-1)/2;
    }
    return res;
}
int64_t check3(vector<int> &A,vector<int> &B,vector<int> &C) {
    int n = A.size();
    vector<int> v(n);
    ll res = 0;
    for(int i = 0; i < n; i++) v[i] = A[i]*40000+B[i]*200+C[i];
    sort(v.begin(),v.end());
    for(int i = 0,j; i < n; i = j) {
        for(j = i; j < n; j++) if(v[i]!=v[j]) break;
        res += (j-i)*(j-i-1)/2;
    }
    return res;
}

int64_t CountSimilarPairs(vector<int> A,vector<int> B,vector<int> C) {
    int n = A.size();
    ll res = 0;
    // AB, BC, CA, 2ABC
    res += check2(A,B);
    res += check2(B,C);
    res += check2(C,A);
    res -= check3(A,B,C)*2;
    return res;
}

컴파일 시 표준 에러 (stderr) 메시지

museum.cpp: In function 'int64_t CountSimilarPairs(std::vector<int>, std::vector<int>, std::vector<int>)':
museum.cpp:33:9: warning: unused variable 'n' [-Wunused-variable]
     int n = A.size();
         ^
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...