https://www.hackerrank.com/contests/w1/challenges/volleyball-match
此题不错,首先可以看出是DP,S(x, y)= S(x - 1, y) + S(x, y - 1)。然后比赛结束状态需要认真判断。三来,最后数据量很大(接近10^9)远超一般DP的数据量,分配数组都不行,里面是有规律的。下面是大数据失败的代码:
#include <vector>
#include <iostream>
#include <cstdlib>
#include <cmath>
using namespace std; vector<vector<int> > ma;
int MOD = 1000000007; bool validEnd(int a, int b) {
// a >= b
if (a < 25)
return false;
if (a == 25) {
return ((a - b) >= 2);
}
else {
return ((a - b) == 2);
}
} bool valid(int a, int b) {
// a >= b
if (a < 0 || b < 0)
return false;
if (a <= 25 && b <= 25)
return true;
else
return ((a - b) <= 2);
} int getCount(int a, int b, bool first) {
if (!valid(a, b)) {
return 0;
}
if (!first && validEnd(a, b)) {
return 0;
}
if (ma[a][b] != -1) {
return ma[a][b];
}
ma[a][b] = (getCount(a - 1, b, false) + getCount(a, b - 1, false)) % MOD;
return ma[a][b];
} int main() {
int a, b;
cin >> a >> b;
if (a < b)
swap(a, b);
if (!validEnd(a, b)) {
cout << 0 << endl;
return 0;
}
ma.resize(a + 1);
for (int i = 0; i < ma.size(); i++) {
ma[i].resize(b + 1, -1);
}
ma[0][0] = 1;
int r = getCount(a, b, true);
cout << r << endl;
}
正确做法是观察到,大于25的比赛结束状态都是由24:24经由一个一个平局过来的,没经过一次增加一倍。这样只要改写main函数基本就行了,主要pow由于次数太高,会远超long的值域,需要改写每次都MOD一下。
int pw(int k) {
if (k == 0)
return 1;
if (k == 1)
return 2;
int q = pw(k / 2);
q = (1LL * q * q) % MOD;
if (k % 2 == 1)
return (q + q) % MOD;
else
return q;
} int main() {
int a, b;
cin >> a >> b;
if (a < b)
swap(a, b);
if (!validEnd(a, b)) {
cout << 0 << endl;
return 0;
}
ma.resize(26);
for (int i = 0; i < ma.size(); i++) {
ma[i].resize(26, -1);
}
ma[0][0] = 1;
int r = 0;
if (a == 25) {
r = getCount(a, b, true);
}
else {
r = getCount(24, 24, true);
r = ((int64_t) r * pw(b - 24)) % MOD;
}
cout << r << endl;
}