题目1 : 对局匹配
时间限制:10000ms 单点时限:1000ms 内存限制:256MB 描述 小Hi开发了一个在线玩斗地主的游戏平台。现在平台上有N名用户正在寻找对局,其中第i名用户的积分是Ai。 小Hi希望自己的平台可以自动将这N名用户匹配成尽量多的3人牌局。同时他希望一局中的3名用户两两之间的积分差不超过K。 你能帮小Hi实现这个自动对局匹配的算法吗? 假设现在正有7人在寻找对局,积分分别是[30, 31, 30, 34, 33, 32, 34]并且K = 1,这时最多可以匹配出2局:[30, 31, 30]和[34, 33, 34]。 输入 第一行包含两个整数,N和K。 (1 ≤ N ≤ 100000, 1 ≤ K ≤ 100000) 第二行包含N整数Ai。(0 ≤ Ai ≤ 100000) 输出 一个整数表示最多能匹配出的对局数量。 样例输入 7 2 30 31 30 34 33 32 34 样例输出 2
总结:这题其实不难,有几个小点需要注意,为了防止越界,自己写for循环里面写的是N-3,,但是没有写等于,这里需要注意,比如N-K,倒数第k个节点,已经是第几个了,所以相当与多减了1,因此需要for循环里面写等于。
这题需要首先排序,这样才好进行处理。
for (int i = 0; i <= N - 3;) { if (check(num,i,K)) { ++cnt; i = i + 3; } else { ++i; } }
#include<iostream> #include<vector> #include<string> #include<algorithm> using namespace std; bool check(vector<int> &num,int begin,int K) { int i = begin; if (num[i + 2] - num[i + 1] > K) { return false; } else if (num[i + 1] - num[i ] > K) { return false; } else if(num[i + 2] - num[i] > K) { return false; } return true; } int main() { int N, K; cin >> N >> K; vector<int> num; for (int i = 0; i < N; ++i) { int tmp; cin >> tmp; num.push_back(tmp); } sort(num.begin(), num.end()); int cnt = 0; for (int i = 0; i <= N - 3;) { if (check(num,i,K)) { ++cnt; i = i + 3; } else { ++i; } } cout << cnt << endl; system("pause"); }
题目2 : 稀疏矩阵乘积
时间限制:10000ms 单点时限:1000ms 内存限制:256MB 描述 给定两个N × N的稀疏矩阵A和B,其中矩阵A有P个元素非0,矩阵B有Q个元素非0。请计算两个矩阵的乘积C = A × B并且输出C中所有非0的元素。 输入 第一行包含三个整数N, P, Q 以下P行每行三个整数i, j, k表示A矩阵的一个非0元素:Aij = k 以下Q行每行三个整数i, j, k表示B矩阵的一个非0元素:Bij = k 对于80%的数据,1 ≤ N, P, Q ≤ 200 对于100%的数据, 1 ≤ N, P, Q ≤ 2000, 1 ≤ i, j ≤ N, 0 ≤ k ≤ 100 输出 输出若干行,按先行后列的顺序输出矩阵C的每一个非0元素 每行三个整数i, j, k表示C矩阵的一个非0元素:Cij = k 样例输入 2 2 4 1 1 1 2 2 1 1 1 1 1 2 2 2 1 3 2 2 4 样例输出 1 1 1 1 2 2 2 1 3 2 2 4
动态输入数组大小,然后将所有位置初始化为0的输入方法*important。
int N; cin >> N; vector<int> tmp; tmp.resize(N);//默认全为0 vector<vector<int>> vec(tmp.size(),vector<int> (tmp.size()));//全部为0
这题有个非常非常非常大的大坑,就是ij其实是第几行和第几列,所以输入的时候需要减去1才能存入数组,检测的方法是输入N= 2,但是有i = 2的行。
这题核心就是记得矩阵相乘的方法:
定义
设A为 的矩阵,B为 的矩阵,那么称 的矩阵C为矩阵A与B的乘积,记作 ,其中矩阵C中的第 行第 列元素可以表示为:
如下所示:
思路和LeetCode的这题很像参考;
#include<iostream> #include<vector> #include<string> #include<algorithm> using namespace std; vector<vector<int>> multiply(vector<vector<int>>& A, vector<vector<int>>& B) { vector<vector<int>> res(A.size(), vector<int>(B[0].size())); for (int i = 0; i < A.size(); ++i) { for (int k = 0; k < A[0].size(); ++k) { if (A[i][k] != 0) { for (int j = 0; j < B[0].size(); ++j) { if (B[k][j] != 0) res[i][j] += A[i][k] * B[k][j]; } } } } return res; } int main() { int N, P, Q; cin >> N >> P >> Q; //cout << N << P << Q; vector<int> tmp; tmp.resize(N); vector<vector<int>> A(tmp.size(),tmp),B(A),res; for (int i = 0; i < P; ++i) { int row = 0, col = 0, value = 0; cin >> row >> col >> value; A[row - 1][col - 1] = value; } for (int j = 0; j < Q; ++j) { int row = 0, col = 0, value = 0; cin >> row >> col >> value; B[row - 1][col - 1] = value; } res = multiply(A, B); for (int i = 0; i < N; ++i) { for (int j = 0; j < N; ++j) { if (res[i][j] != 0) { cout << i + 1 << " " << j + 1 << " " << res[i][j]; cout << endl; } } } system("pause"); }