【ACM】day2：堆与并查集

堆：

并查集：“你的爸爸的爸爸还是你的爸爸”

https://github.com/CSGrandeur/s-1problem1day1ac/discussions/405

 1.中位数

 P1168 中位数 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

#include<bits/stdc++.h>
using namespace std;
long long int a[100010],n,mid;
priority_queue<int,vector<int>,less<int> >que_big;	//大根堆 
priority_queue<int,vector<int>,greater<int> >que_small;		//小根堆 

int main()
{
    scanf("%d",&n);

	scanf("%d",&a[0]);
	mid=a[0];
	printf("%d\n",a[0]);
	
	for(int i=1;i<n;i++)
	{
		scanf("%d",&a[i]);
		if(a[i]>mid)que_small.push(a[i]);
		else que_big.push(a[i]);
		if(i%2==0)
		{
			while(que_big.size()!=que_small.size())
			{
				if(que_big.size()>que_small.size())
				{
					que_small.push(mid);
					mid=que_big.top();
					que_big.pop();
				}
				else
				{
					que_big.push(mid);
					mid=que_small.top();
					que_small.pop();
				}
			}
			printf("%d\n",mid);
		}
	}
}

题解： 难点主要是对顶堆的设计（知道有这东西），核心代码就几行

2.合并果子

（1）超时算法

#include<iostream>
#include<stdio.h>
#include<math.h>
#include <queue>
#include <vector>
#include <functional>
using namespace std;
int a[20010],n;

int main()
{
    priority_queue<int, vector<int>, less<int> >dui;
    
	scanf("%d",&n);
	int ans=0;
	
	for(int i=0;i<n;i++){
		scanf("%d",&a[i]);
		dui.push(a[i]);
	}
	for(int i=1;i<n;i++)		//第一个i代表层数 ,每层递增数目 
	{
		for(int j=0;j<pow(2,i-1);j++)		//第二个j代表每层个数,记录弹出个数 
		{
			ans+=i*dui.top();
			dui.pop();
			
		}
		
	}
	printf("%d",ans);
}

2.采用堆：合并堆

#include<iostream>
#include<stdio.h>
#include<math.h>
#include <queue>
#include <vector>
#include <functional>
using namespace std;
int a,n;

int main()
{
    priority_queue<int, vector<int>, greater<int> >dui;
    
	scanf("%d",&n);
	int ans=0,a1,a2;
	for(int i=0;i<n;i++)
	{
		scanf("%d",&a);
		dui.push(a);
	}
	while(dui.size()>=2)
	{
		a1=dui.top();dui.pop();
		a2=dui.top();dui.pop();
		ans+=a1+a2;
		dui.push(a1+a2);
	}
	
	printf("%d",ans);
}