bzoj3498 PA2009 Cakes（三元环统计）

题意：

给定n个点m条边的无向图，每个点有点权
对于一个三元环，设三个点的点权为x,y,z，那么贡献为max(x,y,z)，
要求计算所有三元环的贡献和。

数据范围：n<=1e5，m<=3e5

解法：

用三元环统计算法直接枚举图中所有三元环统计答案就行了.
视n,m同阶,枚举三元环算法的复杂度为O(n*sqrt(n))

code：

#include
using namespace std;
#define int long long
const int maxm=3e5+5;
vector<int>g[maxm];
int u[maxm],v[maxm];
int mark[maxm];
int d[maxm];
int a[maxm];
int n,m;
signed main(){
    cin>>n>>m;
    for(int i=1;i<=n;i++)cin>>a[i];
    for(int i=1;i<=m;i++){
        cin>>u[i]>>v[i];
        d[u[i]]++,d[v[i]]++;
    }
    for(int i=1;i<=m;i++){
        int x=u[i],y=v[i];
        if(d[x]>d[y]||(d[x]==d[y]&&x>y)){
            g[x].push_back(y);
        }else{
            g[y].push_back(x);
        }
    }
    int ans=0;
    for(int x=1;x<=n;x++){
        for(int y:g[x])mark[y]=1;
        for(int y:g[x]){
            for(int z:g[y]){
                if(mark[z]){
                    ans+=max(a[x],max(a[y],a[z]));
                }
            }
        }
        for(int y:g[x])mark[y]=0;//clear
    }
    cout<<ans<<endl;
    return 0;
}

---