用链表来玩多项式的加法

       周末, 小雨。用链表来玩多项式的加法, 程序如下:

// 要判断malloc的返回值, 为简便示意, 我省略了
// 结点要释放,为简便示意, 我省略了

#include <stdio.h>
#include <stdlib.h>

typedef struct node
{
    int iCoef;   //系数
    int iExpon;  //指数
    struct node* pNext;
}Node;


Node* CreateNode(int coe, int exp)
{
    Node* p = (Node*)malloc(sizeof(Node));
    p->iCoef = coe;
    p->iExpon = exp;
    p->pNext = NULL;
	
    return p;
}


Node* CreatePoly1()
{
    int coe = 0;
	int exp = 0;
    Node* pTmp = NULL;
    Node* pItem = NULL;
    Node* Poly = (Node*)malloc(sizeof( Node)); //带头结点的单链表
    Poly->pNext = NULL;
    pTmp = Poly;

	coe = 1;
	exp = 2;
	pItem = CreateNode(coe, exp);
	pTmp->pNext = pItem;
	pTmp = pItem;

	coe = 2;
	exp = 4;
	pItem=CreateNode(coe, exp);
	pTmp->pNext = pItem;
	pTmp = pItem;

	coe = 9;
	exp = 7;
	pItem = CreateNode(coe, exp);
	pTmp->pNext = pItem;
	pTmp = pItem;

    return Poly;
}


Node* CreatePoly2()
{
    int coe = 0;
	int exp = 0;
    Node* pTmp = NULL;
    Node* pItem = NULL;
    Node* Poly = (Node*)malloc(sizeof(Node));//带头结点的单链表
    Poly->pNext = NULL;
    pTmp = Poly;

	coe = 2;
	exp = 3;
	pItem = CreateNode(coe, exp);
	pTmp->pNext = pItem;
	pTmp = pItem;

	coe = -2;
	exp = 4;
	pItem = CreateNode(coe, exp);
	pTmp->pNext = pItem;
	pTmp = pItem;

	coe = -3;
	exp = 5;
	pItem = CreateNode(coe, exp);
	pTmp->pNext = pItem;
	pTmp = pItem;

	coe = 4;
	exp = 6;
	pItem = CreateNode(coe, exp);
	pTmp->pNext = pItem;
	pTmp = pItem;

    return Poly;
}


void PrintPoly(Node* Poly) // 带头结点, Poly不会为空
{
	Poly = Poly->pNext;
    while(Poly != NULL)
	{
		printf("%2d %2d   ", Poly->iCoef, Poly->iExpon);
        Poly = Poly->pNext;
    }

	printf("\n");
}


void AddPoly(Node* Poly1, Node* Poly2)	// 计算多项式的加法, 结果保存在Poly1链表中, 指针Poly1和Poly2都不为NULL,所以我不加指针判断
{
	Node *p1_prior = Poly1;
	Node *p2_prior = Poly2;
	
    Node *p1 = Poly1->pNext;
    Node *p2 = Poly2->pNext;

    while(p1 && p2)
	{
       if(p1->iExpon == p2->iExpon)
	   	{
            if(p1->iCoef + p2->iCoef == 0) 
			{
				// 删除结点p1, 且后移p1
            	p1_prior->pNext = p1->pNext;
				p1 = p1->pNext;
            }
			else
			{
				// p1后移
				p1->iCoef += p2->iCoef;
				p1_prior = p1;
				p1 = p1->pNext;
			}

			// 结点p2必然要删除, 且p2后移
			p2_prior->pNext = p2->pNext;
			p2 = p2->pNext;
       }
       else if(p1->iExpon > p2->iExpon) // 把结点p2插入到Poly1中
	   {
			p2_prior->pNext = p2->pNext;
			p2->pNext = p1;
           	p1_prior->pNext = p2;
			p1_prior = p2;              // 插入后, p1虽然不变,但p1_prior需要变, 我第一次写这个程序时, 漏了这一句,导致结果错误,TNND.
			p2 = p2_prior->pNext;
       }
       else  // p1后移
	   {
	   		p1_prior = p1;
            p1 = p1->pNext;
       }

    }
	
    if(p2)
	{
        p1_prior->pNext = p2;
    }

    return;
}


int main()
{
    Node *Poly1 = CreatePoly1();  // x^2 + 2x^4 + 9x^7
    Node *Poly2 = CreatePoly2();  // 2x^3 - 2x^4 - 3x^5 + 4x^6
	PrintPoly(Poly1);
	PrintPoly(Poly2);

    AddPoly(Poly1, Poly2);
	PrintPoly(Poly1);            // x^2 + 2x^3 - 3x^5 + 4x^6 + 9x^7
	
    return 0;
}

        结果:

 1  2    2  4    9  7   
 2  3   -2  4   -3  5    4  6   
 1  2    2  3   -3  5    4  6    9  7  

        不多说。

猜你喜欢

转载自blog.csdn.net/stpeace/article/details/81463357