枚举题的做法思路总结

本文主要介绍算法题里面枚举题的做法思路，希望每天都能进步

---

`

一、什么是枚举以及枚举的优化策略？

枚举就是说通过一一列举每一种答案的可能情况，每一次枚举的时候判断当前的答案是否符合条件。
首先枚举需要考虑的是顺序，然后是当会超时的时候如何优化
枚举的优化策略：
1.空间换时间（哈希表）时间复杂度O(1)
原理就是说在枚举的时候可以先把过程中的一些结果存起来，然后再在后面枚举过程中依次查找这些值是否在前面出现过，用哈希表查找效率比较高，C++STL模板库里面unordered_map头文件里面帮助我们实现了哈希表，因此不用手动实现感兴趣的可以去看有关哈希表如何实现的文章，这里不再赘述。

2.二分（两种做法)时间复杂度O(logn)
第一种做法是利用algorithm头文件里面的lower_bound和upper_bound函数。两个函数的用法可以去看这篇博客，但是需要注意的是两个函数只能用于非递减数列。细节决定成败！！
第二种做法是自己手动实现二分，手动实现二分可以看我之前发布的一篇博客。

二、例题及思路

这里选自蓝桥杯的两道例题来进行总结

1.四平方和

本题首先想到暴力，思路就是依次枚举a,b,c,d，如果满足等式就输出，那么如何优化?这里由于是等式，可以考虑采用空间换时间来进行优化

#include<iostream>
#include<unordered_map>
using namespace std;
typedef pair<int,int> PII;
unordered_map<int,PII> S;//定义哈希表
int n;
int main()you
{
	cin>>n;
	//首先枚举c和d，把所有的c*c+d*d都存起来，后面查找的时候再用
	for(int c=0;c<=n;c++)
	{
		for(int d=c;d*d+c*c<=n;d++)
		{
			int t=c*c+d*d;
			if(S.count(t)==0) S[t]={c,d}; //如果哈希表里面没有该数对，那就存起来
		}
	}
	//枚举a和b，看一下跟正确答案还差多少，如果这个差在之前存在于哈希表中，那么就是答案
	for(int a=0;a<=n;a++)
	{
		for(int b=0;b*b+a*a<=n;b++)
		{
			int t=n-a*a-b*b;
			if(S.count(t))
			{
				cout<<a<<" "<<b<<" "<<S[t].first<<" "<<S[t].second;
				return 0; 
			}
		}
	}
	
	return 0;
 } 

2.递增三元组

题意要求我们求出符合条件的递增三元组的个数，因此纯暴力做法就是说三重循环依次枚举每一个数组里面的元素，但是这样的做法时间复杂度是O(n^3)，数据范围是1e5,因此一定会超时，所以考虑优化。
由于本题有较为明显的顺序，即a[i]<b[j]<c[k],可以考虑采用二分。
首先二分需要数组一定要是有序的，所以需要预处理数组。

#include<iostream>
#include<algorithm>
using namespace std;
typedef long long LL;
const int N=100010;
int num[3][N];
int n;
int main()
{
	cin>>n;
    for(int i=0;i<3;i++)
    {
    	for(int j=1;j<=n;j++)
    	{
    		cin>>num[i][j];
		}
	}
	for(int i=0;i<3;i++)
	{
		sort(num[i]+1,num[i]+n+1);
	}
	LL ans=0; 
    for(int i=1;i<=n;i++)
    {
    	int mid=num[1][i];//在b[j]中依次枚举
    	int pos1=lower_bound(num[0]+1,num[0]+1+n,mid)-num[0]-1;//lower_bound函数返回的是数组中第一个大于等于mid的元素下标，这里减一就是小于mid的下标,如果找不到就返回数组尾地址的下一个地址 
    	int pos2=upper_bound(num[2]+1,num[2]+1+n,mid)-num[2];//upper_bound函数返回的是数组中第一个大于mid的元素下标 ，若找不到，同上 
    	if(pos1>=1&&pos2<=n)//对于每一个b[j],如果存在对应的a[i]和c[k] 
    	{
    		ans+=(LL)pos1*(n-pos2+1);//pos1表示的是在a[i]里有多少个数满足小于b[j],由于upper_bound函数返回的是大于b[j]的元素下标，大于b[j]的元素个数就是后面的所有，即n-pos2+1（因为下标从1开始) 
		}
	}
	cout<<ans<<endl;
	return 0;
}

---

总结：

枚举题的优化策略常见的就是这几种，后面如果遇到其他的再继续总结，希望自己变得更强，加油！！！