Submission #72180535

---

Source Code Expand

#include <algorithm>
#include <array>
#include <bitset>
#include <cassert>
#include <chrono>
#include <climits>
#include <cmath>
#include <complex>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <deque>
#include <exception>
#include <fstream>
#include <functional>
#include <iomanip>
#include <iostream>
#include <iterator>
#include <limits>
#include <list>
#include <map>
#include <memory>
#include <new>
#include <numeric>
#include <queue>
#include <set>
#include <sstream>
#include <stack>
#include <stdexcept>
#include <streambuf>
#include <string>
#include <tuple>
#include <typeinfo>
#include <unordered_map>
#include <unordered_set>
#include <utility>
#include <vector>
using ll = long long;
using ld = long double;
using namespace std;
#define endl "\n"
#define ff first
#define ss second
#define forn(i,n) for(int i=0;i<n;i++)
#define dbgv(v) cout<<#v<<" "<<v<<endl
#define dbga(a,n) forn(i,n-1) {cout<<a[i]<<' ';} cout<<a[n-1]<<'\n';
#define all(v) (v).begin(), (v).end()
const ll N = 1e6 + 5, mod = 1e9 + 7;
ll fact[N], modinv[N];
ll fastpow(ll n, ll m) {
    int ret = 1;
    while (m) {
        if (m & 1)
            (ret *= n) %= mod;
        (n *= n) %= mod;
        m /= 2;
    }
    return ret;
}
void pre() {
    fact[0] = 1;
    for (int i = 1; i < N; i++)
        fact[i] = i * fact[i - 1] % mod;
    modinv[N - 1] = fastpow(fact[N - 1], mod - 2);
    for (int i = N - 2; i >= 0; i--)
        modinv[i] = (i + 1) * modinv[i + 1] % mod;
}
ll ncr(ll n, ll r) {
    return fact[n] * modinv[n - r] % mod * modinv[r] % mod;
}
void dfs(int i,int p, vector<vector<int>>&adj){
	for(auto c:adj[i]){
		dfs(c,i,adj);
	}
}
void solve(){
    int n;
    cin>>n;
    vector<ll> a(n);
    map<ll,vector<int>> p7,p5,p3;
    forn(i,n){
        cin>>a[i];
        if(a[i]%7==0) p7[a[i]/7].push_back(i);
        if(a[i]%5==0) p5[a[i]/5].push_back(i);
        if(a[i]%3==0) p3[a[i]/3].push_back(i);
    }
    ll ans=0;
    for(auto&[g,js]:p5){
        if(!p7.count(g)||!p3.count(g)) continue;
        auto&is=p7[g];
        auto&ks=p3[g];
        ll ci=is.size(),ck=ks.size();
        for(int j:js){
            ll il=lower_bound(all(is),j)-is.begin();
            ll ir=ci-il;
            ll kl=lower_bound(all(ks),j)-ks.begin();
            ll kr=ck-kl;
            ans+=il*kl+ir*kr;
        }
    }
    cout<<ans<<endl;
}
int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    int T = 1;
    while(T--){
        solve();
    }
    return 0;
}

Submission Info

Compile Error

./Main.cpp: In function 'void dfs(int, int, std::vector<std::vector<int> >&)':
./Main.cpp:71:20: warning: unused parameter 'p' [-Wunused-parameter]
   71 | void dfs(int i,int p, vector<vector<int>>&adj){
      |                ~~~~^

Judge Result