ACM模式处理输入输出大全（多组数据、数组、链表、二叉树链接整理）

本文链接：https://blog.youkuaiyun.com/qq_41667348/article/details/126012736

文章目录

1、一维数组输入：

关键代码：

int main() {
	int n;
    cin >> n;
    vector<int> nums(n);
    for(int i = 0; i < n; i++) {
	   cin >> nums[i];
   }
}

2、旋转矩阵，手写输入二维矩阵

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

void rotate(vector<vector<int>>& matrix) {
    int m = matrix.size();
    int n = matrix[0].size();

    for (int i = 0; i < m / 2; i++) {
        for (int j = 0; j < n; j++) {
            swap(matrix[i][j], matrix[m - i - 1][j]);
        }
     }

    for (int i = 0; i < m; i++) {
        for (int j = 0; j < i; j++) {
            swap(matrix[i][j], matrix[j][i]);
        }
    }
}

int main() {
    int m, n, temp;
    cin >> m;
    cin >> n;
    vector<vector<int>> matrix(m, vector<int>(n));
    for (int i = 0; i < m; i++) {
        for (int j = 0; j < n; j++) {
            cin >> temp;
            matrix[i][j] = temp;
        }
    }
    rotate(matrix);
    for (int i = 0; i < m; i++) {
        for (int j = 0; j < n; j++) {
            cout << matrix[i][j] << " ";
        }
         cout << endl;
     }
     
     system("PAUSE");
     return 0;
}

在这里插入图片描述

3、合并两个链表，手写输入链表。

第一种写法，自建链表

///#include <bits/stdc++.h> //包含所有c++头文件
//#include <iostream>
//#include <vector>
//using namespace std;
//struct ListNode {
//    int val;
//    ListNode* next;
//    ListNode(int x) : val(x), next(NULL) {}
//};
//ListNode* createList(vector<int>& nums) { //链表转换为数组
//    if (nums.size() == 0)
//        return NULL;
//    ListNode* head = new ListNode(nums[0]);
//    ListNode* curNode = head;
//    for (int i = 1; i < nums.size(); ++i) {
//        curNode->next = new ListNode(nums[i]);
//        curNode = curNode->next;
//    }
//    return head;
//}
//ListNode* MergeTwoListNode(ListNode* l1, ListNode* l2) { //合并两个链表
//    if (!l1)
//        return l2;
//    if (!l2)
//        return l1;
//    if (l1->val < l2->val) {
//        l1->next = MergeTwoListNode(l1->next, l2);
//        return l1;
//    }
//    else {
//        l2->next = MergeTwoListNode(l1, l2->next);
//        return l2;
//    }
//}
//int main()
//{
 //    vector<int> nums1, nums2;
//    int num1, num2;
//    while (cin >> num1)
//    {
//        nums1.push_back(num1);
//        if (cin.get() == '\n')
//            break;
//    }
//    while (cin >> num2)
//    {
//        nums2.push_back(num2);
//        if (cin.get() == '\n')
//            break;
//    }
//    ListNode* head1 = createList(nums1);
//    ListNode* head2 = createList(nums2);
//    ListNode* res = MergeTwoListNode(head1, head2);
//    ListNode* p = res;
//    while (p)
//    {
//        cout << p->val << " ";
//        p = p->next;
//    }
//    return 0;
//}

//方法2，直接放入数组中：

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
int main()
{
	vector<int> nums1, nums2, nums;
 	int num1, num2;
	 while (cin >> num1)
	{
   		nums1.push_back(num1);
    	nums.push_back(num1);
    	if (cin.get() == '\n')
        	break;
	}
	while (cin >> num2)
	{
    	nums2.push_back(num2);
    	nums.push_back(num2);
    	if (cin.get() == '\n')
        	break;
 	}
	sort(nums.begin(), nums.end());
	for (int i = 0;i < nums.size();i++)
    	cout << nums[i] << " ";
	return 0;
}

在这里插入图片描述

4、层序遍历，手写输入输出二叉树

#include <iostream>;
#include<queue>
#include <vector>
using namespace std;

struct TreeNode {
	int val;
	TreeNode* left;
   TreeNode* right;
   TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
};

void treePrint(TreeNode* root) {
	if (!root) return;
	queue<TreeNode*> que;
	que.push(root);
	while (!que.empty()) {
    	int size = que.size();
    	for (int i = 0; i < size; i++) {
        	TreeNode* node = que.front();
        	que.pop();
        	if (!node) cout << "-1 ";
        	else {
            	cout << node->val << " ";
            	que.push(node->left);
            	que.push(node->right);
       		 }
   		}
    	cout << endl;
	}
}

//需要补齐-1当前节点的左右节点
TreeNode* build(const vector<int>& nums) {
	vector<TreeNode*> tree;
	for (auto x : nums) {
    	if (x == -1) tree.push_back(nullptr);
    	else tree.push_back(new TreeNode(x));
	}
	int idx = 1;
	for (int i = 0; i < tree.size(); i++) {
   		if (tree[i] == nullptr) continue;
   	    if (idx >= tree.size()) break;
    	tree[i]->left = tree[idx];
    	tree[i]->right = tree[idx + 1];
    	idx += 2;
	}
   return tree[0];
}

int main() {
	int n;
	cin >> n;
	vector<int> nums(n);
	for (int i = 0; i < n; i++) {
		cin >> nums[i];
	}
	TreeNode* root = build(nums);
	treePrint(root);
	system("pause");
	return 0;
}

在这里插入图片描述