include iostream include fstream using namespace std список смежных ве

  1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
#include <iostream>
#include <fstream>
using namespace std;
//список смежных вершин
struct elem
{
int name;
elem *next;
};
//вершина
struct top
{
int name;//имя - порядковый номер вершины
bool isdeleted;//была ли вершина удалена
elem* list;//указатель на список смежных вершин
} *tops;
int n;//размер массива
int k;//минимальное число членов идеологически невырожденной группы
void print_tops()
{
elem *cur;
for (int i=0; i<n; i++)
{
cout<<tops[i].name<<" "<<tops[i].isdeleted;
cur=tops[i].list;
while (cur)
{
cout<<" "<<cur->name;
cur=cur->next;
}
cout<<endl;
}
}
void add_elem(elem *&a, int b)//добавление элемента в список
{
if (!a)
{
a=new elem;
a->name=b;
a->next=NULL;
}
else
{
elem *cur=a;
while (cur->next) cur=cur->next;
cur->next=new elem;
cur=cur->next;
cur->name=b;
cur->next=NULL;
}
}
void calc_childrens(int index, int &childrens)//рекурсивная ф-ия вычисления кол-ва элементов в поддереве (результат в childrens)
{
elem *cur=tops[index].list;
while (cur)
{
if (!tops[cur->name].isdeleted) calc_childrens(cur->name, ++childrens);
cur=cur->next;
}
}
int main()
{
cout<<"Enter filename with data: ";
char filename[128];
cin>>filename;
ifstream in;
in.open(filename);
if (in.fail())
{
cout<<"ERROR! Can't open file "<<filename<<"."<<endl;
return -1;
}
in>>k>>n;
tops=new top[n];//создаём массив из n вершин
for (int i=0; i<n; i++)//заполняем его
{
tops[i].name=i;//значениями вершин от 0 до n-1
tops[i].isdeleted=false;//изначально вершины не удалены
tops[i].list=NULL;//пока сведений о смежных вершинах нет
}
int temp;
for (int i=1; i<n; i++)
{
in>>temp;
add_elem(tops[temp].list, i);//если родитель i-го элемента temp, то вершина с индексом temp смежна с вершиной i
}
in.close();
elem *deleted_tops=NULL;//список из удалённых вершин (сделан для удобства)
int count=0;//кол-во удалений
int childrens;//кол-во элементов в поддереве + сама вершина
for (int i=n-1; i>=0; i--)//просматриваем массив вершин с конца, т.е. с листьев
{
childrens=1;//у любого поддерева есть хотя бы одна вершина
calc_childrens(i, childrens);//вычисляем кол-во элементов
if (childrens>=k)//проверяем условие
{
tops[i].isdeleted=true;//удалем вершину посредством изменения соответствующего поля
add_elem(deleted_tops, i);//добавляем удалённую вершину в список удалённых
count++;
}
}
cout<<"Enter filename to save data: ";
cin>>filename;
ofstream out;
out.open(filename);
if (out.fail())
{
cout<<"ERROR! Can't save file "<<filename<<"."<<endl;
return -1;
}
out<<count<<endl;
while (deleted_tops)//вывод удалённых вершин
{
out<<deleted_tops->name<<" ";
deleted_tops=deleted_tops->next;
}
return 0;
}