Java B组蓝桥杯第七届国赛：圆圈舞

圆圈舞 

春天温暖的阳光照耀着大地，正是草原上的小动物们最快乐的时候。小动物们在草原上开了一个舞会，欢度这美好的时光。

舞会上最重要的一个环节就是跳圆舞曲，n只小动物手拉手围成一大圈，随着音乐跳起来。在跳的过程中，小动物们可能会变换队形。它们的变换方式是动物A松开自己右手，动物B松开自己的左手，动物A和B手拉到一起，而它们对应的松开的手（如果有的话）也拉到一起。

例如，假设有10只小动物，按顺序围成一圈，动物1的右手拉着动物2的左手，动物2的右手拉着动物3的左手，依次类推，最后动物10的右手拉着动物1的左手。如果通过动物2和8变换队形，则动物2的右手拉着动物8的左手，而对应的动物3的左手拉着动物7的右手，这样形成了1-2-8-9-10和3-4-5-6-7两个圈。如果此时通过动物2和6变换队形，则将形成1-2-6-7-3-4-5-8-9-10一个大圈。注意，如果此时通过动物1和2变换队形，那么队形不会改变，因为动物1的右手和动物2的左手松开后又拉到一起了。

在跳舞的过程中，每个动物i都有一个欢乐值Hi和一个感动值Fi。
如果两个动物在一个圈中，欢乐值会彼此影响，产生欢乐能量。如果两个动物i, j（i≠j）在同一个大小为t的圈中，而动物i在动物j右手的第p个位置（动物j右手的第1个位置就是动物j右手所拉着的动物，而第2个位置就是右手第1个位置的动物右手拉着的动物，依次类推），则产生的欢乐能量为(t-p)*Hj*Fi。在跳舞的过程中，动物们的欢乐值和感动值有可能发生变化。

圆舞曲开始的时候，所有的动物按编号顺序围成一个圈，动物n右手的第i个位置正好是动物i。现在已知小动物们变换队形的过程和欢乐值、感动值变化的过程，求每次变换后所有动物所产生的欢迎能量之和。

【输入格式】
输入的第一行包含一个整数n，表示动物的数量。
接下来n行，每行两个用空格分隔的整数Hi, Fi，按编号顺序给出每只动物的欢乐值和感动值。
接下来一行包含一个整数m，表示队形、欢乐值、感动值的变化次数。
接下来m行，每行三个用空格分隔的整数k, p, q，当k=1时，表示小动物们通过动物p和动物q变换了队形，当k=2时，表示动物p的欢乐值变为q，当k=3时，表示动物p的感动值变为了q。

【输出格式】
输出m行，每行一个整数，表示每次变化后所有动物产生的能量之和。
答案可能很大，你需要计算答案除以1000000007的余数。

【样例输入】
10
1 1
1 1
1 1
1 1
1 1
1 1
1 1
1 1
1 1
1 1
9
1 2 8
1 2 6
2 8 10
3 5 10
1 1 2
1 2 1
2 5 5
1 4 8
1 4 5

【样例输出】
100
450
855
1341
1341
811
923
338
923

【数据规模与约定】
对于20%的数据，2<=n,m<=100。
对于30%的数据，2<=n,m<=1000。
另有20%的数据，只有k=1的操作且Hi,Fi均为1。
另有20%的数据，只有k=1或2的操作且Fi均为1。
对于100%的数据，2<=n,m<=100000，0<=Hi,Fi<=10^9，1<=k<=3，k=1时1<=p,q<=n且p≠q，k=2或3时1<=p<=n且0<=q<=10^9。

资源约定：
峰值内存消耗 < 256M
CPU消耗  < 5000ms

请严格按要求输出，不要画蛇添足地打印类似：“请您输入...” 的多余内容。

所有代码放在同一个源文件中，调试通过后，拷贝提交该源码。
注意：不要使用package语句。不要使用jdk1.7及以上版本的特性。
注意：主类的名字必须是：Main，否则按无效代码处理。
 

这个题除了麻烦，耗时间，难度不是很大，但是比赛的话时间紧迫，嗯~~够呛，里面还埋有几个坑成功让我掉进去了。。。。

公式需要注意，t-p就是指从i到j的距离，我在代码中直接用参数p表示了

测试网址：https://www.dotcpp.com/oj/problem1836.html

代码如下：

import java.util.ArrayList;
import java.util.HashSet;
import java.util.Iterator;
import java.util.List;
import java.util.Scanner;
import java.util.Set;

public class Main {

	int n; // 动物数量
	int m; // 操作次数
	int[][] operate; // 操作内容
	int[] result; // 操作结果
	Animal[] animals;// 动物存放索引
	/*
	 * .这里我用了一个Set集合做为一个圈 .用列表用什么都可以 .只要能找到这个圈所有的动物就行
	 */
	List<Set<Animal>> list = new ArrayList<Set<Animal>>(); // 存放每个圈

	class Animal {
		int number; // 自身编号
		int happy; // 欢乐值
		int affect; // 感动值
		Animal left; // 左手
		Animal right; // 右手

		public Animal(int n, int h, int a) {
			this.number = n;
			this.happy = h;
			this.affect = a;
		}

	}

	public Main() {
		Scanner sn = new Scanner(System.in);
		sn = new Scanner(System.in);
		n = sn.nextInt();
		animals = new Animal[n];
		for (int i = 0; i < n; i++) {
			animals[i] = new Animal(i + 1, sn.nextInt(), sn.nextInt());
		}
		animals[0].left = animals[n - 1];
		animals[0].right = animals[1];
		animals[n - 1].left = animals[n - 2];
		animals[n - 1].right = animals[0];
		for (int i = 1; i < n - 1; i++) {
			animals[i].left = animals[i - 1];
			animals[i].right = animals[i + 1];
		}
		m = sn.nextInt();
		operate = new int[m][3];
		for (int i = 0; i < m; i++) {
			operate[i][0] = sn.nextInt();
			operate[i][1] = sn.nextInt();
			operate[i][2] = sn.nextInt();
		}
		result = new int[m];
		/***************** 以上都是接收输入信息，初始化空间 *******************************/
		list.add(createSet(animals[0]));// 先加入一个完整的圈
		for (int i = 0; i < m; i++) {
			change(operate[i]);
			for (int j = 0; j < list.size(); j++) {
				result[i] += getEnergySum(list.get(j));
			}
		}
		for (int a : result) {
			System.out.println(a);
		}

	}

	// 给出一个动物，利用其右手索引，收集其所在圈的所有动物，存入一个集合
	public Set<Animal> createSet(Animal animal) {
		Set<Animal> set = new HashSet<Animal>();
		while (!set.contains(animal)) {
			set.add(animal);
			animal = animal.right;
		}
		return set;
	}

	// 处理一条命令
	public void change(int[] once) {
		if (once[0] == 1) {
			// 变换队形
			divide(once[1] - 1, once[2] - 1);
		} else if (once[0] == 2) {
			// 修改欢乐值
			animals[once[1] - 1].happy = once[2];
		} else if (once[0] == 3) {
			// 修改感动值
			animals[once[1] - 1].affect = once[2];
		}
	}

	// 变换队列
	public void divide(int i, int j) {
		// 标记： false两个动物不在一个圈，true两个动物在一个圈
		boolean sign = false;
		// 遍历所有圈
		for (int a = 0; a < list.size(); a++) {
			// 当前圈中含有动物i
			if (list.get(a).contains(animals[i])) {
				// 当前圈中含有动物j,表示两个动物在一个圈
				if (list.get(a).contains(animals[j])) {
					// 如果两个在一个圈里，还是连续的，ij按顺序的，不需要处理，跳出函数
					if (i + 1 == j)
						return;
					// 修改标志，移除这个圈
					sign = true;
					list.remove(list.get(a));
					a--;// a--是因为list大小改变了
					// 两个动物在一个圈中，移除过这个圈之后就可以直接中断遍历
					break;
				} else {
					// 两个动物不在一个圈，修改标志，移除这个圈
					sign = false;
					list.remove(list.get(a));
					a--;
					continue;
				}
			}
			// 当前圈中不含有动物i,但是含有j
			if (list.get(a).contains(animals[j])) {
				// 移除动物j所在的圈
				list.remove(list.get(a));
				a--;
			}
		}

		if (sign) {// 两个在一个圈，拆分出两条
			animals[i].right.left = animals[j].left;
			animals[j].left.right = animals[i].right;
			list.add(createSet(animals[i].right));
			animals[i].right = animals[j];
			animals[j].left = animals[i];
			list.add(createSet(animals[i]));
		} else {// 不在一个圈，合并两个圈
			animals[i].right.left = animals[j].left;
			animals[j].left.right = animals[i].right;
			animals[i].right = animals[j];
			animals[j].left = animals[i];
			list.add(createSet(animals[i]));
		}
	}

	// 收集这个圈中的所有能量
	public int getEnergySum(Set<Animal> set) {
		int sum = 0;
		Iterator<Animal> iterator = set.iterator();
		// 遍历圈内的动物,收集每个动物产生的能量
		while (iterator.hasNext()) {
			// 取出一个动物计算该动物和圈内每个动物的能量
			Animal headAnimal = iterator.next();
			// 取出自身编号作为循环参照
			int headnumber = headAnimal.number;
			// 记录一个动物右手到取出的动物的距离
			int p = 0;
			// 先取出右手边的第一个动物方便进入循环
			Animal nextAnimal = headAnimal.right;
			// 如果右手边动物的编号与起点动物的不一致，表示还没统计完一圈
			while (nextAnimal.number != headnumber) {
				p++;
				sum = (p * nextAnimal.happy * headAnimal.affect);// 公式求和
				nextAnimal = nextAnimal.right;// 将位置传给右手边的动物
			}
		}
		return sum;

	}

	public static void main(String[] args) {
		new Main();

	}

}