P7912 [CSP-J 2021] 小熊的果篮

题目传送门

思路 :

由于数据规模很大，直接暴力模拟会时超，只最多能拿到70分。原因是我们再删除元素和处理“块”时，直接用数组十分麻烦

而且n个水果都要取一次，复杂度几乎是n方，在这里我们可以用链表。链表删除元素十分方便，且合并块也不难操作
具体做法：
    在读入的同时便判断出“块”，对每个块进行如下处理：
            在同一个“块”内 ,所有元素均指向它在该块中下一个元素的下标
            头结点（第一个元素）额外存储尾节点的编号，便于查找
            由于尾结点在一个“块”中没有下一个元素，因此尾结点的下一个元素默认为0
    这样，我们就需要用到结构体，每个元素有三个属性：
            1. 读入的值（即是苹果还是桔子）
            2. 在当前“块”中 下一个元素的编号（尾结点为0）   
            3. 尾结点的下标（只有头节点存储）
    接着我们还需要存储每个“块”的第一个元素的下标，为了方便，这里直接用vector存
    在每次删除“块”的开头时元素时（见题意），我们需要删完之后新的头元素的属性修改，并将下标保存
    一轮删除结束时，将新一轮的开头元素存入vector存，用到下一轮操作。

具体请看代码里的注释

代码：

#include<bits/stdc++.h>
using namespace std;
const int N=2e5+10;
struct stu{
    int type;//读入的值
    int next;//下一个节点下标 
    int tail;//尾结点下标 
}fr[N];//链表
vector<int> frhead;//vector变长数组，存储每个“块”的一个节点 
int main(){
    freopen("fruit.in","r",stdin);
    freopen("fruit.out","w",stdout);
    int n,head;//head用来暂时处理每个“块”的第一个节点 
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&fr[i].type);
        if(i==1||fr[i].type!=fr[i-1].type){//是第一个水果或和上一个水果不同，说明有新的“块”出现 
            frhead.push_back(i);//push_back(i),vector函数，将i加入最后
                                //当前节点为该“块”的第一个节点，加入vector 
            head=i;//记录第一个节点 
        }
        else{//不然说明与上一个元素在同一个“块”内 
            fr[i-1].next=i;//该节点为上一个节点的下一个 
            fr[head].tail=i;//由于该节点是目前“块”中最晚的，所以可能为尾结点，
                            //之后出现相同的也这样更新，最后指向的一定是尾节点 
        }
    }
    int cnt=0;//cnt记录删除的元素个数 
    while(cnt<n){//n个元素没被全部删完，继续循环 
        vector<int> last_head;//新一轮修改后的每个块的第一个节点 
        for(int i=0;i<frhead.size();i++){//size()，vector函数，返回vector中元素个数
                                         //因为vector中的元素个数就是“块”的个数，所以循环frhead.size()次 
            int head=frhead[i];//记录该“块”开头元素的下标，在后面的调用中更好理解。 
            printf("%d ",head);//先将下标输出 
            cnt++;//删除的元素个数加一 
            int nxt=fr[head].next;//由于该节点被删除，下一节点便成为新一轮的第一个节点 
            if(nxt==0) continue;//这里要特判一下：如果nxt为0，说明刚刚删除的是该块中的最后一个元素，
                                //即在该块中没有下一个元素，也就是说该块的元素被取光了，那么就不需要操作，直接continue掉 
            if(last_head.empty()||fr[nxt].type!=fr[last_head.back()].type){
            //empty(),vector函数，表示vector是否为空，vector为空，说明是第一个块
            //back(),vector函数，返回vector末尾元素的值，在这里为上一个块开头的下标 
            //无论哪种情况，都说明当前依旧是一个独立的块，需要修改这个块的信息 
                last_head.push_back(nxt);//nxt为新的开头元素，将其记录 
                fr[nxt].tail=fr[head].tail;//开头元素还要记录尾元素的下标 
            }
            else if(fr[nxt].type==fr[last_head.back()].type){//与上一个块相等，说明需要合并 
                int lasthead=last_head.back();//lasthead保存上一个块开头元素，方便调用 
                int lasttail=fr[lasthead].tail;//lasttail保存上一个块结尾元素，方便调用 
                fr[lasttail].next=nxt;//上一个块的末尾元素的下一个就是当前块的开头元素 
                fr[lasthead].tail=fr[head].tail;//由于两个块合并了，所以尾节点为当前块的尾结点 
            }
        }
        printf("\n");//全取完后别忘了加回车 
        frhead=last_head;//将新一轮的开头元素放到原vector中，这里与正常变量的赋值写法一样，这也是vector的方便之处。 
    }
    return 0;
} 

开个玩笑，这个是正确的代码：

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int maxn=200005;
int n,a[maxn],b[maxn],t[maxn],hd[maxn],hd2[maxn],nxt[maxn],fa[maxn],sum,cnt,c,ttt;
inline int find(int i){
	if(fa[i]!=i) return fa[i]=find(fa[i]);
	return i;
}
inline void merge(int x,int y){
	x=find(x),y=find(y);
	if(x>y) swap(x,y);
	fa[x]=y;
}
signed main(){
	scanf("%d",&n);
	for(int i=1;i<=n;++i){
		scanf("%d",&a[i]);
		fa[i]=i;
	}
	++cnt;
	b[1]=1,hd[1]=1;
	for(int i=2;i<=n;++i){
		if(a[i]!=a[i-1]) ++cnt,hd[cnt]=i;
		++b[cnt];
	}
	while(sum<n){
		for(int i=1;i<cnt;++i){
			printf("%d ",hd[i]);
			merge(hd[i]-1,hd[i]);
		}
		printf("%d\n",hd[cnt]);
		merge(hd[cnt]-1,hd[cnt]);
		sum+=cnt;
		c=0;
		ttt=0;
		for(int i=1;i<=cnt;++i){
			if(b[i]>1){
				if(b[i-1]>1||c==0||ttt%2==0) ++c,hd2[c]=find(hd[i])+1,t[c]=0;
				t[c]+=b[i]-1;
				ttt=0;
			}
			else ++ttt;
		}
		cnt=c;
		for(int i=1;i<=cnt;++i){
			b[i]=t[i],hd[i]=hd2[i];
		}
	}
	return 0;
}

制作不易，点个赞吧，求求啦！！！O(∩_∩)O谢谢