bzoj 3498: PA2009 Cakes【瞎搞】

参考:https://www.cnblogs.com/spfa/p/7495438.html
为什么邻接表会TTTTTTTLE啊...只能用vector?
把点按照点权从大到小排序,把无向边变成排名靠前的点连向排名靠后的点的有向边并记录出度d[u],用map记录一下联通,这样可以避免重复计算
按排名遍历,设遍历到u点,扫u的邻接点v,如果v的出度小于根号m,就遍历v的所有邻接点,判断是否与u相连即可;否则,再次枚举u的邻接点v,用map判断当前v是否与新扫描的v相连;
每次判断成功都在ans里加上u的点权,因为边总是从点权大的点连向点权小的点,所以判断出的三元环的价值总是当前u的点权
至于为什么复杂度是对的.......O(玄学)吧......

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<map>
#include<cmath>
#include<vector>
using namespace std;
const int N=100005;
int n,m,d[N],cnt,h[N],lk[N],rk[N];
map<int,int>mp[N];
vector<int>g[N];
struct qwe
{
	int ne,to;
}e[N<<2];
struct dian
{
	int id,v;
}a[N];
bool cmp(const dian &a,const dian &b)
{
	return a.v>b.v;
}
int read()
{
	int r=0,f=1;
	char p=getchar();
	while(p>'9'||p<'0')
	{
		if(p=='-')
			f=-1;
		p=getchar();
	}
	while(p>='0'&&p<='9')
	{
		r=r*10+p-48;
		p=getchar();
	}
	return r*f;
}
void add(int u,int v)
{
	d[u]++;
	mp[u][v]=1;
	g[u].push_back(v);
}
int main()
{
	n=read(),m=read();
	for(int i=1;i<=n;i++)
		a[i].v=read(),a[i].id=i;
	sort(a+1,a+1+n,cmp);
	for(int i=1;i<=n;i++)
		rk[a[i].id]=i;
	for(int i=1;i<=m;i++)
	{
		int x=read(),y=read();
		if(rk[x]<rk[y])
			add(x,y);
		else
			add(y,x);
	}
	long long ans=0;
	int bs=sqrt(m)+1;
	for(int k=1;k<=n;k++)
	{
		int u=a[k].id,v=a[k].v;;
		for(int i=0,len=g[u].size();i<len;i++)
			lk[g[u][i]]=u;
		for(int i=0,len=g[u].size();i<len;i++)
		{
			int y=g[u][i];
			if(d[y]>bs)
			{
				for(int j=0;j<len;j++)
					if(mp[y][g[u][j]])
						ans+=v;
			}
			else
			{
				for(int j=0,le=g[y].size();j<le;j++)
					if(lk[g[y][j]]==u)
						ans+=v;
			}
		}
	}
	printf("%lld
",ans);
	return 0;
}
原文地址:https://www.cnblogs.com/lokiii/p/8953077.html