XORSums/code2.txt at main · SaarthakMarkandey/XORSums · GitHub

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
65
66
67
68
69
70
71
72
#include <iostream>
#include <stdio.h>
using namespace std;
#define lli long long int

lli sum1;
lli sum2;
void combinationUtil(int arr[], int n, int r,
                     int index, int data[], int i);

void printCombination(int arr[], int n, int r)
{

    int data[r];
    sum1=0;

    combinationUtil(arr, n, r, 0, data, 0);
}

void combinationUtil(int arr[], int n, int r, int index,
                     int data[], int i)
{

    if (index == r) {
        sum2=data[0];
        for (int j = 1; j < r; j++)
            sum2=sum2^data[j];

            sum1+=sum2;
        return;
    }

    if (i >= n)
        return;

    data[index] = arr[i];
    combinationUtil(arr, n, r, index + 1, data, i + 1);


    combinationUtil(arr, n, r, index, data, i + 1);
}


int main() {
	int N;
	cin>>N;
	int num[N];
	lli sumin=0;
	for(int i=0;i<N;i++){
	    cin>>num[i];
	    sumin+=num[i];
	}
	lli finalsum;
	int Q,M;
	cin>>Q;
	while(Q--){
	    cin>>M;
	    finalsum=0;
	    if(M!=1){
	    for(int i=2; i<=M;i++){
	        printCombination(num,N,i);
	        finalsum+=sum1;
	    }
	      finalsum+=sumin;
	    }
	    else {
	        finalsum=sumin;
	    }
	    cout << finalsum%998244353 <<endl;
	}
	return 0;
}