题意:
给定n个点m条边的无向图,每个点有点权
对于一个三元环,设三个点的点权为x,y,z,那么贡献为max(x,y,z),
要求计算所有三元环的贡献和。
数据范围:n<=1e5,m<=3e5
解法:
用三元环统计算法直接枚举图中所有三元环统计答案就行了.
视n,m同阶,枚举三元环算法的复杂度为O(n*sqrt(n))
code:
#include<bits/stdc++.h>
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;
}