Pagini recente » Cod sursa (job #1118707) | Cod sursa (job #1349613) | Cod sursa (job #1862440) | Cod sursa (job #1070377) | Cod sursa (job #1316030)
#include <fstream>
using namespace std;
ifstream fin("algsort.in");
ofstream fout("algsort.out");
struct node
{
int inf;
unsigned char inaltime; // simplifica niste operatii
node* left;
node* right;
node(int k) { inf = k; left = right = 0; inaltime = 1; }
};
unsigned char inaltime(node* p) // necesara pentru pointeri nuli
{
return p ? p->inaltime : 0;
}
int bfactor(node* p) // poate returna -2,-1,0,1,2
{
return inaltime(p->right) - inaltime(p->left);
}
void calculinaltime(node* p) //calculul inaltimii
{
unsigned char hl = inaltime(p->left);
unsigned char hr = inaltime(p->right);
p->inaltime = (hl>hr ? hl : hr) + 1;
}
node* rotdr(node* p)
{
node* q = p->left;
p->left = q->right;
q->right = p;
calculinaltime(p);
calculinaltime(q);
return q;
}
node* rotst(node* q)
{
node* p = q->right;
q->right = p->left;
p->left = q;
calculinaltime(q);
calculinaltime(p);
return p;
}
node* balance(node* p)
{
calculinaltime(p);
if (bfactor(p) == 2)
{
if (bfactor(p->right) < 0)
p->right = rotdr(p->right);
return rotst(p);
}
if (bfactor(p) == -2)
{
if (bfactor(p->left) > 0)
p->left = rotst(p->left);
return rotdr(p);
}
return p;
}
node* ins(node* p, int k)
{
if (!p) return new node(k);
if (k<p->inf)
p->left = ins(p->left, k);
else
p->right = ins(p->right, k);
return balance(p);
}
node* gasmin(node* p)
{
return p->left ? gasmin(p->left) : p;
}
node* stergemin(node* p)
{
if (p->left == 0)
return p->right;
p->left = stergemin(p->left);
return balance(p);
}
node* sterge(node* p, int k)
{
if (!p) return 0;
if (k < p->inf)
p->left = sterge(p->left, k);
else if (k > p->inf)
p->right = sterge(p->right, k);
else
{
node* q = p->left;
node* r = p->right;
delete p;
if (!r) return q; // min=tatal lui q
node* min = gasmin(r);
min->right = stergemin(r);
min->left = q;
return balance(min);
}
return balance(p);
}
void tip(node *p) // tiparire in inordine
{
if (p->left)
tip(p->left);
fout << p->inf << ' ';
if (p->right)
tip(p->right);
}
int main()
{
int n;
fin >> n;
int aux;
fin >> aux;
node* r = new node(aux);
for (int i = 2; i <= n; i++)
{
fin >> aux;
r = ins(r, aux);
}
tip(r);
}