[HDOJ 4879] ZCC loves march [模拟+数据结构]

给一个10^18*10^18的棋盘，上边一共有10^5个棋子，棋子可以重叠。现在告诉你棋子的初始位置，以及10^5个操作。一种操作是让某个棋子向某个方向移动若干步，另一种操作是让某个棋子把同行同列的棋子都聚集到和他一个格子上。要求对每个第二种操作输出所需费用。费用定义为每个棋子移动的距离的平方的和。

以格子为单位，记录这个格子的横纵坐标，格子内棋子个数，以及这个格子是否被整体迁移到了某个其他的格子上。

使用一个map来记录某个横纵坐标位置上，实际放的是哪个格子。

再使用一个map来记录某行都有哪些格子，再使用一个map来记录某列都有哪些格子。

#include <cstring>
#include <cstdio>
#include <set>
#include <utility>
#include <vector>
#include <map>

using namespace std;

const int mod=1000000007;

struct Node {
	long long x,y;
	int to,troop;
	void clear(long long xx,long long yy) {
		to=-1;
		x=xx;y=yy;
		troop=1;
	}
};

Node b[200101];
int a[100100],bp;
map<pair<long long,long long>,int> c;
map<long long,set<int> > row;
map<long long,set<int> > col;
int n;
long long m;

void print() {
	for (int i=1;i<bp;i++) {
		printf("Node %d: (%I64d,%I64d) %d num:%d\n",i,b[i].x,b[i].y,b[i].to,b[i].troop);
	}
	for (int i=0;i<n;i++) {
		printf("%d ",a[i]);
	}
	printf("\n");
	for (map<pair<long long,long long>,int>::iterator it=c.begin();it!=c.end();it++) {
		printf("Pos (%I64d,%I64d): %d\n",it->first.first,it->first.second,it->second);
	}
	for (map<long long,set<int> >::iterator it=row.begin();it!=row.end();it++) {
		printf("---row %I64d\n",it->first);
		for (set<int>::iterator itr=it->second.begin();itr!=it->second.end();itr++) {
			printf("%d ",*itr);
		}
		printf("\n");
	}
	for (map<long long,set<int> >::iterator it=col.begin();it!=col.end();it++) {
		printf("---col %I64d\n",it->first);
		for (set<int>::iterator itr=it->second.begin();itr!=it->second.end();itr++) {
			printf("%d ",*itr);
		}
		printf("\n");
	}
}
void init() {
	bp=1;
	c.clear();
	row.clear();
	col.clear();
}
void addTroop(int i,long long x,long long y) {
	int tmp=c[make_pair(x,y)];
	if (tmp==0) {
		c[make_pair(x,y)]=bp;
		row[x].insert(bp);
		col[y].insert(bp);
		b[bp].clear(x,y);
		tmp=bp++;
	} else {
		b[tmp].troop++;
	}
	a[i]=tmp;
}
int getPos(int i) {
	int tmp=a[i];
	while (b[tmp].to!=-1) tmp=b[tmp].to;
	return a[i]=tmp;
}
void delTroop(int i) {
	int tmp=getPos(i);
	b[tmp].troop--;
	if (b[tmp].troop==0) {
		row[b[tmp].x].erase(tmp);
		col[b[tmp].y].erase(tmp);
		c[make_pair(b[tmp].x,b[tmp].y)]=0;
	}
}
int regroup(int i) {
	long long x=b[i].x;
	long long y=b[i].y;
	int ans=0;
	set<int> &rr=row[x];
	for (set<int>::iterator it=rr.begin();it!=rr.end();it++) {
		if (*it!=i) {
			b[*it].to=i;
			b[i].troop+=b[*it].troop;
			long long tmpy=b[*it].y;
			c[make_pair(x,tmpy)]=0;
			col[tmpy].erase(*it);
			tmpy=(tmpy-y)%mod;
			ans=(ans+tmpy*tmpy%mod*b[*it].troop)%mod;
		}
	}
	rr.clear();
	rr.insert(i);
	set<int> &cc=col[y];
	for (set<int>::iterator it=cc.begin();it!=cc.end();it++) {
		if (*it!=i) {
			b[*it].to=i;
			b[i].troop+=b[*it].troop;
			long long tmpx=b[*it].x;
			c[make_pair(tmpx,y)]=0;
			row[tmpx].erase(*it);
			col[y].erase(*it);
			tmpx=(tmpx-x)%mod;
			ans=(ans+tmpx*tmpx%mod*b[*it].troop)%mod;
		}
	}
	cc.clear();
	cc.insert(i);
	return ans;
}

int main() {
	int i,lastans=0;
	long long x,y;
	char cc;
	while (scanf("%d%I64d",&n,&m)!=EOF) {
		init();
		for (i=0;i<n;i++) {
			scanf("%I64d%I64d",&x,&y);
			addTroop(i,x,y);
		}
		int t;
		//print();
		scanf("%d",&t);
		while (t--) {
			scanf(" %c",&cc);
			if (cc=='U') {
				scanf("%d%I64d",&i,&x);
				i^=lastans;
				i--;
				int tmp=getPos(i);
				x=b[tmp].x-x;
				y=b[tmp].y;
				delTroop(i);
				addTroop(i,x,y);
			} else if (cc=='L') {
				scanf("%d%I64d",&i,&y);
				i^=lastans;
				i--;
				int tmp=getPos(i);
				x=b[tmp].x;
				y=b[tmp].y-y;
				delTroop(i);
				addTroop(i,x,y);
			} else if (cc=='D') {
				scanf("%d%I64d",&i,&x);
				i^=lastans;
				i--;
				int tmp=getPos(i);
				x=b[tmp].x+x;
				y=b[tmp].y;
				delTroop(i);
				addTroop(i,x,y);
			} else if (cc=='R') {
				scanf("%d%I64d",&i,&y);
				i^=lastans;
				i--;
				int tmp=getPos(i);
				x=b[tmp].x;
				y=b[tmp].y+y;
				delTroop(i);
				addTroop(i,x,y);
			} else if (cc=='Q') {
				scanf("%d",&i);
				i^=lastans;
				int tmp=getPos(i-1);
				int ans=regroup(tmp);
				printf("%d\n",ans);
				lastans=ans;
			}
			//print();
		}
	}
	return 0;
}