# |
Submission time |
Handle |
Problem |
Language |
Result |
Execution time |
Memory |
144500 |
2019-08-17T00:04:24 Z |
12tqian |
Lottery (CEOI18_lot) |
C++14 |
|
0 ms |
0 KB |
#pragma comment(linker, "/stack:200000000")
#pragma GCC optimize("Ofast")
#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
//#pragma GCC optimize ("O3")
//#pragma GCC target ("sse4")
#include <bits/stdc++.h>
#include <ext/pb_ds/tree_policy.hpp>
#include <ext/pb_ds/assoc_container.hpp>
using namespace std;
using namespace __gnu_pbds;
template <class T> using Tree = tree<T, null_type, less<T>, rb_tree_tag,tree_order_statistics_node_update>;
const double PI = 4 * atan(1);
#define sz(x) (int)(x).size()
#define ll long long
#define ld long double
#define mp make_pair
#define pb push_back
#define eb emplace_back
#define pii pair <int, int>
#define vi vector<int>
#define f first
#define s second
#define lb lower_bound
#define ub upper_bound
#define all(x) x.begin(), x.end()
#define vpi vector<pair<int, int>>
#define vpd vector<pair<double, double>>
#define pd pair<double, double>
#define f0r(i,a) for(int i=0;i<a;i++)
#define f1r(i,a,b) for(int i=a;i<b;i++)
#define trav(a, x) for (auto& a : x)
template<typename A, typename B> ostream& operator<<(ostream &cout, pair<A, B> const &p) { return cout << "(" << p.f << ", " << p.s << ")"; }
template<typename A> ostream& operator<<(ostream &cout, vector<A> const &v) {
cout << "["; for(int i = 0; i < v.size(); i++) {if (i) cout << ", "; cout << v[i];} return cout << "]";
}
void fast_io(){
ios_base::sync_with_stdio(0);
cin.tie(NULL);
cout.tie(NULL);
}
void io(string taskname){
string fin = taskname + ".in";
string fout = taskname + ".out";
const char* FIN = fin.c_str();
const char* FOUT = fout.c_str();
freopen(FIN, "r", stdin);
freopen(FOUT, "w", stdout);
fast_io();
}
const int MAX = 1e4 + 5;
vector<short int> a;
int n, l;
short int** pre;
int** cnt;
inline short int gap(int i, int j){
if(i>j) swap(i, j);
return l - (pre[j-i][j+l-1] -(j == 0?0: pre[j-i][j-1]));
}
int main(){
fast_io();
scanf("%d %d", &n, &l);
f0r(i, n){
short int ai;
scanf("%d", &ai);
a.eb(ai);
}
pre = new int*[MAX];
cnt = new int*[MAX];
f0r(i, n){
pre[i] = new int[MAX];
}
f0r(i, n) f0r(j, n) pre[i][j] = 0;
f1r(diff, 1, n){
f0r(i, n){
if(i-diff<0) continue;
if(i-diff==0) pre[diff][i] = (a[i] == a[i-diff]);
else pre[diff][i] = pre[diff][i-1] + (a[i] == a[i-diff]);
}
}
f0r(i, n){
cnt[i] = new int[MAX];
}
f0r(i, n) f0r(j, n) cnt[i][j] = 0;
f1r(d,1, n){
f0r(i, n-l+1){
int j = i +d;
if(j>=n-l+1)continue;
int val = gap(i, j);
cnt[i][val]++;
cnt[j][val]++;
}
delete[] pre[d];
}
delete[] pre;
f0r(i, n-l+1){
f1r(j,1, n){
cnt[i][j] += cnt[i][j-1];
}
}
int q;
scanf("%d", &q);
//cout << q << endl;
while(q--){
int k;
scanf("%d", &k);
f0r(i, n-l + 1){
printf("%d ", cnt[i][k]);
}
printf("\n");
}
return 0;
}
Compilation message
lot.cpp:1:0: warning: ignoring #pragma comment [-Wunknown-pragmas]
#pragma comment(linker, "/stack:200000000")
lot.cpp: In function 'int main()':
lot.cpp:70:24: warning: format '%d' expects argument of type 'int*', but argument 2 has type 'short int*' [-Wformat=]
scanf("%d", &ai);
~~~^
lot.cpp:73:23: error: cannot convert 'int**' to 'short int**' in assignment
pre = new int*[MAX];
^
lot.cpp:76:29: error: cannot convert 'int*' to 'short int*' in assignment
pre[i] = new int[MAX];
^
lot.cpp: In function 'void io(std::__cxx11::string)':
lot.cpp:52:12: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)', declared with attribute warn_unused_result [-Wunused-result]
freopen(FIN, "r", stdin);
~~~~~~~^~~~~~~~~~~~~~~~~
lot.cpp:53:12: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)', declared with attribute warn_unused_result [-Wunused-result]
freopen(FOUT, "w", stdout);
~~~~~~~^~~~~~~~~~~~~~~~~~~
lot.cpp: In function 'int main()':
lot.cpp:67:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%d %d", &n, &l);
~~~~~^~~~~~~~~~~~~~~~~
lot.cpp:70:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%d", &ai);
~~~~~^~~~~~~~~~~
lot.cpp:107:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%d", &q);
~~~~~^~~~~~~~~~
lot.cpp:111:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%d", &k);
~~~~~^~~~~~~~~~