【NC16622】多项式输出

题目

多项式输出

模拟，注意细节

思路

一道模拟题，按要求模拟即可，没有算法上的难度，有的是动手的复杂性。这本来可以直接在主函数中写处理过程的，但是为了其更有通用性，在这里将其提取为一个函数，具体见代码。

代码

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

/**
 * @brief 根据系数数组和最高次幂生成降幂排列的多项式
 *
 * @param coef 系数数组，长度为 n + 1
 * @param n 多项式的最高次幂
 * @param var 变量使用的字符，比如 x
 * @return char* 堆中的字符串，由调用者释放
 */
char* gen(int* coef, int n, char var) {
    // 如果字符串长度可能很长，则将这里的N修改即可
    const int N = 1000;
    char* res = (char*)malloc(N * sizeof(char));
    int t = 0;
    for (int i = n; i >= 0; i--) {
        t = coef[n - i];
        // 系数为0，直接跳过
        if (!t) continue;
        if (i < n) {
        	// 如果不是第一项，则系数前面要带符号
            sprintf(res + strlen(res), "%c", t > 0 ? '+' : '-');
        } else {
        	// 如果是第一项，则只有负数才有符号
            if (t < 0) sprintf(res + strlen(res), "-");
        }
        // 接下来是输出系数
        if (t > 1 || t < -1) {
        	// 如果是变量项且系数绝对值为 1 的话，就不用输出
        	// 只用输出绝对值大于 1 的
            sprintf(res + strlen(res), "%d", t > 0 ? t : -t);
        } else {
        	// 如果是最后一项，即常数项，则无论绝对值是多少都要输出
            if (!i) sprintf(res + strlen(res), "%d", t > 0 ? t : -t);
        }
        if (i > 0) {
        	// 如果是变量项
            if (i != 1) {
            	// 则只有指数不为 1 才输出指数
                sprintf(res + strlen(res), "%c^%d", var, i);
            } else {
            	// 否则不用输出指数
                sprintf(res + strlen(res), "%c", var);
            }
        }
    }
    // 这里是将多余的空间截断，让字符串恰好占用其长度以及一个\0的空间
    n = strlen(res);
    realloc(res, n + 1);
    res[n] = 0;
    return res;
}

int main(void) {
    int n = 0, t = 0;
    scanf("%d", &n);
    int coef[n + 1];
    for (int i = 0; i <= n; i++) {
        scanf("%d", coef + i);
    }
    char* ans = gen(coef, n, 'x');
    printf("%s\n", ans);
    free(ans);
    return 0;
}