贪心算法笔记
大概内容
贪心就是对于一个问题有很多个步骤,我们在每一个步骤中都选取最优的那一个,最后得出答案。就是在一些函数中可行,但是有些比如二次函数,因为它的转折点不一定最优,就是不可行的。那么如何判断贪心呢?有这么几种
- 看时间复杂度,一般的就是 O ( n ) O(n) O(n) 或者是排序 O ( n l o g n ) O(n \ log n) O(n logn)
- 或者猜测,看着像就可以试试。
- 自己用数学证明方法,比如归纳法,交换法,就是如果交换之后答案变得小或者大就可以了。
那贪心在比赛中怎么办呢?可以先尝试一下大小样例,再尝试对拍,有个保底,如果可以也可以证明一下。
然后还有一种贪心,叫做反悔贪心,就是可以撤销,也没别的。
最后因为贪心每次取局部最优,所以代码不长,而且时间复杂度不大。
例题讲解
凌乱的yyy / 线段覆盖
题目大意:有 n n n 个线段,问最多有多少个不重叠的线段。
思路:贪心,然后按照 r i r_{i} ri 排序,每次选取尽可能早的场次并且要合法,也就是不能前面重叠。
证明:如果不相交,那么图片如下。
很明显,一场弄完,另一场。如果包含,图片如下:
那样我们就可以选择第一场,因为它结束得早,可以取另外的场次。所以证明取更早的是永远比更晚的要更优。
时间复杂度: O ( n log n ) O(n \log n) O(nlogn)
#include<bits/stdc++.h>
struct noip//有同学不会结构体的多开几个数组
{
int begin;//开始时间
int finish;//结束时间
}a[2000005];
bool cmp(noip x,noip y)
{
return x.finish<y.finish;//按结束时间排
}
using namespace std;
int main()
{
int n,ans=1;//初始化
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>a[i].begin>>a[i].finish;
}
sort(a+1,a+n+1,cmp);//快速排序(不用我多说了吧)
int mini=a[1].finish;//定义mini统计最后结束时间
int j=1;
while(j<=n)//我用了while,感兴趣的同学可以用for(提示:for不用定义j)
{
j++;
if(a[j].begin>=mini)//新的比赛开始时间要晚于mini
{
ans++;//统计比赛数
mini=a[j].finish;//更新mini
}
}
cout<<ans<<endl;//输出
return 0;//功德圆满
}
Teleporters (Easy Version)
思路:直接 a i + i a_{i}+i ai+i 排序,就好了。
证明:就是每次都是走过去,传回来,所以直接排序就可以了。
时间复杂度: O ( n log n ) O(n \log n) O(nlogn)
#include<bits/stdc++.h>
using namespace std;
struct node
{
int x,id;
}c[200005];
int n,m,tmp;
bool cmp(node a,node b)
{
return a.x+a.id<=b.x+b.id;//跟思路一样排序
}
int main()
{
int t;
cin>>t;
while(t--)
{
memset(c,0,sizeof(c));
cin>>n>>m;
for(int i=1;i<=n;i++)
{
cin>>tmp;
c[i].x=tmp;
c[i].id=i;
}
sort(c+1,c+n+1,cmp);
int i=0;
while(m>=0&&i<=n)
{
i++;
m=m-c[i].x-c[i].id;
}
cout<<i-1<<endl;
}
return 0;
}
Teleporters (Hard Version)
思路:这一题就是先前缀和记录一下能用的传送门的价值之和,然后在使用二分答案,但是因为起点是 0 0 0 的缘故,所以还要另外枚举 0 0 0 点。
#include<bits/stdc++.h>
#define maxn 2900001
#define int long long
using namespace std;
int T,n,m,ans,a[maxn],s[maxn];
int check(int c,int x,int id)
{
//二分最长的合法前缀区间
int l=1,r=n,sum=-1;
while(l<=r)
{
int mid=l+r>>1;
if(s[mid]-(mid>=id?x:0)<=c)
sum=mid+(mid>=id?-1:0),l=mid+1;//特判x的影响
else
r=mid-1;
}
return sum==-1?0:sum;//可能无解
}
signed main()
{
scanf("%lld",&T);
while(T--)
{
ans=0;
map<int,int>mp;//记录位置
scanf("%lld%lld",&n,&m);
for(int i=1;i<=n;i++)
{
scanf("%lld",&a[i]);
s[i]=min(a[i]+(n-i+1),a[i]+i);//取最小花费
}
sort(s+1,s+n+1);
for(int i=1;i<=n;i++)
mp[s[i]]=i,s[i]+=s[i-1];//前缀和
for(int i=1;i<=n;i++)
{
if(m-a[i]-i<0)
continue;//可能不合法
ans=max(ans,check(m-a[i]-i,min(a[i]+(n-i+1),a[i]+i),mp[min(a[i]+(n-i+1),a[i]+i)])+1);//注意要加上当前点i
}
printf("%lld\n",ans);
}
return 0;
}
国王游戏
思路:就是按照 a i b i a_ib_i aibi的大小从小到大排序
证明:我们不妨设两个人分别写了 a 1 , b 1 , a 2 , b 2 a_1,b_1,a_2,b_2 a1,b1,a2,b2 且满足 a 1 b 1 > a 2 b 2 a_1b_1>a_2b_2 a1b1>a2b2 则有 a 2 b 1 = a 1 b 2 \tfrac{a_2}{b_1}=\tfrac{a_1}{b_2} b1a2=b2a1
所以有一下两种情况
- 1在2的前面
- 1在2的后面
设在设在 1 , 2 1,2 1,2 之前所有人左手乘积为 k k k,那么对于第一种情况,我们的答案就是
a n s 1 = max ( k b 1 , k a 1 b 2 )