Chương trình C++ tính tích K nút đầu của danh sách liên kết

Cho danh sách liên kết có vài phần tử. Tìm tích của k phần tử đầu. Giá trị k cho trước. Nếu danh sách [5, 7, 3, 5, 6, 9] và k = 3, kết quả = 5 * 7 * 3 = 105.

Quy trình đơn giản: đọc phần tử từ đầu, nhân vào tích (khởi tạo 1), khi đã duyệt k phần tử thì dừng.

Ví dụ mã nguồn

#include <iostream>
#include <cmath>
using namespace std;

class Node {
public:
    int data;
    Node *next;
};

Node* getNode(int data) {
    Node *newNode = new Node;
    newNode->data = data;
    newNode->next = NULL;
    return newNode;
}

void append(struct Node** start, int key) {
    Node* new_node = getNode(key);
    Node *p = (*start);
    if (p == NULL) {
        (*start) = new_node;
        return;
    }
    while (p->next != NULL) {
        p = p->next;
    }
    p->next = new_node;
}

long long prodFirstKElements(Node *start, int k) {
    long long res = 1;
    int count = 0;
    Node* temp = start;
    while (temp != NULL && count != k) {
        res *= temp->data;
        count++;
        temp = temp->next;
    }
    return res;
}

int main() {
    Node *start = NULL;
    int arr[] = {5, 7, 3, 5, 6, 9};
    int n = sizeof(arr)/sizeof(arr[0]);
    int k = 3;
    for (int i = 0; i < n; i++) {
        append(&start, arr[i]);
    }
    cout << "Product of first k elements: " << prodFirstKElements(start, k);
}

Kết quả đầu ra

Product of first k elements: 105