Submission #1066123

#TimeUsernameProblemLanguageResultExecution timeMemory
1066123vjudge1Swapping Cities (APIO20_swap)C++17
Compilation error
0 ms0 KiB
///*** Sown_Vipro ***/// /// ->TUYEN QUOC GIA<- /// #include<bits/stdc++.h> using namespace std; //#pragma GCC optimize ("O3") //#pragma GCC optimize ("unroll-loops") //#pragma GCC target("popcnt") #define F first #define S second #define pb push_back #define pi pair<int, int> #define pii pair<int, pair<int, int> > #define all(a) a.begin(), a.end() #define FOR(i, a, b) for(int i = a; i <= b; ++i) #define REP(i, a, b) for(int i = a; i >= b; --i) #define inp(name) if(fopen(name, "r")) freopen(name, "r", stdin); #define out(name) if(fopen(name, "w")) freopen(name, "w", stdout); //#define szz(s) int(s.size()) #define int long long const int N = (1 << 20) + 5, MAX = 1e6, oo = 1e9 + 5, MOD = 1e9 + 7; int n, k, l, r, res; int a[N]; void fixmod(int &x){ if(x >= MOD) x -= MOD; } namespace sub1{ int f[260][260][260]; void solve(){ FOR(i, 1, n){ f[i][1][a[i]] += 1; fixmod(f[i][1][a[i]]); FOR(t, 1, k){ FOR(x, 0, min(r, 255ll)){ f[i + 1][t][x] += f[i][t][x]; fixmod(f[i + 1][t][x]); if((x | a[i + 1]) <= r){ f[i + 1][t + 1][x | a[i + 1]] += f[i][t][x]; fixmod(f[i + 1][t + 1][x | a[i + 1]]); } } } } // cout << f[2][1][3]; for(int x = l; x <= min(r, 255ll); ++x) if(x % 3 == 0){ // cout << x << " " << f[n][k][x] << "\n"; res += f[n][k][x]; fixmod(res); } cout << res; } } namespace sub2{ int f[N], cnt[N][25], t; int gt[N], nd[N], c[N]; int mul(int a, int b){ if(b == 0) return 1; a %= MOD; int r = mul(a, b / 2); int t = (1ll * r * r) % MOD; if(b % 2) return (1ll * t * a) % MOD; else return t; } void pre(){ gt[0] = nd[0] = 1; for(int i = 1; i <= n; ++i){ gt[i] = (1ll * gt[i - 1] * i) % MOD; nd[i] = mul(gt[i], MOD - 2); } } int C(int k, int n){ if(k < 0) return 0; if(k > n) return 0; return ((((1ll * gt[n] * nd[k]) % MOD) * 1ll * nd[n - k]) % MOD) % MOD; } void solve(){ if(l % 3 != 0){ cout << 0; return; } pre(); t = __builtin_popcount(l) % 2; FOR(i, 1, n) ++cnt[a[i]][0]; // cout << cnt[1][0] << "\n"; FOR(mask, 0, (1 << 20) - 1){ if((mask | l) != l) continue; FOR(k, 1, 20){ if(mask & (1 << k - 1)){ cnt[mask][k] = (cnt[mask][k - 1] + cnt[mask ^ (1 << k - 1)][k - 1]); fixmod(cnt[mask][k]); }else cnt[mask][k] = cnt[mask][k - 1]; } f[mask] = C(k, cnt[mask][20]); int tmp = __builtin_popcount(mask) % 2; int s = (tmp == t ? 1 : -1); // cout << mask << " " << cnt[mask][20] << " " << f[mask] << "\n"; res = (res + s * f[mask] + MOD) % MOD; } // cout << cnt[9][1] << "\n"; cout << res; } } void solve(){ cin >> n >> k >> l >> r; FOR(i, 1, n) cin >> a[i]; if(l == r) sub2::solve(); else sub1::solve(); } signed main(){ ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0); inp("in.txt"); // out("out.txt"); int t = 1; // cin >> t; while(t--){ solve(); } }

Compilation message (stderr)

swap.cpp: In function 'void sub2::solve()':
swap.cpp:91:35: warning: suggest parentheses around '-' inside '<<' [-Wparentheses]
   91 |                 if(mask & (1 << k - 1)){
      |                                 ~~^~~
swap.cpp:92:75: warning: suggest parentheses around '-' inside '<<' [-Wparentheses]
   92 |                     cnt[mask][k] = (cnt[mask][k - 1] + cnt[mask ^ (1 << k - 1)][k - 1]);
      |                                                                         ~~^~~
swap.cpp: In function 'int main()':
swap.cpp:17:47: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   17 | #define inp(name) if(fopen(name, "r")) freopen(name, "r", stdin);
      |                                        ~~~~~~~^~~~~~~~~~~~~~~~~~
swap.cpp:118:5: note: in expansion of macro 'inp'
  118 |     inp("in.txt");
      |     ^~~
/usr/bin/ld: /tmp/ccl0V4Rj.o: in function `main':
grader.cpp:(.text.startup+0x0): multiple definition of `main'; /tmp/ccj6DLok.o:swap.cpp:(.text.startup+0x0): first defined here
/usr/bin/ld: /tmp/ccl0V4Rj.o: in function `main':
grader.cpp:(.text.startup+0x1c3): undefined reference to `init(int, int, std::vector<int, std::allocator<int> >, std::vector<int, std::allocator<int> >, std::vector<int, std::allocator<int> >)'
/usr/bin/ld: grader.cpp:(.text.startup+0x217): undefined reference to `getMinimumFuelCapacity(int, int)'
collect2: error: ld returned 1 exit status