bzoj 3498

统计三元环

很多代码在bzoj都T诶

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <string>

#define gc getchar()
inline int read() {int x = , f = ; char c = gc; while(c < '' || c > '') {if(c == '-') f = -; c = gc;}
    while(c >= '' && c <= '') x = x *  + c - '', c = gc; return x;}
#undef gc

using namespace std;
const int N = 1e5 + , M = 2e5 + 5e4 + ;

int n, m;
int A[M], B[M], W[N];
int du[N];
int vis[N];
int cnt, head[N];
struct Node {int v, nxt;} G[M];

inline void Add(int u, int v) {G[++ cnt].v = v, G[cnt].nxt = head[u], head[u] = cnt;}
inline int Max(int a, int b) {if(a > b) return a; return b;}

int main() {
    n = read(), m = read();
    for(int i = ; i <= n; i ++) W[i] = read();
    for(int i = ; i <= m; i ++) {
        A[i] = read(), B[i] = read();
        du[A[i]] ++, du[B[i]] ++;
    }
    for(int i = ; i <= n; i ++) head[i] = -;
    for(int i = ; i <= m; i ++) {
        if(du[A[i]] > du[B[i]] || (du[A[i]] == du[B[i]] && A[i] > B[i])) swap(A[i], B[i]);
        Add(A[i], B[i]);
    }
    long long Answer = ;
    for(int k = ; k <= m; k ++) {
        for(int i = head[A[k]]; ~ i; i = G[i].nxt) {
            vis[G[i].v] = k;
        }
        for(int i = head[B[k]]; ~ i; i = G[i].nxt) {
            if(G[i].v != A[k] && vis[G[i].v] == k) {
                Answer += Max(Max(W[A[k]], W[B[k]]), W[G[i].v]);
            }
        }
    }
    printf("%lld", Answer);
    return ;
}