求给定单链表中前 k 个节点的积
假设一个单链表中有一些元素。我们要找出前 k 个元素的乘积。也会提供 k 的值。因此,如果链表为 [5, 7, 3, 5, 6, 9],且 k = 3,则结果为 5 * 7 * 3 = 105。
该过程非常简单。我们从链表的左端开始依次读取元素,然后将其与 prod 相乘(最初 prod 为 1),当遍历完 k 个元素时就停止。
示例
#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); }
输出
Product of first k elements: 105
广告