UVa : 12028 (A Gift from the Setter)



// http://uva.onlinejudge.org/external/120/12028.html
// Runtime: 0.512s
// Tag: Adhoc 

// @BEGIN_OF_SOURCE_CODE

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <string>
#include <cctype>
#include <stack>
#include <queue>
#include <list>
#include <vector>
#include <map>
#include <sstream>
#include <cmath>
#include <bitset>
#include <utility>
#include <set>
#include <numeric>

#define INF_MAX 2147483647
#define INF_MIN -2147483647
#define pi acos(-1.0)
#define N 1000000
#define LL long long

#define For(i, a, b) for( int i = (a); i < (b); i++ )
#define Fors(i, sz) for( size_t i = 0; i < sz.size (); i++ )
#define Fore(it, x) for(typeof (x.begin()) it = x.begin(); it != x.end (); it++)
#define Set(a, s) memset(a, s, sizeof (a))
#define Read(r) freopen(r, "r", stdin)
#define Write(w) freopen(w, "w", stdout)

int dr [] = {-1, -1, 0, 1, 1, 1, 0, -1};
int dc [] = {0, 1, 1, 1, 0, -1, -1, -1};

using namespace std;

long long a [100000 + 10];
long long accu_sum [100000 + 10];

int main ()
{
    int testCase; scanf ("%d", &testCase);
    int cases = 0;

    while ( testCase-- ) {
        long long k, c, n; scanf ("%lld %lld %lld %lld", &k, &c, &n, &a [0]);

        for ( int i = 1; i < n; i++ )
            a [i] = (k * a [i - 1] + c) % 1000007;

        sort (a, a + n);

        for ( int i = 0; i < n; i++ ) accu_sum [i] = a [i];

        for ( int i = 1; i < n; i++ ) accu_sum [i] += accu_sum [i - 1];

        long long sum = 0;

        for ( int i = 0; i < n; i++ ) {
            long long first = a [i] * (n - i - 1);
            long long second = accu_sum [n - 1] - accu_sum [i];
            sum += abs (first - second);
        }

        printf ("Case %d: %lld\n", ++cases, sum);

    }

    return 0;
}

// @END_OF_SOURCE_CODE

Leave a Reply

Fill in your details below or click an icon to log in:

WordPress.com Logo

You are commenting using your WordPress.com account. Log Out / Change )

Twitter picture

You are commenting using your Twitter account. Log Out / Change )

Facebook photo

You are commenting using your Facebook account. Log Out / Change )

Google+ photo

You are commenting using your Google+ account. Log Out / Change )

Connecting to %s