難易度の2倍以上の問題に対して選ぶことが出来る．
まずは，難易度でソート．
今見ている問題数をiとすると，i+1問目でその問題を選ぶことが出来る場所に足していく．
縦を問題，横を問題数と見て，遷移出来る場所を見つけたい．これを愚直に探すと，探索にかかってしまう．しかし，難易度でソートされているので，lower_boundでで見つけることが出来る．最初に見つけた場所に足していって，累積和を取った．全体でで間に合う．
Code
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
| #include <iostream>
#include <vector>
#include <string>
#include <cstring>
#include <algorithm>
#include <sstream>
#include <map>
#include <set>
#define REP(i,k,n) for(int i=k;i<n;i++)
#define rep(i,n) for(int i=0;i<n;i++)
#define INF 1<<30
#define pb push_back
#define mp make_pair
#define MOD 1000000007
using namespace std;
typedef long long ll;
typedef pair<int,int> P;
int main() {
int n;
cin >> n;
ll vmax = 0;
vector<ll> v(n);
rep(i,n) {
cin >> v[i];
vmax = max(vmax, v[i]);
}
sort(v.begin(), v.end());
ll cnt[100005][4];
memset(cnt, 0, sizeof(cnt)) ;
rep(i,n) {
cnt[i][0]++;
}
rep(i,3) {
rep(j,n) {
if(v[j]*2 > vmax) continue;
int id = lower_bound(v.begin(), v.end(), v[j]*2) - v.begin();
cnt[id][i+1] += cnt[j][i];
cnt[id][i+1] %= MOD;
}
REP(j,1,n) {
cnt[j][i+1] += cnt[j-1][i+1];
cnt[j][i+1] %= MOD;
}
}
ll ans = 0;
rep(i,n) {
ans += cnt[i][3];
ans %= MOD;
}
cout << ans << endl;
return 0;
}
|
2倍するので，オーバーフローしてWA．llに直した時に，cntの初期化を消してしまったらしく，それに気づかずWAを連発してしまった．．．