题解 P4008 【[NOI2003]文本编辑器】
块状链表及其应用
思路楼上已经说的很清楚了
看代码注释
代码很丑
#include<cstdio>
#include<cctype>
#include<cstring>
//#define gc() getchar()
#define MAXIN 100000
#define gc() (SS==TT&&(TT=(SS=IN)+fread(IN,1,MAXIN,stdin),SS==TT)?EOF:*SS++)
const int N=1024*1024*2+5;
const int MaxSize=/*1700*/4500,MaxNum=/*1700*2*/N*2/MaxSize+100;
int num,nxt[MaxNum],sz[MaxNum],pool[MaxNum];
char data[MaxNum][MaxSize],str[N],IN[MAXIN],*SS=IN,*TT=IN;
inline int read() {
int now=0,f=1;
register char c=gc();
for(; !isdigit(c); c=gc()) if(c=='-') f=-1;
for(; isdigit(c); now=now*10+c-'0',c=gc());
return now*f;
}
inline int New_Block() {
return pool[++num];
}
inline void Del_Block(int v) {
pool[num--]=v;
}
void Init() {
for(int i=1; i<MaxNum; ++i) pool[i]=i;
sz[0]=0, nxt[0]=-1;//新建一个0节点，方便 表头就是0
}
void Merge(int cur,int Nxt) {
memcpy(data[cur]+sz[cur],data[Nxt],sz[Nxt]);
nxt[cur]=nxt[Nxt], sz[cur]+=sz[Nxt];
Del_Block(Nxt);
}
void Maintain() {
for(int cur=0,Nxt=nxt[0]; ~cur; cur=nxt[cur],Nxt=nxt[cur])
while((~Nxt) && sz[cur]+sz[Nxt]<=MaxSize)
Merge(cur,Nxt), Nxt=nxt[cur];//最好不用nxt[Nxt]，因为已经合并、删掉了，虽然不影响答案，但还是不该写
}
int Get_Index(int &pos) { //找到pos所在的块，并将pos定位为块内位置
int cur=0;
while((~cur) && pos>sz[cur])//把cur定位到某一块的末尾，不用下一块开头了(pos>=sz)
pos-=sz[cur], cur=nxt[cur];
return cur;
}
void Update(int cur,int Nxt,int len,char *s) { //给新的块Nxt设置数据及指针
nxt[Nxt]=nxt[cur], nxt[cur]=Nxt, sz[Nxt]=len;
memcpy(data[Nxt],s,len);
}
void Split(int cur,int pos) {
if(cur==-1||pos==sz[cur]) return;//不能判!pos！因为后边会直接用nxt[cur]，不分裂会跳过该块；而在=sz时不分是没问题的
int Nxt=New_Block();
Update(cur,Nxt,sz[cur]-pos,data[cur]+pos);
sz[cur]=pos;
}
void Insert(int pos,int len) {
int cur=Get_Index(pos),sum=0,Nxt;
Split(cur,pos);
while(sum+MaxSize<=len) {
Nxt=New_Block();
Update(cur,Nxt,MaxSize,str+sum);//先分成尽可能多的整块
sum+=MaxSize, cur=Nxt;
}
if(len-sum)//剩余的单独放到一块
Nxt=New_Block(), Update(cur,Nxt,len-sum,str+sum);
Maintain();
}
void Erase(int pos,int len) {
int cur=Get_Index(pos),Nxt;
Split(cur,pos);
Nxt=nxt[cur];//because of here
while((~Nxt) && len>sz[Nxt])
len-=sz[Nxt], Del_Block(Nxt), Nxt=nxt[Nxt];
Split(Nxt,len);
Del_Block(Nxt), nxt[cur]=nxt[Nxt];
Maintain();
}
void Get_Data(int pos,int len) {
int cur=Get_Index(pos),sum=sz[cur]-pos;
if(len<sum) sum=len;
memcpy(str,data[cur]+pos,sum);
cur=nxt[cur];
while((~cur) && sum+sz[cur]<=len)
memcpy(str+sum,data[cur],sz[cur]), sum+=sz[cur] ,cur=nxt[cur];
if((~cur) && len-sum)
memcpy(str+sum,data[cur],len-sum);
str[len]='\0';
printf("%s\n",str);
}
inline char Get_opt() {
register char c=gc();
while(c!='M'&&c!='I'&&c!='D'&&c!='G'&&c!='P'&&c!='N') c=gc();
return c;
}
int main() {
Init();
int t=read(),pos=0,len;
char s[10];
while(t--) {
switch(Get_opt()) {
case 'M':
pos=read();
break;
case 'I':
len=read();
for(int i=0; i<len; ++i) {
str[i]=gc();
if(str[i]<32||str[i]>126) --i;
}
Insert(pos,len);
break;
case 'D':
len=read(),Erase(pos,len);
break;
case 'G':
len=read(),Get_Data(pos,len);
break;
case 'P':
--pos;
break;
case 'N':
++pos;
break;
}
}
return 0;
}