这题我们数据范围太大,用二维肯定是不行的,我们可以采用一维线性存储。
如题意,我们可以将每组奶牛编号都存在一维数组里面,只需记录每组的头尾指针就可以了。
如题中样例我们就可以存储成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;
}
算法小白的刷题日记.