SRM601 Div1 Easy WinterAndPresents

問題概要

N個の袋があり,i個目にはa_i個のりんごとb_i個のオレンジが入っている.それぞれの袋からx個ずつ持ってきて1つの袋にまとめるとき,りんごとオレンジの個数の組み合わせは何通りあるか.

解法

xの範囲は[1\leq x,min\{a_1+b_1,a_2+b_2,...,a_N+b_N\}]である.
それぞれについて,袋iから取ってくることができるりんごの個数は[x-min\{b_i,x\}],max\{a_i,x\}で,すべての袋についてそのminを取ったものをl,maxを取ったものをrとすると,その間の値はすべて取ることができるので,r-l+1通り.これらの和を取れば良い.

ソースコード

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define FOR(i,n,m) for(int i=(n);i<(m);i++)
#define REP(i,n) FOR(i,0,n)
#define REPR(i,n) for(int i=(n);i>=0;i--)
#define all(vec) vec.begin(),vec.end()
using vi=vector<int>;
using vvi=vector<vi>;
using vl=vector<ll>;
using vvl=vector<vl>;
using P=pair<ll,ll>;
using PP=pair<ll,P>;
using vp=vector<P>;
using vpp=vector<PP>;
using vs=vector<string>;
#define fi first
#define se second
#define pb push_back
template<class T>bool chmax(T &a,const T &b){if(a<b){a=b;return true;}return false;}
template<class T>bool chmin(T &a,const T &b){if(a>b){a=b;return true;}return false;}
const ll MOD=1000000007LL;
const int INF=1<<30;
const ll LINF=1LL<<60;

class WinterAndPresents{
    public:
    ll getNumber(vi a,vi b){
        int n=a.size();
        int mi=INF;
        REP(i,n){
            chmin(mi,a[i]+b[i]);
        }
        ll ans=0;
        FOR(i,1,mi+1){
            int r=0;
            REP(j,n){
                r+=min(a[j],i);
            }
            int l=0;
            REP(j,n){
                l+=i-min(b[j],i);
            }
            ans+=r-l+1;
        }
        return ans;
    }
};