天仙配[心得: 学多点c++数据结构, STL]

题目描述

今年的七夕异常的热闹，很多对新人都把结婚日子选在了那天。 于是，他们决定一起举办“天上地下姻缘一线牵”活动，在广场上举行集体婚礼。
为了活跃气氛，司仪想出了一个很有意思的游戏：会场有n(1≤n≤10000)对新人，司仪在地上画出一排(共2n个)格子， 每个格子里都写着一个随机的整数Ai(1≤Ai≤10000)。
游戏开始后，让新人们任意地站成一排。等他们都站好以后，司仪开始计算他们每个人自己的得分， 记分规则是：男方的分数等于把从自己所站的位置开始一直累加到开头，女方的分数等于从自己所站 位置开始一直累加到末尾。如果某一对新人的得分数是相同的，那你们就获胜。可以得到一份司仪精心准备的礼物。^_
比如，有3对新人，地上的那一排数字为:3，6，2，4，5，2。
如果男方站在第三个位置(2)，他的得分为：3+6+2=11；女方站在第4个位置(4)，她的得分为4+5+2=11。两人得分相同，可以获胜。
或者男方站第6个位置(2)，女方站第1个位置(3)，他们的得分都等于22，也可以获胜。
这么高兴的日子不能太扫兴，所以我们发现，无论地上的数字填成什么样子，女方站在开头和男方站在末尾就一定可以获奖。 我们不得不感叹司仪的用心良苦。呵呵。
碰巧，nowcoder的姐姐和姐夫也在那天结婚，为了帮姐姐夺得奖品，nowcoder 来请你帮忙，计算一下他姐姐获得奖品有多少种站法。

输入描述:

输入包括多组测试数据。
每组测试数据包括两行。
第一行为一个数据n，即新人的对数。
第二行有2n个数据，代表地上的数字。
输入以0结束，这一行不做处理。

输出描述:

每组输出占一行。
输出共有几种站法。

示例1

输入

3
3 6 2 4 5 2
0

输出

2

solution

他山之石大佬牛客地址

#include<iostream>
#include<vector>
#include<string>
#include<algorithm>
#include<functional>
#include <map>
#include <set>
#include <unordered_set>
#include <unordered_map>
#include <exception>
#include <iomanip>
#include <memory>
#include <sstream>
 
#define INF 1000000
using namespace std;
 
int main(int argc, char** argv)
{
    //freopen("in.txt", "r", stdin);
    int n;
    while (cin >> n && n > 0)
    {
        n <<= 1;
        vector<int> nums(n);
        for (int i = 0; i < n; ++i) cin >> nums[i];
 
        unordered_map<int,int> hs;
        int partSum = 0;
        for (int i = 0; i < n; ++i)
        {
            partSum += nums[i];
            hs.emplace(partSum, i);
        }
        int res = 0;
        partSum = 0;
        for (int i = n - 1; i >= 0; --i)
        {
            partSum += nums[i];
            if (hs.find(partSum) != hs.end() && hs[partSum] != i) ++res;
        }
 
        cout << res << endl;
    }
 
    return 0;
}

my code[还不能通过]

#include"iostream"
using namespace std;
#include"vector"

int main(){
    int n, n2, res = 1, last, g, cnt;
    cin >> n;
    n2 = 2*n;
    vector<int> v(n2);
    vector<long long> man(n2), girl(n2+1);
    for(int i = 0; i < n2; i++){
        scanf("%d", &v[i]);
        //cin >> v[i];
        printf("%d ", v[i]);
    }
    cin >> girl[n2]; //v[n2] = 0;
    man[0] = v[0];
    for(int i = 1; i < n2; i++){
        man[i] = v[i];
        man[i] += man[i-1];
        girl[n2-i] = v[n2-i];
        girl[n2-i] += girl[n2-i+1];
    }
    for(int i = 0; i < n2; i++)
        printf("v[%d] : %d, man[%d] : %d; girl[%d] : %d\n", i, v[i], i, man[i], i, girl[i]);
    g = last = cnt = n2 - 1;
    for(int i = 0; i < cnt; i++)
        for(; g; g--)
            if(i != g && man[i] == girl[g]){
                res++;
                --g;
                break;
            }
            else if(girl[g] > man[i]) 
                break;
    cout << "res == "; 
    cout << res;
    return 0;
}