洛谷：P3068 [USACO13JAN] Party Invitations S（枚举、前缀和）

这题我们数据范围太大，用二维肯定是不行的，我们可以采用一维线性存储。

如题意，我们可以将每组奶牛编号都存在一维数组里面，只需记录每组的头尾指针就可以了。

如题中样例我们就可以存储成1 3 3 4 1 2 3 4 5 6 7 4 3 2 1

然后第一组头尾指针就是 1,2 。

第二组头尾指针就是3,4。

第二组头尾指针就是 5,11。

第二组头尾指针就是12,15。

#include<iostream>
using namespace std;
const int N = 1e6 + 10;
pair<int, int> q[N]; //q用于记录每组的头和尾
int e[N], d[N]; //d用于记录每组的数量，e用于将所有组的奶牛线性存储
int n, g, ans;
bool f[N]; //用于判断当前编号是否被邀请过
bool fz[N];//用于判断每组是否都邀请过
int main()
{
	cin >> n >> g;
	int idx = 0;
	for(int i=1;i<=g;i++)
	{
		cin >> d[i];
		for (int j = 1; j <= d[i]; j++)
		{
			cin >> e[++idx];  //将所有组奶牛线性存储，
		}
		//存储每组的头尾指针
		//如例题e中存入的是1 3 3 4 1 2 3 4 5 6 4 3 2 1
		//第一组是1 3，那么q[1]的头尾指针就对应1 3下标，也就是1，2
		int head = idx - d[i] + 1; 
		int tail = idx;
		q[i] = { head,idx };
	}
	f[1] = true;
	int mm = 0;
	do
	{
		mm = 0;
		for (int i = 1; i <= g; i++)
		{
			int hh = q[i].first;
			int tt = q[i].second;
			int c = 0; //当前组邀请的人数
			int flag = 0; //记录没被邀请人的下标
			for (int j = hh; j <= tt; j++)
			{
				if (f[e[j]]) //如果当前编号已经被邀请了，当前组邀请的人数++
					c++;
				else
					flag = j;
			}
			if (c == d[i] - 1)
			{
				f[e[flag]] = true; //邀请前组唯一一个没被邀请的人
				fz[i] = true;  //现在当前组就全部被邀请了
				mm = 1;  //既然已经新邀请了一个人，那么就要判断是否还要再继续邀请
			}
			if (c == d[i]) //如果相等就说明改组已经全部邀请
				fz[i] = true;
		}
	} while (mm == 1); //如果mm=1才说明需要进一步邀请，否则就不需要邀请了

	for (int i = 1; i <= n; i++)
		if (f[i])ans++; //如果被邀请数量就+1
	cout << ans << endl;

	return 0;
}

算法小白的刷题日记.