ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

【ybt高效进阶 21161】复杂问题(图论)(分类讨论)(MIM / Meet In Middle)

2021-09-30 17:35:27  阅读:210  来源: 互联网

标签:MIM 进阶 int ybt 然后 四元 100001 lnk 权值


复杂问题

题目链接:ybt高效进阶 21161

题目大意

给你一个无向图,我们定义一个四元环为一个四个点的集合,这四个点(u,v,x,y)构成的子图,存在边 (u,v),(v,x),(x,y),(y,u)。
然后点有点权,四元环的点权就是那四个点的点权之和。
然后要你求这个图所有四元环的点权和。

思路

在做这道题之前呢,先做做这道题:
无向图三元环计数
(下文默认会了上面的方法)

显然就是把三元环变成了四元环。
那我们考虑用一种叫做 MIM(Meet In Middle)的方法。
好吧其实就算是折半搜索的折半思想。

我们可以类似三元环的做法,然后看看怎么搞。
在这里插入图片描述

在这里插入图片描述
那它肯定就是这两种结构,我们分别叫它情况一和情况二。

对于情况一,我们可以先走两步,然后记录第三个点确定的时候,有多少种走法,中间点的权值和是多少。然后你再枚举第三个点(应该不能直接 \(1\sim n\)?没试过,但最好用一个数组装着有走法的点),然后再扩展一步,看一开始的点是否能到。(不难想到这样的复杂度是跟走两步一样的)

对于情况二,我们可以先预处理出走到最后的点的时候,中间有多少种走法,中间点权值和是否限购它那个。(其实就是跟情况一的走两步一样,可以一次处理出来)
然后我们就重新跑一次,然后就拿这次跑的路径和之前跑到这个终点的所有路径匹配权值。
(当然,要去掉两次都是走的这个中间点的路径,减一下就好了)

然后我们就能搞出答案啦。
(记得枚举每个出发点结束之后要清空个数啊,权值和啊那些的)

代码

#include<queue>
#include<cstdio>
#include<vector>
#define ll long long
#define mo 1000000007

using namespace std;

int n, m, x, y, du[100001];
vector <int> lnk[100001], son[100001];
queue <int> q;
ll ans, val[100001], nm[100001], a[100001];
int pl[100001];
bool in[100001];

int main() {
//	freopen("fz.in", "r", stdin);
//	freopen("fz.out", "w", stdout);
	
	scanf("%d %d", &n, &m);
	for (int i = 1; i <= n; i++) scanf("%lld", &a[i]);
	for (int i = 1; i <= m; i++) {
		scanf("%d %d", &x, &y);
		lnk[x].push_back(y); lnk[y].push_back(x);
	}
	
	for (int i = 1; i <= n; i++) {
		for (int j = 0; j < lnk[i].size(); j++)
			if (i < lnk[i][j]) {
				if (lnk[i].size() >= lnk[lnk[i][j]].size()) {
					son[i].push_back(lnk[i][j]); du[lnk[i][j]]++;
				}
				else son[lnk[i][j]].push_back(i), du[i]++;
			}
	}
	
	for (int i = 1; i <= n; i++) {
		if (!du[i]) q.push(i);
	}
	while (!q.empty()) {
		int now = q.front();
		q.pop();
		
		for (int i = 0; i < son[now].size(); i++) {//记录是否能到(情况1要用)
			du[son[now][i]]--;
			if (!du[son[now][i]]) q.push(son[now][i]);
			in[son[now][i]] = 1;
		}
		for (int i = 0; i < son[now].size(); i++) {//预处理走两步(MIM)
			for (int j = 0; j < son[son[now][i]].size(); j++) {
				val[son[son[now][i]][j]] += a[son[now][i]];
				val[son[son[now][i]][j]] %= mo;
				nm[son[son[now][i]][j]]++;
				if (nm[son[son[now][i]][j]] == 1)
					pl[++pl[0]] = son[son[now][i]][j];
			}
		}
		for (int i = 1; i <= pl[0]; i++) {//情况1
			int x = pl[i];
			for (int j = 0; j < son[x].size(); j++) {
				int y = son[x][j];
				if (in[y]) {
					ans += (val[x] + (a[now] + a[x] + a[y]) % mo * nm[x]) % mo;
					ans %= mo;
				}
			}
		}
		for (int i = 0; i < son[now].size(); i++) {//情况2
			for (int j = 0; j < son[son[now][i]].size(); j++) {
				ans += (val[son[son[now][i]][j]] + (a[son[now][i]] + a[now] + a[son[son[now][i]][j]]) * nm[son[son[now][i]][j]] % mo) % mo;
				ans -= (a[now] + a[son[son[now][i]][j]] + 2ll * a[son[now][i]]) % mo;
				ans = (ans % mo + mo) % mo; 
			}
		}
		for (int i = 0; i < son[now].size(); i++) {//清空
			in[son[now][i]] = 0;
			for (int j = 0; j < son[son[now][i]].size(); j++) {
				val[son[son[now][i]][j]] = nm[son[son[now][i]][j]] = 0;
			}
		}
		pl[0] = 0;
	}
	
	printf("%lld", ans);
	
	return 0;
}

标签:MIM,进阶,int,ybt,然后,四元,100001,lnk,权值
来源: https://www.cnblogs.com/Sakura-TJH/p/YBT_GXJJ_21161.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有